128 lines
2.5 KiB
Go
128 lines
2.5 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/miner"
|
|
)
|
|
|
|
type TestNode struct {
|
|
api.FullNode
|
|
}
|
|
|
|
type TestStorageNode struct {
|
|
api.StorageMiner
|
|
|
|
MineOne func(context.Context, miner.MineReq) error
|
|
}
|
|
|
|
var PresealGenesis = -1
|
|
|
|
const GenesisPreseals = 2
|
|
|
|
type StorageMiner struct {
|
|
Full int
|
|
Preseal int
|
|
}
|
|
|
|
// APIBuilder is a function which is invoked in test suite to provide
|
|
// test nodes and networks
|
|
//
|
|
// storage array defines storage nodes, numbers in the array specify full node
|
|
// index the storage node 'belongs' to
|
|
type APIBuilder func(t *testing.T, nFull int, storage []StorageMiner) ([]TestNode, []TestStorageNode)
|
|
type testSuite struct {
|
|
makeNodes APIBuilder
|
|
}
|
|
|
|
// TestApis is the entry point to API test suite
|
|
func TestApis(t *testing.T, b APIBuilder) {
|
|
ts := testSuite{
|
|
makeNodes: b,
|
|
}
|
|
|
|
t.Run("version", ts.testVersion)
|
|
t.Run("id", ts.testID)
|
|
t.Run("testConnectTwo", ts.testConnectTwo)
|
|
t.Run("testMining", ts.testMining)
|
|
t.Run("testMiningReal", ts.testMiningReal)
|
|
}
|
|
|
|
var oneMiner = []StorageMiner{{Full: 0, Preseal: PresealGenesis}}
|
|
|
|
func (ts *testSuite) testVersion(t *testing.T) {
|
|
ctx := context.Background()
|
|
apis, _ := ts.makeNodes(t, 1, oneMiner)
|
|
api := apis[0]
|
|
|
|
v, err := api.Version(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
require.Equal(t, v.Version, build.BuildVersion)
|
|
}
|
|
|
|
func (ts *testSuite) testID(t *testing.T) {
|
|
ctx := context.Background()
|
|
apis, _ := ts.makeNodes(t, 1, oneMiner)
|
|
api := apis[0]
|
|
|
|
id, err := api.ID(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Regexp(t, "^12", id.Pretty())
|
|
}
|
|
|
|
func (ts *testSuite) testConnectTwo(t *testing.T) {
|
|
ctx := context.Background()
|
|
apis, _ := ts.makeNodes(t, 2, oneMiner)
|
|
|
|
p, err := apis[0].NetPeers(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(p) != 0 {
|
|
t.Error("Node 0 has a peer")
|
|
}
|
|
|
|
p, err = apis[1].NetPeers(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(p) != 0 {
|
|
t.Error("Node 1 has a peer")
|
|
}
|
|
|
|
addrs, err := apis[1].NetAddrsListen(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := apis[0].NetConnect(ctx, addrs); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
p, err = apis[0].NetPeers(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(p) != 1 {
|
|
t.Error("Node 0 doesn't have 1 peer")
|
|
}
|
|
|
|
p, err = apis[1].NetPeers(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(p) != 1 {
|
|
t.Error("Node 0 doesn't have 1 peer")
|
|
}
|
|
}
|