Update for the new bad block manager (#607)

Signed-off-by: Fabio Di Fabio <[email protected]>
Consensys · Feb 21, 2024 · e6f14e9 · e6f14e9
1 parent f16aabc
commit e6f14e9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/reference-tests/src/test/java/net/consensys/linea/BlockchainReferenceTestTools.java b/reference-tests/src/test/java/net/consensys/linea/BlockchainReferenceTestTools.java
@@ -121,7 +121,7 @@ public static void executeTest(final BlockchainReferenceTestCaseSpec spec) {
         final ProtocolSpec protocolSpec = schedule.getByBlockHeader(block.getHeader());
 
         final MainnetBlockImporter blockImporter =
-            getMainnetBlockImporter(protocolSpec, schedule, zkTracer);
+            getMainnetBlockImporter(context, protocolSpec, schedule, zkTracer);
 
         final HeaderValidationMode validationMode =
             "NoProof".equalsIgnoreCase(spec.getSealEngine())
@@ -143,7 +143,10 @@ public static void executeTest(final BlockchainReferenceTestCaseSpec spec) {
   }
 
   private static MainnetBlockImporter getMainnetBlockImporter(
-      final ProtocolSpec protocolSpec, final ProtocolSchedule schedule, final ZkTracer zkTracer) {
+      final ProtocolContext context,
+      final ProtocolSpec protocolSpec,
+      final ProtocolSchedule schedule,
+      final ZkTracer zkTracer) {
     CorsetBlockProcessor corsetBlockProcessor =
         new CorsetBlockProcessor(
             protocolSpec.getTransactionProcessor(),
@@ -159,7 +162,7 @@ private static MainnetBlockImporter getMainnetBlockImporter(
             protocolSpec.getBlockHeaderValidator(),
             protocolSpec.getBlockBodyValidator(),
             corsetBlockProcessor,
-            protocolSpec.getBadBlocksManager());
+            context.getBadBlockManager());
 
     return new MainnetBlockImporter(blockValidator);
   }