coreapi/dht: fix test panic

License: MIT Signed-off-by: Łukasz Magiera <magik6k@gmail.com>
ipfs · Mar 30, 2018 · ae14365 · ae14365
1 parent 5f75628
commit ae14365
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/core/coreapi/dht_test.go b/core/coreapi/dht_test.go
@@ -27,8 +27,8 @@ func TestDhtFindPeer(t *testing.T) {
 
 	addr := <-out
 
-	if addr.String() != "/ip4/127.0.0.1/tcp/4001" {
-		t.Errorf("got unexpected address from FindPeer: %s", addr.String())
+	if addr == nil || addr.String() != "/ip4/127.0.0.1/tcp/4001" {
+		t.Errorf("got unexpected address from FindPeer: %s", addr)
 	}
 
 	out, err = apis[1].Dht().FindPeer(ctx, peer.ID(nds[2].Identity))
@@ -38,8 +38,8 @@ func TestDhtFindPeer(t *testing.T) {
 
 	addr = <-out
 
-	if addr.String() != "/ip4/127.0.2.1/tcp/4001" {
-		t.Errorf("got unexpected address from FindPeer: %s", addr.String())
+	if addr == nil || addr.String() != "/ip4/127.0.2.1/tcp/4001" {
+		t.Errorf("got unexpected address from FindPeer: %s", addr)
 	}
 }