90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/lotus/api/test"
|
|
test2 "github.com/filecoin-project/lotus/node/test"
|
|
)
|
|
|
|
func StartOneNodeOneMiner(ctx context.Context, t *testing.T, blocktime time.Duration) (test.TestNode, address.Address) {
|
|
n, sn := test2.RPCMockSbBuilder(t, test.OneFull, test.OneMiner)
|
|
|
|
full := n[0]
|
|
miner := sn[0]
|
|
|
|
// Get everyone connected
|
|
addrs, err := full.NetAddrsListen(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := miner.NetConnect(ctx, addrs); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Start mining blocks
|
|
bm := test.NewBlockMiner(ctx, t, miner, blocktime)
|
|
bm.MineBlocks()
|
|
t.Cleanup(bm.Stop)
|
|
|
|
// Get the full node's wallet address
|
|
fullAddr, err := full.WalletDefaultAddress(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Create mock CLI
|
|
return full, fullAddr
|
|
}
|
|
|
|
func StartTwoNodesOneMiner(ctx context.Context, t *testing.T, blocktime time.Duration) ([]test.TestNode, []address.Address) {
|
|
n, sn := test2.RPCMockSbBuilder(t, test.TwoFull, test.OneMiner)
|
|
|
|
fullNode1 := n[0]
|
|
fullNode2 := n[1]
|
|
miner := sn[0]
|
|
|
|
// Get everyone connected
|
|
addrs, err := fullNode1.NetAddrsListen(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := fullNode2.NetConnect(ctx, addrs); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := miner.NetConnect(ctx, addrs); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Start mining blocks
|
|
bm := test.NewBlockMiner(ctx, t, miner, blocktime)
|
|
bm.MineBlocks()
|
|
t.Cleanup(bm.Stop)
|
|
|
|
// Send some funds to register the second node
|
|
fullNodeAddr2, err := fullNode2.WalletNew(ctx, types.KTSecp256k1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
test.SendFunds(ctx, t, fullNode1, fullNodeAddr2, abi.NewTokenAmount(1e18))
|
|
|
|
// Get the first node's address
|
|
fullNodeAddr1, err := fullNode1.WalletDefaultAddress(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Create mock CLI
|
|
return n, []address.Address{fullNodeAddr1, fullNodeAddr2}
|
|
}
|