lotus/itests/ccupgrade_test.go
2021-06-15 08:21:02 +02:00

134 lines
3.0 KiB
Go

package itests
import (
"context"
"fmt"
"testing"
"time"
"github.com/filecoin-project/lotus/itests/kit"
"github.com/filecoin-project/lotus/itests/kit2"
"github.com/stretchr/testify/require"
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/lotus/chain/types"
)
func TestCCUpgrade(t *testing.T) {
kit2.QuietMiningLogs()
for _, height := range []abi.ChainEpoch{
-1, // before
//162, // while sealing
//530, // after upgrade deal
//5000, // after
} {
height := height // make linters happy by copying
t.Run(fmt.Sprintf("upgrade-%d", height), func(t *testing.T) {
runTestCCUpgrade(t, height)
})
}
}
func runTestCCUpgrade(t *testing.T, upgradeHeight abi.ChainEpoch) {
ctx := context.Background()
blockTime := 5 * time.Millisecond
client, miner, ens := kit2.EnsembleMinimal(t, kit2.MockProofs(), kit2.LatestActorsAt(upgradeHeight))
ens.InterconnectAll().BeginMining(blockTime)
//b := kit.MockMinerBuilder
//n, sn := b(t, []kit.FullNodeOpts{kit.FullNodeWithLatestActorsAt(upgradeHeight)}, kit.OneMiner)
//client := n[0].FullNode.(*impl.FullNodeAPI)
//miner := sn[0]
//addrinfo, err := client.NetAddrsListen(ctx)
//if err != nil {
// t.Fatal(err)
//}
//
//if err := miner.NetConnect(ctx, addrinfo); err != nil {
// t.Fatal(err)
//}
//time.Sleep(time.Second)
//
//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, kit.MineNext); err != nil {
// t.Error(err)
// }
// }
//}()
maddr, err := miner.ActorAddress(ctx)
if err != nil {
t.Fatal(err)
}
CC := abi.SectorNumber(kit.GenesisPreseals + 1)
Upgraded := CC + 1
miner.PledgeSectors(ctx, 1, 0, nil)
sl, err := miner.SectorsList(ctx)
if err != nil {
t.Fatal(err)
}
if len(sl) != 1 {
t.Fatal("expected 1 sector")
}
if sl[0] != CC {
t.Fatal("bad")
}
{
si, err := client.StateSectorGetInfo(ctx, maddr, CC, types.EmptyTSK)
require.NoError(t, err)
require.Less(t, 50000, int(si.Expiration))
}
if err := miner.SectorMarkForUpgrade(ctx, sl[0]); err != nil {
t.Fatal(err)
}
dh := kit2.NewDealHarness(t, client, miner)
dh.MakeOnlineDeal(context.Background(), 6, false, 0)
// Validate upgrade
{
exp, err := client.StateSectorExpiration(ctx, maddr, CC, types.EmptyTSK)
require.NoError(t, err)
require.NotNil(t, exp)
require.Greater(t, 50000, int(exp.OnTime))
}
{
exp, err := client.StateSectorExpiration(ctx, maddr, Upgraded, types.EmptyTSK)
require.NoError(t, err)
require.Less(t, 50000, int(exp.OnTime))
}
dlInfo, err := client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
require.NoError(t, err)
// Sector should expire.
for {
// Wait for the sector to expire.
status, err := miner.SectorsStatus(ctx, CC, true)
require.NoError(t, err)
if status.OnTime == 0 && status.Early == 0 {
break
}
t.Log("waiting for sector to expire")
// wait one deadline per loop.
time.Sleep(time.Duration(dlInfo.WPoStChallengeWindow) * blockTime)
}
}