fix: fix error in test cases

Author: s-h-ubham

consensus/polybft/consensus_runtime_test.go

@@ -8,6 +8,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/0xPolygon/polygon-edge/chain"
 	"github.com/0xPolygon/polygon-edge/consensus"
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/bitmap"
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/contractsapi"
@@ -434,6 +435,11 @@ func TestConsensusRuntime_FSM_NotEndOfEpoch_NotEndOfSprint(t *testing.T) {
 		},
 		Key:        wallet.NewKey(validators.GetPrivateIdentities()[0]),
 		blockchain: blockchainMock,
+		consensusConfig: &consensus.Config{
+			Params: &chain.Params{
+				Forks: &chain.Forks{},
+			},
+		},
 	}
 	runtime := &consensusRuntime{
 		proposerCalculator: NewProposerCalculatorFromSnapshot(
@@ -452,6 +458,10 @@ func TestConsensusRuntime_FSM_NotEndOfEpoch_NotEndOfSprint(t *testing.T) {
 		state:             newTestState(t),
 		stateSyncManager:  &dummyStateSyncManager{},
 		checkpointManager: &dummyCheckpointManager{},
+		stakeManager:      &dummyStakeManager{},
+		rewardWalletCalculator: &rewardWalletCalculator{
+			blockchain: blockchainMock,
+		},
 	}
 	runtime.setIsActiveValidator(true)
 
@@ -518,6 +528,11 @@ func TestConsensusRuntime_FSM_EndOfEpoch_BuildCommitEpoch(t *testing.T) {
 		},
 		Key:        validatorAccounts.GetValidator("A").Key(),
 		blockchain: blockchainMock,
+		consensusConfig: &consensus.Config{
+			Params: &chain.Params{
+				Forks: &chain.Forks{},
+			},
+		},
 	}
 
 	snapshot := NewProposerSnapshot(1, nil)

consensus/polybft/fsm_test.go

@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/0xPolygon/polygon-edge/bls"
+	"github.com/0xPolygon/polygon-edge/chain"
 	"github.com/0xPolygon/polygon-edge/consensus"
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/bitmap"
 	"github.com/0xPolygon/polygon-edge/consensus/polybft/contractsapi"
@@ -590,6 +591,7 @@ func TestFSM_VerifyStateTransactions_MiddleOfEpochWithoutTransaction(t *testing.
 	fsm := &fsm{
 		commitEpochInput: createTestCommitEpochInput(t, 0, allAccounts, 10),
 		parent:           &types.Header{},
+		forks:            &chain.Forks{},
 	}
 	err := fsm.VerifyStateTransactions([]*types.Transaction{})
 	assert.NoError(t, err)
@@ -604,6 +606,8 @@ func TestFSM_VerifyStateTransactions_EndOfEpochWithoutTransaction(t *testing.T)
 	fsm := &fsm{
 		isEndOfEpoch:     true,
 		commitEpochInput: createTestCommitEpochInput(t, 0, allAccounts, 10),
+		forks:            &chain.Forks{},
+		parent:           &types.Header{},
 	}
 	assert.EqualError(t, fsm.VerifyStateTransactions([]*types.Transaction{}),
 		"commit epoch transaction is not found in the epoch ending block")
@@ -857,6 +861,7 @@ func TestFSM_VerifyStateTransactions_EndOfEpochMissingFundRewardWalletTx(t *test
 		commitEpochInput:       createTestCommitEpochInput(t, 0, allAccounts, 10),
 		rewardWalletFundAmount: common.GetTwoThirdOfMaxUint256(),
 		logger:                 hclog.NewNullLogger(),
+		forks:                  &chain.Forks{},
 	}
 
 	// create commit epoch transaction in order to add it to the transactions list
@@ -899,6 +904,7 @@ func TestFSM_VerifyStateTransactions_EndOfEpochWithoutFundRewardWalletTx(t *test
 		),
 		distributeDAOIncentiveInput: createTestDistributeDAOIncentiveInput(t),
 		logger:                      hclog.NewNullLogger(),
+		forks:                       &chain.Forks{},
 	}
 
 	// create commit epoch transaction in order to add it to the transactions list
@@ -1008,6 +1014,7 @@ func TestFSM_VerifyStateTransactions_EndOfEpochMissingDistributeVaultFundsTx(t *
 		),
 		distributeDAOIncentiveInput: createTestDistributeDAOIncentiveInput(t),
 		logger:                      hclog.NewNullLogger(),
+		forks:                       &chain.Forks{},
 	}
 
 	// create commit epoch transaction in order to add it to the transactions list
