add e2etest for migrate

iotexproject · Jun 12, 2024 · 0ca40b9 · 0ca40b9
1 parent 1de44c9
commit 0ca40b9
Show file tree

Hide file tree

Showing 3 changed files with 121 additions and 7 deletions.
diff --git a/e2etest/e2etest.go b/e2etest/e2etest.go
@@ -90,17 +90,18 @@ func (e *e2etest) runCase(ctx context.Context, c *testcase) {
 func (e *e2etest) teardown() {
 	require := require.New(e.t)
 	// clean up
+	require.NoError(e.svr.Stop(context.Background()))
 	testutil.CleanupPath(e.cfg.Chain.ChainDBPath)
 	testutil.CleanupPath(e.cfg.Chain.TrieDBPath)
 	testutil.CleanupPath(e.cfg.Chain.BloomfilterIndexDBPath)
 	testutil.CleanupPath(e.cfg.Chain.CandidateIndexDBPath)
 	testutil.CleanupPath(e.cfg.Chain.StakingIndexDBPath)
 	testutil.CleanupPath(e.cfg.Chain.ContractStakingIndexDBPath)
+	testutil.CleanupPath(e.cfg.Chain.ContractStakingIndexDBPathV2)
 	testutil.CleanupPath(e.cfg.DB.DbPath)
 	testutil.CleanupPath(e.cfg.Chain.IndexDBPath)
 	testutil.CleanupPath(e.cfg.System.SystemLogDBPath)
 	testutil.CleanupPath(e.cfg.Chain.SGDIndexDBPath)
-	require.NoError(e.svr.Stop(context.Background()))
 }
 
 func (e *e2etest) withTest(t *testing.T) *e2etest {

diff --git a/e2etest/expect.go b/e2etest/expect.go
@@ -31,6 +31,9 @@ type (
 	bucketExpect struct {
 		bucket *iotextypes.VoteBucket
 	}
+	executionExpect struct {
+		contractAddress string
+	}
 )
 
 func (be *basicActionExpect) expect(test *e2etest, act *action.SealedEnvelope, receipt *action.Receipt, err error) {
@@ -78,7 +81,7 @@ func (ce *candidateExpect) expect(test *e2etest, act *action.SealedEnvelope, rec
 func (be *bucketExpect) expect(test *e2etest, act *action.SealedEnvelope, receipt *action.Receipt, err error) {
 	require := require.New(test.t)
 	method := &iotexapi.ReadStakingDataMethod{
-		Method: iotexapi.ReadStakingDataMethod_BUCKETS_BY_INDEXES,
+		Method: iotexapi.ReadStakingDataMethod_COMPOSITE_BUCKETS_BY_INDEXES,
 	}
 	methodBytes, err := proto.Marshal(method)
 	require.NoError(err)
@@ -110,6 +113,12 @@ func (be *bucketExpect) expect(test *e2etest, act *action.SealedEnvelope, receip
 	idx := slices.IndexFunc(vbs.Buckets, func(vb *iotextypes.VoteBucket) bool {
 		return vb.ContractAddress == be.bucket.ContractAddress
 	})
-	require.True(idx != -1)
+	require.Greater(idx, -1)
 	require.EqualValues(be.bucket.String(), vbs.Buckets[idx].String())
 }
+
+func (ee *executionExpect) expect(test *e2etest, act *action.SealedEnvelope, receipt *action.Receipt, err error) {
+	require := require.New(test.t)
+	require.NoError(err)
+	require.Equal(ee.contractAddress, receipt.ContractAddress)
+}