update test

iotexproject · Jun 25, 2024 · cfe1123 · cfe1123
1 parent 182388a
commit cfe1123
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/action/protocol/staking/vote_reviser_test.go b/action/protocol/staking/vote_reviser_test.go
@@ -227,7 +227,6 @@ func TestVoteReviser(t *testing.T) {
 	r.Nil(csm.GetByName(oldCand.Name))
 }
 
-// TODO: merge TestVoteRevise_CorrectEndorsement and TestVoteRevise_CorrectSelfStake
 func TestVoteRevise_CorrectEndorsement(t *testing.T) {
 	r := require.New(t)
 	t.Run("correct endorse height", func(t *testing.T) {
@@ -282,10 +281,6 @@ func TestVoteRevise_CorrectEndorsement(t *testing.T) {
 			r.ErrorIs(err, state.ErrStateNotExist)
 		})
 	})
-}
-
-func TestVoteRevise_CorrectSelfStake(t *testing.T) {
-	r := require.New(t)
 	t.Run("correct selfstake height", func(t *testing.T) {
 		revise := NewVoteReviser(ReviseConfig{
 			SelfStakeBucketReviseHeights: []uint64{1, 3},
@@ -324,6 +319,7 @@ func TestVoteRevise_CorrectSelfStake(t *testing.T) {
 			revise := NewVoteReviser(ReviseConfig{
 				VoteWeight:                   g.VoteWeightCalConsts,
 				SelfStakeBucketReviseHeights: []uint64{2},
+				ReviseHeights:                []uint64{2},
 			})
 			bkt := &VoteBucket{
 				Index:          0,
@@ -359,6 +355,7 @@ func TestVoteRevise_CorrectSelfStake(t *testing.T) {
 			revise := NewVoteReviser(ReviseConfig{
 				VoteWeight:                   g.VoteWeightCalConsts,
 				SelfStakeBucketReviseHeights: []uint64{2},
+				ReviseHeights:                []uint64{2},
 			})
 			bkt := &VoteBucket{
 				Index:          1,
@@ -398,6 +395,7 @@ func TestVoteRevise_CorrectSelfStake(t *testing.T) {
 			revise := NewVoteReviser(ReviseConfig{
 				VoteWeight:                   g.VoteWeightCalConsts,
 				SelfStakeBucketReviseHeights: []uint64{2},
+				ReviseHeights:                []uint64{2},
 			})
 			bkt := &VoteBucket{
 				Candidate:      identityset.Address(3),
@@ -434,6 +432,7 @@ func TestVoteRevise_CorrectSelfStake(t *testing.T) {
 			revise := NewVoteReviser(ReviseConfig{
 				VoteWeight:                   g.VoteWeightCalConsts,
 				SelfStakeBucketReviseHeights: []uint64{2},
+				ReviseHeights:                []uint64{2},
 			})
 			bkt := &VoteBucket{
 				Candidate:      identityset.Address(4),
@@ -470,6 +469,7 @@ func TestVoteRevise_CorrectSelfStake(t *testing.T) {
 			revise := NewVoteReviser(ReviseConfig{
 				VoteWeight:                   g.VoteWeightCalConsts,
 				SelfStakeBucketReviseHeights: []uint64{2},
+				ReviseHeights:                []uint64{2},
 			})
 			bkt := &VoteBucket{
 				Candidate:      identityset.Address(5),