fix: hello darkness my old friend

Author: LHerskind

yarn-project/end-to-end/src/e2e_epochs/epochs_proof_fails.test.ts

@@ -2,8 +2,9 @@ import { type Logger, getTimestampRangeForEpoch, sleep } from '@aztec/aztec.js';
 import { BatchedBlob } from '@aztec/blob-lib';
 import type { ViemClient } from '@aztec/ethereum';
 import { RollupContract } from '@aztec/ethereum/contracts';
-import { ChainMonitor, type Delayer, waitUntilL1Timestamp } from '@aztec/ethereum/test';
+import { ChainMonitor, DelayedTxUtils, type Delayer, waitUntilL1Timestamp } from '@aztec/ethereum/test';
 import { promiseWithResolvers } from '@aztec/foundation/promise';
+import type { ProverNodePublisher } from '@aztec/prover-node';
 import type { TestProverNode } from '@aztec/prover-node/test';
 import type { L1RollupConstants } from '@aztec/stdlib/epoch-helpers';
 import { Proof } from '@aztec/stdlib/proofs';
@@ -34,8 +35,9 @@ describe('e2e_epochs/epochs_proof_fails', () => {
   beforeEach(async () => {
     // Bumping the epoch duration to 5 because otherwise it takes a full epoch before the actual test starts,
     // which means the prover node is attempting to prove before we setup the mocks.
-    test = await EpochsTestContext.setup({ ethereumSlotDuration: 8, aztecEpochDuration: 5 });
-    ({ proverDelayer, sequencerDelayer, context, l1Client, rollup, constants, logger, monitor } = test);
+    // Set startProverNode to false to prevent automatic startup
+    test = await EpochsTestContext.setup({ ethereumSlotDuration: 8, startProverNode: false });
+    ({ sequencerDelayer, context, l1Client, rollup, constants, logger, monitor } = test);
     ({ L1_BLOCK_TIME_IN_S, L2_SLOT_DURATION_IN_S } = test);
   });
 
@@ -48,6 +50,14 @@ describe('e2e_epochs/epochs_proof_fails', () => {
     // Here we cause a re-org by not publishing the proof for epoch 0 until after the end of epoch 1
     // The proof will be rejected and a re-org will take place
 
+    // Create prover node after test setup to avoid early proving
+    const proverNode = await test.createProverNode();
+    context.proverNode = proverNode;
+
+    // Get the prover delayer from the newly created prover node
+    proverDelayer = (((proverNode as TestProverNode).publisher as ProverNodePublisher).l1TxUtils as DelayedTxUtils)
+      .delayer!;
+
     // Hold off prover tx until end epoch 1
     const [epoch2Start] = getTimestampRangeForEpoch(2n, constants);
     proverDelayer.pauseNextTxUntilTimestamp(epoch2Start);
@@ -85,16 +95,23 @@ describe('e2e_epochs/epochs_proof_fails', () => {
   });
 
   it('aborts proving if end of next epoch is reached', async () => {
+    // Create prover node after test setup to avoid early proving
+    const proverNode = await test.createProverNode();
+
+    // Get the prover delayer from the newly created prover node
+    proverDelayer = (((proverNode as TestProverNode).publisher as ProverNodePublisher).l1TxUtils as DelayedTxUtils)
+      .delayer!;
+
     // Inject a delay in prover node proving equal to the length of an epoch, to make sure deadline will be hit
-    const epochProverManager = (context.proverNode as TestProverNode).prover;
+    const epochProverManager = (proverNode as TestProverNode).prover;
     const originalCreate = epochProverManager.createEpochProver.bind(epochProverManager);
     const finalizeEpochPromise = promiseWithResolvers<void>();
     jest.spyOn(epochProverManager, 'createEpochProver').mockImplementation(() => {
       const prover = originalCreate();
       jest.spyOn(prover, 'finalizeEpoch').mockImplementation(async () => {
-        const seconds = L2_SLOT_DURATION_IN_S * test.epochDuration;
+        const seconds = L2_SLOT_DURATION_IN_S * (test.epochDuration * 2); // Forgive me for I have sinned.
         logger.warn(`Finalize epoch: sleeping ${seconds}s.`);
-        await sleep(L2_SLOT_DURATION_IN_S * test.epochDuration * 1000);
+        await sleep(seconds * 1000);
         logger.warn(`Finalize epoch: returning.`);
         finalizeEpochPromise.resolve();
         const ourPublicInputs = RootRollupPublicInputs.random();
@@ -109,6 +126,7 @@ describe('e2e_epochs/epochs_proof_fails', () => {
       });
       return prover;
     });
+    context.proverNode = proverNode;
 
     await test.waitUntilEpochStarts(1);
     logger.info(`Starting epoch 1`);