Commit e19b2b82 authored by Steven Allen's avatar Steven Allen

test: fix unique addr check

If we have, e.g., multiple transports, we'll have multiple addresses.
Really, we were just trying to make sure the DHT returned unique
addresses.
parent 2a8c43b2
...@@ -15,6 +15,7 @@ import ( ...@@ -15,6 +15,7 @@ import (
record "github.com/libp2p/go-libp2p-record" record "github.com/libp2p/go-libp2p-record"
swarmt "github.com/libp2p/go-libp2p-swarm/testing" swarmt "github.com/libp2p/go-libp2p-swarm/testing"
bhost "github.com/libp2p/go-libp2p/p2p/host/basic" bhost "github.com/libp2p/go-libp2p/p2p/host/basic"
"github.com/multiformats/go-multiaddr"
) )
var wancid, lancid cid.Cid var wancid, lancid cid.Cid
...@@ -356,14 +357,20 @@ func TestFindPeer(t *testing.T) { ...@@ -356,14 +357,20 @@ func TestFindPeer(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if len(p.Addrs) != 1 { assertUniqueMultiaddrs(t, p.Addrs)
t.Fatalf("expeced find peer to find 1 address, found %d", len(p.Addrs))
}
p, err = d.FindPeer(ctx, wan.PeerID()) p, err = d.FindPeer(ctx, wan.PeerID())
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if len(p.Addrs) != 1 { assertUniqueMultiaddrs(t, p.Addrs)
t.Fatalf("expeced find peer to find addresses, found %d", len(p.Addrs)) }
func assertUniqueMultiaddrs(t *testing.T, addrs []multiaddr.Multiaddr) {
set := make(map[string]bool)
for _, addr := range addrs {
if set[string(addr.Bytes())] {
t.Errorf("duplicate address %s", addr)
}
set[string(addr.Bytes())] = true
} }
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment