mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-25 04:17:44 +08:00
07b064010e
refactor test peer creation to be deterministic and reliable a bit of cleanup trying to figure out TestGetFailure add test to verify deterministic peer creation switch put RPC over to use getClosestPeers rm 0xDEADC0DE fix queries not searching peer if its not actually closer
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"crypto/rand"
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
|
|
ds "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"
|
|
"github.com/jbenet/go-ipfs/blocks/blockstore"
|
|
blockservice "github.com/jbenet/go-ipfs/blockservice"
|
|
ci "github.com/jbenet/go-ipfs/crypto"
|
|
"github.com/jbenet/go-ipfs/exchange/offline"
|
|
mdag "github.com/jbenet/go-ipfs/merkledag"
|
|
nsys "github.com/jbenet/go-ipfs/namesys"
|
|
"github.com/jbenet/go-ipfs/net/mock"
|
|
path "github.com/jbenet/go-ipfs/path"
|
|
peer "github.com/jbenet/go-ipfs/peer"
|
|
dht "github.com/jbenet/go-ipfs/routing/dht"
|
|
ds2 "github.com/jbenet/go-ipfs/util/datastore2"
|
|
"github.com/jbenet/go-ipfs/util/testutil"
|
|
)
|
|
|
|
// TODO this is super sketch. Deprecate and initialize one that shares code
|
|
// with the actual core constructor. Lots of fields aren't initialized.
|
|
// Additionally, the context group isn't wired up. This is as good as broken.
|
|
|
|
// NewMockNode constructs an IpfsNode for use in tests.
|
|
func NewMockNode() (*IpfsNode, error) {
|
|
ctx := context.TODO()
|
|
nd := new(IpfsNode)
|
|
|
|
// Generate Identity
|
|
sk, pk, err := ci.GenerateKeyPair(ci.RSA, 1024, rand.Reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
p, err := peer.IDFromPublicKey(pk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
nd.Identity = p
|
|
nd.PrivateKey = sk
|
|
nd.Peerstore = peer.NewPeerstore()
|
|
nd.Peerstore.AddPrivKey(p, sk)
|
|
nd.Peerstore.AddPubKey(p, pk)
|
|
nd.Network, err = mocknet.New(ctx).AddPeer(sk, testutil.RandLocalTCPAddress()) // effectively offline
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Temp Datastore
|
|
dstore := ds.NewMapDatastore()
|
|
nd.Datastore = ds2.CloserWrap(syncds.MutexWrap(dstore))
|
|
|
|
// Routing
|
|
dht := dht.NewDHT(ctx, nd.Identity, nd.Network, nd.Datastore)
|
|
nd.Routing = dht
|
|
|
|
// Bitswap
|
|
bstore := blockstore.NewBlockstore(nd.Datastore)
|
|
bserv, err := blockservice.New(bstore, offline.Exchange(bstore))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
nd.DAG = mdag.NewDAGService(bserv)
|
|
|
|
// Namespace resolver
|
|
nd.Namesys = nsys.NewNameSystem(dht)
|
|
|
|
// Path resolver
|
|
nd.Resolver = &path.Resolver{DAG: nd.DAG}
|
|
|
|
return nd, nil
|
|
}
|