Clean up reprovide test to reflect cleanup up routing mock
License: MIT Signed-off-by: Zander Mackie <zmackie@gmail.com>
Zander Mackie committed
Jan 20, 2017 at 16:40 UTC
32e92c72ebd01033f8d394766a81cdf989b190fc
1 file changed
+9
-5
exchange/reprovide/reprovide_test.go
+9
-5
@@ -10,6 +10,7 @@ import (
10
testutil "github.com/ipfs/go-ipfs/thirdparty/testutil"
11
ds "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore"
12
dssync "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore/sync"
13
+ pstore "gx/ipfs/QmeXj9VAjmYQZxpmVz7VzccbJrpmr8qkCDSjfVNsPTWTYU/go-libp2p-peerstore"
14
15
. "github.com/ipfs/go-ipfs/exchange/reprovide"
16
)
@@ -37,16 +38,19 @@ func TestReprovide(t *testing.T) {
38
t.Fatal(err)
39
}
40
40
- provs, err := clB.FindProviders(ctx, blk.Cid())
41
- if err != nil {
42
- t.Fatal(err)
41
+ var providers []pstore.PeerInfo
42
+ maxProvs := 100
43
+
44
+ provChan := clB.FindProvidersAsync(ctx, blk.Cid(), maxProvs)
45
+ for p := range provChan {
46
+ providers = append(providers, p)
47
}
48
45
- if len(provs) == 0 {
49
+ if len(providers) == 0 {
50
t.Fatal("Should have gotten a provider")
51
}
52
49
- if provs[0].ID != idA.ID() {
53
+ if providers[0].ID != idA.ID() {
54
t.Fatal("Somehow got the wrong peer back as a provider.")
55
}
56
}