1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-07-01 02:30:39 +08:00

Clean up reprovide test to reflect cleanup up routing mock

License: MIT
Signed-off-by: Zander Mackie <zmackie@gmail.com>
This commit is contained in:
Zander Mackie
2017-01-20 16:40:42 -05:00
parent 3f479d1653
commit 32e92c72eb

View File

@ -10,6 +10,7 @@ import (
testutil "github.com/ipfs/go-ipfs/thirdparty/testutil" testutil "github.com/ipfs/go-ipfs/thirdparty/testutil"
ds "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore" ds "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore"
dssync "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore/sync" dssync "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore/sync"
pstore "gx/ipfs/QmeXj9VAjmYQZxpmVz7VzccbJrpmr8qkCDSjfVNsPTWTYU/go-libp2p-peerstore"
. "github.com/ipfs/go-ipfs/exchange/reprovide" . "github.com/ipfs/go-ipfs/exchange/reprovide"
) )
@ -37,16 +38,19 @@ func TestReprovide(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
provs, err := clB.FindProviders(ctx, blk.Cid()) var providers []pstore.PeerInfo
if err != nil { maxProvs := 100
t.Fatal(err)
provChan := clB.FindProvidersAsync(ctx, blk.Cid(), maxProvs)
for p := range provChan {
providers = append(providers, p)
} }
if len(provs) == 0 { if len(providers) == 0 {
t.Fatal("Should have gotten a provider") t.Fatal("Should have gotten a provider")
} }
if provs[0].ID != idA.ID() { if providers[0].ID != idA.ID() {
t.Fatal("Somehow got the wrong peer back as a provider.") t.Fatal("Somehow got the wrong peer back as a provider.")
} }
} }