freezer: fix some broken uts;

bnb-chain · Jul 16, 2024 · fb16c4a · fb16c4a
1 parent 92aa77a
commit fb16c4a
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/core/blockchain_repair_test.go b/core/blockchain_repair_test.go
@@ -26,6 +26,8 @@ import (
  "testing"
  "time"
 
+ "github.com/ethereum/go-ethereum/ethdb"
+
  "github.com/ethereum/go-ethereum/common"
  "github.com/ethereum/go-ethereum/consensus/ethash"
  "github.com/ethereum/go-ethereum/core/rawdb"
@@ -1795,6 +1797,13 @@ func testRepairWithScheme(t *testing.T, tt *rewindTest, snapshots bool, scheme s
  config.SnapshotWait = true
  }
  config.TriesInMemory = 128
+
+ if err = db.SetupFreezerEnv(&ethdb.FreezerEnv{
+ ChainCfg: gspec.Config,
+ BlobExtraReserve: params.DefaultExtraReserveForBlobRequests,
+ }); err != nil {
+ t.Fatalf("Failed to create chain: %v", err)
+ }
  chain, err := NewBlockChain(db, config, gspec, nil, engine, vm.Config{}, nil, nil)
  if err != nil {
  t.Fatalf("Failed to create chain: %v", err)

diff --git a/core/blockchain_sethead_test.go b/core/blockchain_sethead_test.go
@@ -27,6 +27,8 @@ import (
  "testing"
  "time"
 
+ "github.com/ethereum/go-ethereum/ethdb"
+
  "github.com/ethereum/go-ethereum/common"
  "github.com/ethereum/go-ethereum/consensus/ethash"
  "github.com/ethereum/go-ethereum/core/rawdb"
@@ -1998,6 +2000,13 @@ func testSetHeadWithScheme(t *testing.T, tt *rewindTest, snapshots bool, scheme
  config.SnapshotWait = true
  }
  config.TriesInMemory = 128
+
+ if err = db.SetupFreezerEnv(&ethdb.FreezerEnv{
+ ChainCfg: gspec.Config,
+ BlobExtraReserve: params.DefaultExtraReserveForBlobRequests,
+ }); err != nil {
+ t.Fatalf("Failed to create chain: %v", err)
+ }
  chain, err := NewBlockChain(db, config, gspec, nil, engine, vm.Config{}, nil, nil)
  if err != nil {
  t.Fatalf("Failed to create chain: %v", err)