@@ -1202,6 +1209,7 @@ func TestFSM_VerifyStateTransactions_StateTransactionPass(t *testing.T) {
 		),
 		distributeDAOIncentiveInput: createTestDistributeDAOIncentiveInput(t),
 		logger:                      hclog.NewNullLogger(),
+		forks:                       &chain.Forks{},
 	}
 
 	// create commit epoch transaction in order to add it to the transactions list
@@ -1251,6 +1259,7 @@ func TestFSM_VerifyStateTransaction_StartOfEpochInvalidSyncValidatorsDataTxIndex
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, allAccounts),
 		newValidatorsDelta:      newValidatorDelta,
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	// create sync validators data transaction to add it in the list of state transactions to verify
@@ -1303,6 +1312,7 @@ func TestFSM_VerifyStateTransaction_StartOfEpochInvalidSyncValidatorsDataTxErr(t
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, allAccounts),
 		newValidatorsDelta:      newValidatorDelta,
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	syncValidatorsDataInputEncoded, err := fsm.syncValidatorsDataInput.EncodeAbi()
@@ -1349,6 +1359,7 @@ func TestFSM_VerifyStateTransaction_StartOfEpochSingleSyncValidatorsDataTxRequir
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, allAccounts),
 		newValidatorsDelta:      newValidatorDelta,
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	// create sync validators data transaction to add it in the list of state transactions to verify
@@ -1383,6 +1394,7 @@ func TestFSM_VerifyStateTransaction_StartOfEpochUnexpectedSyncValidatorsDataTxEr
 		validators:              validatorSet,
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, allAccounts),
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	// create sync validators data transaction to add it in the list of state transactions to verify
@@ -1452,6 +1464,7 @@ func TestFSM_VerifyStateTransaction_StartOfEpochMissingSyncValidatorsDataTx(t *t
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, validators),
 		newValidatorsDelta:      newDelta,
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	assert.ErrorContains(
@@ -1483,6 +1496,7 @@ func TestFSM_VerifyStateTransaction_SyncValidatorsData(t *testing.T) {
 		syncValidatorsDataInput: createTestSyncValidatorsDataInput(t, allAccounts),
 		newValidatorsDelta:      newValidatorDelta,
 		logger:                  hclog.NewNullLogger(),
+		forks:                   &chain.Forks{},
 	}
 
 	// create sync validators data transaction to add it in the list of state transactions to verify
@@ -1791,6 +1805,7 @@ func TestFSM_Validate_ExitEventRootNotExpected(t *testing.T) {
 		exitEventRootHash: types.BytesToHash(
 			[]byte{0, 1, 2, 3, 4},
 		), // expect this to be in proposal extra
+		forks: &chain.Forks{},
 	}
 
 	err = fsm.Validate(proposal)
@@ -1928,6 +1943,7 @@ func TestFSM_Validate_EpochEndingBlock_MismatchInDeltas(t *testing.T) {
 		polybftBackend:              polybftBackendMock,
 		newValidatorsDelta:          newValidatorDelta,
 		config:                      &PolyBFTConfig{BlockTimeDrift: 1},
+		forks:                       &chain.Forks{},
 	}
 
 	err = fsm.Validate(proposal)
@@ -2027,6 +2043,7 @@ func TestFSM_Validate_EpochEndingBlock_UpdatingValidatorSetInNonEpochEndingBlock
 		logger:         hclog.NewNullLogger(),
 		polybftBackend: polybftBackendMock,
 		config:         &PolyBFTConfig{BlockTimeDrift: 1},
+		forks:          &chain.Forks{},
 	}
 
 	err = fsm.Validate(proposal)
@@ -2485,11 +2502,11 @@ func TestFSM_DecodeCommitEpochStateTx(t *testing.T) {
 // func TestFSM_VerifyStateTransaction_ValidBothTypesOfStateTransactions(t *testing.T) {
 // 	t.Parallel()
 
-// 	var (
-// 		commitments       [2]*PendingCommitment
-// 		stateSyncs        [2][]*contractsapi.StateSyncedEvent
-// 		signedCommitments [2]*CommitmentMessageSigned
-// 	)
+// var (
+// 	commitments       [2]*PendingCommitment
+// 	stateSyncs        [2][]*contractsapi.StateSyncedEvent
+// 	signedCommitments [2]*CommitmentMessageSigned
+// )
 
 // validators := validator.NewTestValidatorsWithAliases(t, []string{"A", "B", "C", "D", "E"})
 // commitments[0], signedCommitments[0], stateSyncs[0] = buildCommitmentAndStateSyncs(t, 10, uint64(3), 2)