mirror of
https://github.com/ipfs/kubo.git
synced 2025-06-07 00:39:36 +08:00
123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/ipfs/go-filestore"
|
|
keystore "github.com/ipfs/go-ipfs-keystore"
|
|
"github.com/ipfs/kubo/core"
|
|
"github.com/ipfs/kubo/core/bootstrap"
|
|
"github.com/ipfs/kubo/core/coreapi"
|
|
mock "github.com/ipfs/kubo/core/mock"
|
|
"github.com/ipfs/kubo/core/node/libp2p"
|
|
"github.com/ipfs/kubo/repo"
|
|
|
|
"github.com/ipfs/go-datastore"
|
|
syncds "github.com/ipfs/go-datastore/sync"
|
|
coreiface "github.com/ipfs/interface-go-ipfs-core"
|
|
"github.com/ipfs/interface-go-ipfs-core/tests"
|
|
"github.com/ipfs/kubo/config"
|
|
"github.com/libp2p/go-libp2p/core/crypto"
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
mocknet "github.com/libp2p/go-libp2p/p2p/net/mock"
|
|
)
|
|
|
|
const testPeerID = "QmTFauExutTsy4XP6JbMFcw2Wa9645HJt2bTqL6qYDCKfe"
|
|
|
|
type NodeProvider struct{}
|
|
|
|
func (NodeProvider) MakeAPISwarm(ctx context.Context, fullIdentity bool, n int) ([]coreiface.CoreAPI, error) {
|
|
mn := mocknet.New()
|
|
|
|
nodes := make([]*core.IpfsNode, n)
|
|
apis := make([]coreiface.CoreAPI, n)
|
|
|
|
for i := 0; i < n; i++ {
|
|
var ident config.Identity
|
|
if fullIdentity {
|
|
sk, pk, err := crypto.GenerateKeyPair(crypto.RSA, 2048)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
id, err := peer.IDFromPublicKey(pk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
kbytes, err := crypto.MarshalPrivateKey(sk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ident = config.Identity{
|
|
PeerID: id.Pretty(),
|
|
PrivKey: base64.StdEncoding.EncodeToString(kbytes),
|
|
}
|
|
} else {
|
|
ident = config.Identity{
|
|
PeerID: testPeerID,
|
|
}
|
|
}
|
|
|
|
c := config.Config{}
|
|
c.Addresses.Swarm = []string{fmt.Sprintf("/ip4/18.0.%d.1/tcp/4001", i)}
|
|
c.Identity = ident
|
|
c.Experimental.FilestoreEnabled = true
|
|
|
|
ds := syncds.MutexWrap(datastore.NewMapDatastore())
|
|
r := &repo.Mock{
|
|
C: c,
|
|
D: ds,
|
|
K: keystore.NewMemKeystore(),
|
|
F: filestore.NewFileManager(ds, filepath.Dir(os.TempDir())),
|
|
}
|
|
|
|
node, err := core.NewNode(ctx, &core.BuildCfg{
|
|
Routing: libp2p.DHTServerOption,
|
|
Repo: r,
|
|
Host: mock.MockHostOption(mn),
|
|
Online: fullIdentity,
|
|
ExtraOpts: map[string]bool{
|
|
"pubsub": true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
nodes[i] = node
|
|
apis[i], err = coreapi.NewCoreAPI(node)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
err := mn.LinkAll()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
bsinf := bootstrap.BootstrapConfigWithPeers(
|
|
[]peer.AddrInfo{
|
|
nodes[0].Peerstore.PeerInfo(nodes[0].Identity),
|
|
},
|
|
)
|
|
|
|
for _, n := range nodes[1:] {
|
|
if err := n.Bootstrap(bsinf); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return apis, nil
|
|
}
|
|
|
|
func TestIface(t *testing.T) {
|
|
tests.TestApi(&NodeProvider{})(t)
|
|
}
|