128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/node/impl"
|
|
)
|
|
|
|
func TestCCUpgrade(t *testing.T, b APIBuilder, blocktime time.Duration) {
|
|
for _, height := range []abi.ChainEpoch{
|
|
1, // before
|
|
162, // while sealing
|
|
520, // after upgrade deal
|
|
5000, // after
|
|
} {
|
|
height := height // make linters happy by copying
|
|
t.Run(fmt.Sprintf("upgrade-%d", height), func(t *testing.T) {
|
|
testCCUpgrade(t, b, blocktime, height)
|
|
})
|
|
}
|
|
}
|
|
|
|
func testCCUpgrade(t *testing.T, b APIBuilder, blocktime time.Duration, upgradeHeight abi.ChainEpoch) {
|
|
ctx := context.Background()
|
|
n, sn := b(t, []FullNodeOpts{FullNodeWithActorsV2At(upgradeHeight)}, 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, MineNext); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
maddr, err := miner.ActorAddress(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
CC := abi.SectorNumber(GenesisPreseals + 1)
|
|
Upgraded := CC + 1
|
|
|
|
pledgeSectors(t, ctx, miner, 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)
|
|
}
|
|
|
|
MakeDeal(t, ctx, 6, client, miner, false, 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)
|
|
}
|
|
|
|
fmt.Println("shutting down mining")
|
|
atomic.AddInt64(&mine, -1)
|
|
<-done
|
|
}
|