提交 f17ede83 作者: Brian Tiger Chow

refactor(integrationtest) use RoutingOptions in mocknet repo

上级 9e5b3b77
...@@ -104,12 +104,12 @@ func DirectAddCat(data []byte, conf testutil.LatencyConfig) error { ...@@ -104,12 +104,12 @@ func DirectAddCat(data []byte, conf testutil.LatencyConfig) error {
return errors.New("test initialization error") return errors.New("test initialization error")
} }
adder, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(peers[0], mn.Host(peers[0]), conf))) adder, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(peers[0], mn.Host(peers[0]), conf, core.DHTOption)))
if err != nil { if err != nil {
return err return err
} }
defer adder.Close() defer adder.Close()
catter, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(peers[1], mn.Host(peers[1]), conf))) catter, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(peers[1], mn.Host(peers[1]), conf, core.DHTOption)))
if err != nil { if err != nil {
return err return err
} }
......
...@@ -34,7 +34,7 @@ func TestBitswapWithoutRouting(t *testing.T) { ...@@ -34,7 +34,7 @@ func TestBitswapWithoutRouting(t *testing.T) {
var nodes []*core.IpfsNode var nodes []*core.IpfsNode
for _, p := range peers { for _, p := range peers {
n, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(p, mn.Host(p), conf))) n, err := core.NewIPFSNode(ctx, core.ConfigOption(MocknetTestRepo(p, mn.Host(p), conf, core.DHTOption)))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
......
...@@ -4,6 +4,7 @@ import ( ...@@ -4,6 +4,7 @@ import (
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context" context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
"github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore" "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore"
syncds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore/sync" syncds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore/sync"
ds2 "github.com/jbenet/go-ipfs/util/datastore2"
blockstore "github.com/jbenet/go-ipfs/blocks/blockstore" blockstore "github.com/jbenet/go-ipfs/blocks/blockstore"
core "github.com/jbenet/go-ipfs/core" core "github.com/jbenet/go-ipfs/core"
bitswap "github.com/jbenet/go-ipfs/exchange/bitswap" bitswap "github.com/jbenet/go-ipfs/exchange/bitswap"
...@@ -11,16 +12,14 @@ import ( ...@@ -11,16 +12,14 @@ import (
host "github.com/jbenet/go-ipfs/p2p/host" host "github.com/jbenet/go-ipfs/p2p/host"
peer "github.com/jbenet/go-ipfs/p2p/peer" peer "github.com/jbenet/go-ipfs/p2p/peer"
"github.com/jbenet/go-ipfs/repo" "github.com/jbenet/go-ipfs/repo"
dht "github.com/jbenet/go-ipfs/routing/dht"
delay "github.com/jbenet/go-ipfs/thirdparty/delay" delay "github.com/jbenet/go-ipfs/thirdparty/delay"
eventlog "github.com/jbenet/go-ipfs/thirdparty/eventlog" eventlog "github.com/jbenet/go-ipfs/thirdparty/eventlog"
ds2 "github.com/jbenet/go-ipfs/util/datastore2"
testutil "github.com/jbenet/go-ipfs/util/testutil" testutil "github.com/jbenet/go-ipfs/util/testutil"
) )
var log = eventlog.Logger("epictest") var log = eventlog.Logger("epictest")
func MocknetTestRepo(p peer.ID, h host.Host, conf testutil.LatencyConfig) core.ConfigOption { func MocknetTestRepo(p peer.ID, h host.Host, conf testutil.LatencyConfig, routing core.RoutingOption) core.ConfigOption {
return func(ctx context.Context) (*core.IpfsNode, error) { return func(ctx context.Context) (*core.IpfsNode, error) {
const kWriteCacheElems = 100 const kWriteCacheElems = 100
const alwaysSendToPeer = true const alwaysSendToPeer = true
...@@ -30,22 +29,26 @@ func MocknetTestRepo(p peer.ID, h host.Host, conf testutil.LatencyConfig) core.C ...@@ -30,22 +29,26 @@ func MocknetTestRepo(p peer.ID, h host.Host, conf testutil.LatencyConfig) core.C
} }
ds := r.Datastore() ds := r.Datastore()
log.Debugf("MocknetTestRepo: %s %s %s", p, h.ID(), h) n := &core.IpfsNode{
dhtt := dht.NewDHT(ctx, h, ds) Peerstore: h.Peerstore(),
Repo: r,
PeerHost: h,
Identity: p,
}
dhtt, err := routing(ctx, n)
if err != nil {
return nil, err
}
bsn := bsnet.NewFromIpfsHost(h, dhtt) bsn := bsnet.NewFromIpfsHost(h, dhtt)
bstore, err := blockstore.WriteCached(blockstore.NewBlockstore(ds), kWriteCacheElems) bstore, err := blockstore.WriteCached(blockstore.NewBlockstore(ds), kWriteCacheElems)
if err != nil { if err != nil {
return nil, err return nil, err
} }
exch := bitswap.New(ctx, p, bsn, bstore, alwaysSendToPeer) exch := bitswap.New(ctx, p, bsn, bstore, alwaysSendToPeer)
return &core.IpfsNode{ n.Blockstore = bstore
Peerstore: h.Peerstore(), n.Exchange = exch
Blockstore: bstore, n.Routing = dhtt
Exchange: exch, return n, nil
Repo: r,
PeerHost: h,
Routing: dhtt,
Identity: p,
}, nil
} }
} }
...@@ -80,17 +80,17 @@ func RunThreeLeggedCat(data []byte, conf testutil.LatencyConfig) error { ...@@ -80,17 +80,17 @@ func RunThreeLeggedCat(data []byte, conf testutil.LatencyConfig) error {
if len(peers) < numPeers { if len(peers) < numPeers {
return errors.New("test initialization error") return errors.New("test initialization error")
} }
bootstrap, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[2], mn.Host(peers[2]), conf)) bootstrap, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[2], mn.Host(peers[2]), conf, core.DHTOption))
if err != nil { if err != nil {
return err return err
} }
defer bootstrap.Close() defer bootstrap.Close()
adder, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[0], mn.Host(peers[0]), conf)) adder, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[0], mn.Host(peers[0]), conf, core.DHTOption))
if err != nil { if err != nil {
return err return err
} }
defer adder.Close() defer adder.Close()
catter, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[1], mn.Host(peers[1]), conf)) catter, err := core.NewIPFSNode(ctx, MocknetTestRepo(peers[1], mn.Host(peers[1]), conf, core.DHTOption))
if err != nil { if err != nil {
return err return err
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论