106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
package itests
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCCUpgrade(t *testing.T) {
|
|
kit.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
|
|
|
|
opts := kit.ConstructorOpts(kit.LatestActorsAt(upgradeHeight))
|
|
client, miner, ens := kit.EnsembleMinimal(t, kit.MockProofs(), opts)
|
|
ens.InterconnectAll().BeginMining(blockTime)
|
|
|
|
maddr, err := miner.ActorAddress(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
CC := abi.SectorNumber(kit.DefaultPresealsPerBootstrapMiner + 1)
|
|
Upgraded := CC + 1
|
|
|
|
miner.PledgeSectors(ctx, 1, 0, nil)
|
|
|
|
sl, err := miner.SectorsList(ctx)
|
|
require.NoError(t, err)
|
|
require.Len(t, sl, 1, "expected 1 sector")
|
|
require.Equal(t, CC, sl[0], "unexpected sector number")
|
|
|
|
{
|
|
si, err := client.StateSectorGetInfo(ctx, maddr, CC, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Less(t, 50000, int(si.Expiration))
|
|
}
|
|
|
|
err = miner.SectorMarkForUpgrade(ctx, sl[0])
|
|
require.NoError(t, err)
|
|
|
|
dh := kit.NewDealHarness(t, client, miner)
|
|
deal, res, inPath := dh.MakeOnlineDeal(ctx, kit.MakeFullDealParams{
|
|
Rseed: 6,
|
|
SuspendUntilCryptoeconStable: true,
|
|
})
|
|
outPath := dh.PerformRetrieval(context.Background(), deal, res.Root, false)
|
|
kit.AssertFilesEqual(t, inPath, outPath)
|
|
|
|
// Validate upgrade
|
|
|
|
{
|
|
exp, err := client.StateSectorExpiration(ctx, maddr, CC, types.EmptyTSK)
|
|
if err != nil {
|
|
require.Contains(t, err.Error(), "failed to find sector 3") // already cleaned up
|
|
} else {
|
|
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)
|
|
}
|
|
}
|