Fix UTs

0xPolygon · Sep 5, 2023 · 9a75e4c · 9a75e4c
1 parent a0e79f2
commit 9a75e4c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/consensus/polybft/fsm_test.go b/consensus/polybft/fsm_test.go
@@ -586,6 +586,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -603,6 +604,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -621,6 +623,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			validators:             validatorSet,
 			distributeRewardsInput: createTestDistributeRewardsInput(t, 0, validators.GetPublicIdentities(), 10),
 			logger:                 hclog.NewNullLogger(),
+			forks:                  &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		distributeRewardsTx, err := fsm.createDistributeRewardsTx()
@@ -638,6 +641,7 @@ func TestFSM_VerifyStateTransactions_DistributeRewards(t *testing.T) {
 			parent:              &types.Header{Number: 9},
 			validators:          validatorSet,
 			logger:              hclog.NewNullLogger(),
+			forks:               &chain.Forks{chain.Governance: chain.NewFork(0)},
 		}
 
 		err := fsm.VerifyStateTransactions([]*types.Transaction{})