mirror of
https://github.com/ipfs/kubo.git
synced 2025-08-06 19:44:01 +08:00
122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/ipfs/go-ipfs/filestore"
|
|
|
|
"github.com/ipfs/go-ipfs/core"
|
|
"github.com/ipfs/go-ipfs/core/coreapi"
|
|
mock "github.com/ipfs/go-ipfs/core/mock"
|
|
"github.com/ipfs/go-ipfs/keystore"
|
|
"github.com/ipfs/go-ipfs/repo"
|
|
|
|
"github.com/ipfs/go-datastore"
|
|
syncds "github.com/ipfs/go-datastore/sync"
|
|
"github.com/ipfs/go-ipfs-config"
|
|
coreiface "github.com/ipfs/interface-go-ipfs-core"
|
|
"github.com/ipfs/interface-go-ipfs-core/tests"
|
|
ci "github.com/libp2p/go-libp2p-crypto"
|
|
"github.com/libp2p/go-libp2p-peer"
|
|
pstore "github.com/libp2p/go-libp2p-peerstore"
|
|
"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(ctx)
|
|
|
|
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 := ci.GenerateKeyPair(ci.RSA, 512)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
id, err := peer.IDFromPublicKey(pk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
kbytes, err := sk.Bytes()
|
|
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/127.0.%d.1/tcp/4001", i)}
|
|
c.Identity = ident
|
|
c.Experimental.FilestoreEnabled = true
|
|
|
|
ds := datastore.NewMapDatastore()
|
|
r := &repo.Mock{
|
|
C: c,
|
|
D: syncds.MutexWrap(ds),
|
|
K: keystore.NewMemKeystore(),
|
|
F: filestore.NewFileManager(ds, filepath.Dir(os.TempDir())),
|
|
}
|
|
|
|
node, err := core.NewNode(ctx, &core.BuildCfg{
|
|
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 := core.BootstrapConfigWithPeers(
|
|
[]pstore.PeerInfo{
|
|
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)
|
|
}
|