mirror of
https://github.com/fluencelabs/go-libp2p-kad-dht
synced 2025-07-03 22:51:35 +00:00
dht/test: providers test id compare
This commit is contained in:
13
dht_test.go
13
dht_test.go
@ -334,10 +334,13 @@ func TestProvidesMany(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var providers = map[u.Key]peer.ID{}
|
||||||
|
|
||||||
d := 0
|
d := 0
|
||||||
for k, v := range testCaseValues {
|
for k, v := range testCaseValues {
|
||||||
d = (d + 1) % len(dhts)
|
d = (d + 1) % len(dhts)
|
||||||
dht := dhts[d]
|
dht := dhts[d]
|
||||||
|
providers[k] = dht.self
|
||||||
|
|
||||||
t.Logf("adding local values for %s = %s (on %s)", k, v, dht.self)
|
t.Logf("adding local values for %s = %s (on %s)", k, v, dht.self)
|
||||||
err := dht.putLocal(k, v)
|
err := dht.putLocal(k, v)
|
||||||
@ -370,13 +373,17 @@ func TestProvidesMany(t *testing.T) {
|
|||||||
getProvider := func(dht *IpfsDHT, k u.Key) {
|
getProvider := func(dht *IpfsDHT, k u.Key) {
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
|
|
||||||
|
expected := providers[k]
|
||||||
|
|
||||||
provchan := dht.FindProvidersAsync(ctxT, k, 1)
|
provchan := dht.FindProvidersAsync(ctxT, k, 1)
|
||||||
select {
|
select {
|
||||||
case prov := <-provchan:
|
case prov := <-provchan:
|
||||||
if prov.ID == "" {
|
actual := prov.ID
|
||||||
|
if actual == "" {
|
||||||
errchan <- fmt.Errorf("Got back nil provider (%s at %s)", k, dht.self)
|
errchan <- fmt.Errorf("Got back nil provider (%s at %s)", k, dht.self)
|
||||||
} else if prov.ID != dhts[3].self {
|
} else if actual != expected {
|
||||||
errchan <- fmt.Errorf("Got back wrong provider (%s at %s)", k, dht.self)
|
errchan <- fmt.Errorf("Got back wrong provider (%s != %s) (%s at %s)",
|
||||||
|
expected, actual, k, dht.self)
|
||||||
}
|
}
|
||||||
case <-ctxT.Done():
|
case <-ctxT.Done():
|
||||||
errchan <- fmt.Errorf("Did not get a provider back (%s at %s)", k, dht.self)
|
errchan <- fmt.Errorf("Did not get a provider back (%s at %s)", k, dht.self)
|
||||||
|
Reference in New Issue
Block a user