package itests import ( "context" "fmt" "testing" "time" "github.com/stretchr/testify/require" "github.com/filecoin-project/go-state-types/abi" "github.com/filecoin-project/go-state-types/network" "github.com/filecoin-project/lotus/api" "github.com/filecoin-project/lotus/chain/types" "github.com/filecoin-project/lotus/itests/kit" sealing "github.com/filecoin-project/lotus/storage/pipeline" ) func TestAbortUpgradeAvailable(t *testing.T) { kit.QuietMiningLogs() ctx := context.Background() blockTime := 1 * time.Millisecond client, miner, ens := kit.EnsembleMinimal(t, kit.GenesisNetworkVersion(network.Version15), kit.ThroughRPC()) ens.InterconnectAll().BeginMiningMustPost(blockTime) maddr, err := miner.ActorAddress(ctx) if err != nil { t.Fatal(err) } CCUpgrade := abi.SectorNumber(kit.DefaultPresealsPerBootstrapMiner + 1) fmt.Printf("CCUpgrade: %d\n", CCUpgrade) 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, CCUpgrade, sl[0], "unexpected sector number") { si, err := client.StateSectorGetInfo(ctx, maddr, CCUpgrade, types.EmptyTSK) require.NoError(t, err) require.Less(t, 50000, int(si.Expiration)) } client.WaitForSectorActive(ctx, t, CCUpgrade, maddr) err = miner.SectorMarkForUpgrade(ctx, sl[0], true) require.NoError(t, err) sl, err = miner.SectorsList(ctx) require.NoError(t, err) require.Len(t, sl, 1, "expected 1 sector") ss, err := miner.SectorsStatus(ctx, sl[0], false) require.NoError(t, err) for i := 0; i < 100; i++ { ss, err = miner.SectorsStatus(ctx, sl[0], false) require.NoError(t, err) if ss.State == api.SectorState(sealing.Proving) { time.Sleep(50 * time.Millisecond) continue } require.Equal(t, api.SectorState(sealing.Available), ss.State) break } require.NoError(t, miner.SectorAbortUpgrade(ctx, sl[0])) for i := 0; i < 100; i++ { ss, err = miner.SectorsStatus(ctx, sl[0], false) require.NoError(t, err) if ss.State == api.SectorState(sealing.Available) { time.Sleep(50 * time.Millisecond) continue } require.Equal(t, api.SectorState(sealing.Proving), ss.State) break } }