85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
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)
|
|
fmt.Printf("CCUpgrade: %d\n", CCUpgrade)
|
|
|
|
miner.PledgeSectors(ctx, 1, 0, nil)
|
|
sl, err := miner.SectorsListNonGenesis(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.SectorsListNonGenesis(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
|
|
}
|
|
}
|