132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"regexp"
|
|
"strconv"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
logging "github.com/ipfs/go-log/v2"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/api/test"
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/actors/policy"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/lib/lotuslog"
|
|
"github.com/filecoin-project/lotus/node/repo"
|
|
builder "github.com/filecoin-project/lotus/node/test"
|
|
)
|
|
|
|
func TestWorkerKeyChange(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode")
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
_ = logging.SetLogLevel("*", "INFO")
|
|
|
|
policy.SetConsensusMinerMinPower(abi.NewStoragePower(2048))
|
|
policy.SetSupportedProofTypes(abi.RegisteredSealProof_StackedDrg2KiBV1)
|
|
policy.SetMinVerifiedDealSize(abi.NewStoragePower(256))
|
|
|
|
lotuslog.SetupLogLevels()
|
|
logging.SetLogLevel("miner", "ERROR")
|
|
logging.SetLogLevel("chainstore", "ERROR")
|
|
logging.SetLogLevel("chain", "ERROR")
|
|
logging.SetLogLevel("sub", "ERROR")
|
|
logging.SetLogLevel("storageminer", "ERROR")
|
|
|
|
blocktime := 1 * time.Millisecond
|
|
|
|
n, sn := builder.MockSbBuilder(t, []test.FullNodeOpts{test.FullNodeWithUpgradeAt(1)}, test.OneMiner)
|
|
|
|
output := bytes.NewBuffer(nil)
|
|
run := func(cmd *cli.Command, args ...string) error {
|
|
app := cli.NewApp()
|
|
app.Metadata = map[string]interface{}{
|
|
"repoType": repo.StorageMiner,
|
|
"testnode-full": n[0],
|
|
"testnode-storage": sn[0],
|
|
}
|
|
app.Writer = output
|
|
build.RunningNodeType = build.NodeMiner
|
|
|
|
fs := flag.NewFlagSet("", flag.ContinueOnError)
|
|
for _, f := range cmd.Flags {
|
|
if err := f.Apply(fs); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
require.NoError(t, fs.Parse(args))
|
|
|
|
cctx := cli.NewContext(app, fs, nil)
|
|
return cmd.Action(cctx)
|
|
}
|
|
|
|
// setup miner
|
|
mine := int64(1)
|
|
done := make(chan struct{})
|
|
go func() {
|
|
defer close(done)
|
|
for atomic.LoadInt64(&mine) == 1 {
|
|
time.Sleep(blocktime)
|
|
if err := sn[0].MineOne(ctx, test.MineNext); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
}()
|
|
defer func() {
|
|
atomic.AddInt64(&mine, -1)
|
|
fmt.Println("shutting down mining")
|
|
<-done
|
|
}()
|
|
|
|
client := n[0]
|
|
|
|
newKey, err := client.WalletNew(ctx, types.KTBLS)
|
|
require.NoError(t, err)
|
|
|
|
// Initialize wallet.
|
|
test.SendFunds(ctx, t, client, newKey, abi.NewTokenAmount(0))
|
|
|
|
require.NoError(t, run(actorProposeChangeWorker, "--really-do-it", newKey.String()))
|
|
|
|
result := output.String()
|
|
output.Reset()
|
|
|
|
require.Contains(t, result, fmt.Sprintf("Worker key change to %s successfully proposed.", newKey))
|
|
|
|
epochRe := regexp.MustCompile("at or after height (?P<epoch>[0-9]+) to complete")
|
|
matches := epochRe.FindStringSubmatch(result)
|
|
require.NotNil(t, matches)
|
|
targetEpoch, err := strconv.Atoi(matches[1])
|
|
require.NoError(t, err)
|
|
require.NotZero(t, targetEpoch)
|
|
|
|
// Too early.
|
|
require.Error(t, run(actorConfirmChangeWorker, "--really-do-it", newKey.String()))
|
|
output.Reset()
|
|
|
|
for {
|
|
head, err := client.ChainHead(ctx)
|
|
require.NoError(t, err)
|
|
if head.Height() >= abi.ChainEpoch(targetEpoch) {
|
|
break
|
|
}
|
|
build.Clock.Sleep(10 * blocktime)
|
|
}
|
|
require.NoError(t, run(actorConfirmChangeWorker, "--really-do-it", newKey.String()))
|
|
output.Reset()
|
|
}
|