lotus/itests/kit/net.go

88 lines
1.8 KiB
Go
Raw Normal View History

2021-05-18 21:01:10 +00:00
package kit
import (
"context"
"testing"
"time"
2020-10-26 11:01:33 +00:00
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/go-address"
)
func StartOneNodeOneMiner(ctx context.Context, t *testing.T, blocktime time.Duration) (TestFullNode, address.Address) {
2021-05-17 12:28:09 +00:00
n, sn := RPCMockSbBuilder(t, OneFull, 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
2021-05-19 09:51:32 +00:00
bm := NewBlockMiner(t, miner)
bm.MineBlocks(ctx, blocktime)
2020-11-12 15:23:46 +00:00
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
}
2020-10-26 11:01:33 +00:00
func StartTwoNodesOneMiner(ctx context.Context, t *testing.T, blocktime time.Duration) ([]TestFullNode, []address.Address) {
2021-05-17 12:28:09 +00:00
n, sn := RPCMockSbBuilder(t, TwoFull, OneMiner)
2020-10-26 11:01:33 +00:00
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
2021-05-19 09:51:32 +00:00
bm := NewBlockMiner(t, miner)
bm.MineBlocks(ctx, blocktime)
2020-11-12 15:23:46 +00:00
t.Cleanup(bm.Stop)
2020-10-26 11:01:33 +00:00
// Send some funds to register the second node
fullNodeAddr2, err := fullNode2.WalletNew(ctx, types.KTSecp256k1)
if err != nil {
t.Fatal(err)
}
SendFunds(ctx, t, fullNode1, fullNodeAddr2, abi.NewTokenAmount(1e18))
2020-10-26 11:01:33 +00:00
// 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}
}