Fix unit tests

0xPolygon · Sep 8, 2023 · caea32b · caea32b
1 parent 3160d04
commit caea32b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/blockchain/blockchain.go b/blockchain/blockchain.go
@@ -12,7 +12,6 @@ import (
 
 	"github.com/0xPolygon/polygon-edge/blockchain/storage"
 	"github.com/0xPolygon/polygon-edge/chain"
-	"github.com/0xPolygon/polygon-edge/forkmanager"
 	"github.com/0xPolygon/polygon-edge/helper/common"
 	"github.com/0xPolygon/polygon-edge/state"
 	"github.com/0xPolygon/polygon-edge/types"
@@ -1393,7 +1392,7 @@ func (b *Blockchain) CalculateBaseFee(parent *types.Header) uint64 {
 
 func (b *Blockchain) calcBaseFeeDelta(gasUsedDelta, parentGasTarget, baseFee uint64) uint64 {
 	baseFeeChangeDenom := chain.BaseFeeChangeDenom
-	if forkmanager.GetInstance().IsForkEnabled(chain.Governance, b.Header().Number) {
+	if b.config.Params.Forks.IsActive(chain.Governance, b.Header().Number) {
 		baseFeeChangeDenom = b.Config().BaseFeeChangeDenom
 	}
 

diff --git a/blockchain/blockchain_test.go b/blockchain/blockchain_test.go
@@ -1403,6 +1403,13 @@ func TestBlockchain_CalculateBaseFee(t *testing.T) {
 				},
 			}
 
+			blockchain.setCurrentHeader(&types.Header{
+				Number:   test.blockNumber + 1,
+				GasLimit: test.parentGasLimit,
+				GasUsed:  test.parentGasUsed,
+				BaseFee:  test.parentBaseFee,
+			}, big.NewInt(1))
+
 			parent := &types.Header{
 				Number:   test.blockNumber,
 				GasLimit: test.parentGasLimit,

diff --git a/consensus/polybft/consensus_runtime_test.go b/consensus/polybft/consensus_runtime_test.go
@@ -490,6 +490,7 @@ func Test_NewConsensusRuntime(t *testing.T) {
 	config := &runtimeConfig{
 		polybftBackend: polybftBackendMock,
 		State:          newTestState(t),
+		genesisParams:  &chain.Params{Engine: map[string]interface{}{polyCommon.ConsensusName: polyBftConfig}},
 		GenesisConfig:  polyBftConfig,
 		DataDir:        tmpDir,
 		Key:            createTestKey(t),