Skip to content

Commit

Permalink
Rough draft
Browse files Browse the repository at this point in the history
  • Loading branch information
ZenGround0 committed Jun 24, 2024
1 parent 286fada commit f57dae9
Showing 1 changed file with 69 additions and 0 deletions.
69 changes: 69 additions & 0 deletions itests/migration_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
"testing"
"time"

"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"

"github.com/filecoin-project/go-address"
Expand Down Expand Up @@ -828,3 +829,71 @@ func TestMigrationNV21(t *testing.T) {
//todo @zen Direct data onboarding tests

}

func TestMigrationNV23(t *testing.T) {
kit.QuietMiningLogs()

nv23epoch := abi.ChainEpoch(100)
testClient, _, ens := kit.EnsembleMinimal(t, kit.MockProofs(),
kit.UpgradeSchedule(stmgr.Upgrade{
Network: network.Version22,
Height: -1,
}, stmgr.Upgrade{
Network: network.Version23,
Height: nv23epoch,
Migration: filcns.UpgradeActorsV13,
},
))

ens.InterconnectAll().BeginMining(10 * time.Millisecond)

clientApi := testClient.FullNode.(*impl.FullNodeAPI)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

testClient.WaitTillChain(ctx, kit.HeightAtLeast(nv23epoch+5))

bs := blockstore.NewAPIBlockstore(testClient)
ctxStore := gstStore.WrapBlockStore(ctx, bs)

preMigrationTs, err := clientApi.ChainGetTipSetByHeight(ctx, nv23epoch, types.EmptyTSK)
require.NoError(t, err)

preStateTree, err := state.LoadStateTree(ctxStore, preMigrationTs.Blocks()[0].ParentStateRoot)
require.NoError(t, err)
require.Equal(t, types.StateTreeVersion5, preStateTree.Version())

// Check f090 actor before migration
f090Addr, err := address.NewIDAddress(90)
require.NoError(t, err)
msigCodeNv22, ok := actors.GetActorCodeID(actorstypes.Version13, manifest.MultisigKey)
assert.True(t, ok)
f090ActorPre, err := preStateTree.GetActor(f090Addr)
require.NoError(t, err)
require.True(t, f090ActorPre.Code.Equals(msigCodeNv22))

// Perform the migration
testClient.WaitTillChain(ctx, kit.HeightAtLeast(nv23epoch+1))
postMigrationTs, err := clientApi.ChainHead(ctx)
require.NoError(t, err)
postStateTree, err := state.LoadStateTree(ctxStore, postMigrationTs.Blocks()[0].ParentStateRoot)
require.NoError(t, err)

// Check the new system actor
systemAct, err := postStateTree.GetActor(builtin.SystemActorAddr)
require.NoError(t, err)
systemCode, ok := actors.GetActorCodeID(actorstypes.Version14, manifest.SystemKey)
require.True(t, ok)
require.Equal(t, systemCode, systemAct.Code)

// Check f090 actor after migration
f090ActorPost, err := postStateTree.GetActor(f090Addr)
require.NoError(t, err)
accountNV23, ok := actors.GetActorCodeID(actorstypes.Version14, manifest.AccountKey)
assert.True(t, ok)
require.True(t, f090ActorPost.Code.Equals(accountNV23))
f090StatePost, err := clientApi.StateReadState(ctx, f090Addr, types.EmptyTSK)
require.NoError(t, err)
state := f090StatePost.State.(map[string]interface{})
require.Equal(t, state["Address"], f090Addr)
}

0 comments on commit f57dae9

Please sign in to comment.