test(fast-inbox): streaming inbox mocks for the validator and prover-node unit suites (A-1384)

With streaming as the only consumption path, the validator proposal handler and the prover node derive
consumed messages from L1-to-L2 tree leaf counts and the proposal bucket reference. Give the mocked block
headers a leaf count, carry a genesis-bucket reference on test proposals, and resolve bucket queries to the
genesis bucket so the acceptance checks pass with an empty bundle.

Author: spalladino

yarn-project/prover-node/src/prover-node.test.ts

@@ -173,7 +173,7 @@ describe('ProverNode', () => {
     // getBlockData also feeds collectRegisterData when the rebuild re-registers, so it carries a header too.
     l2BlockSource.getBlockData.mockResolvedValue({
       checkpointNumber: CheckpointNumber(2),
-      header: { lastArchive: { root: Fr.ZERO } },
+      header: { lastArchive: { root: Fr.ZERO }, state: { l1ToL2MessageTree: { nextAvailableLeafIndex: 0 } } },
     } as any);
     await proverNode.handleBlockStreamEvent({
       type: 'chain-pruned',
@@ -639,9 +639,9 @@ describe('ProverNode', () => {
     l2BlockSource.getBlockNumber.mockResolvedValue(undefined);
     setupRegistrationSuccess();
     // getBlockData returns a header that lets isEpochFullyProven bail out as "not proven"
-    // and supplies a lastArchive.root for collectRegisterData.
+    // and supplies a lastArchive.root plus an L1-to-L2 leaf count for collectRegisterData.
     l2BlockSource.getBlockData.mockResolvedValue({
-      header: { lastArchive: { root: Fr.ZERO } },
+      header: { lastArchive: { root: Fr.ZERO }, state: { l1ToL2MessageTree: { nextAvailableLeafIndex: 0 } } },
     } as any);
   }
 
@@ -655,7 +655,7 @@ describe('ProverNode', () => {
     worldState.syncImmediate.mockResolvedValue(undefined as any);
     l1ToL2MessageSource.getL1ToL2Messages.mockResolvedValue([]);
     l2BlockSource.getBlockData.mockResolvedValue({
-      header: { lastArchive: { root: Fr.ZERO } },
+      header: { lastArchive: { root: Fr.ZERO }, state: { l1ToL2MessageTree: { nextAvailableLeafIndex: 0 } } },
     } as any);
     worldState.getSnapshot.mockReturnValue({
       getTreeInfo: () => Promise.resolve({ size: 1n }),
@@ -673,7 +673,15 @@ describe('ProverNode', () => {
       number: CheckpointNumber(checkpointNumber),
       header: { slotNumber: SlotNumber(slot), inboxRollingHash: Fr.ZERO },
       archive: { root: archiveRoot },
-      blocks: [{ number: blockNumber, header: { hash: () => Promise.resolve('0x01') } }],
+      blocks: [
+        {
+          number: blockNumber,
+          header: {
+            hash: () => Promise.resolve('0x01'),
+            state: { l1ToL2MessageTree: { nextAvailableLeafIndex: 0 } },
+          },
+        },
+      ],
       hash: () => new Fr(checkpointNumber),
     } as unknown as Checkpoint;
   }

yarn-project/validator-client/src/validator.test.ts

@@ -32,7 +32,12 @@ import { AztecAddress } from '@aztec/stdlib/aztec-address';
 import { type BlockData, BlockHash, L2Block, type L2BlockSink, type L2BlockSource } from '@aztec/stdlib/block';
 import { type Checkpoint, CheckpointReexecutionTracker, type ProposedCheckpointData } from '@aztec/stdlib/checkpoint';
 import type { SlasherConfig, WorldStateSynchronizer } from '@aztec/stdlib/interfaces/server';
-import { type L1ToL2MessageSource, computeInHashFromL1ToL2Messages } from '@aztec/stdlib/messaging';
+import {
+  type InboxBucket,
+  InboxBucketRef,
+  type L1ToL2MessageSource,
+  computeInHashFromL1ToL2Messages,
+} from '@aztec/stdlib/messaging';
 import type { BlockProposal } from '@aztec/stdlib/p2p';
 import { CheckpointHeader } from '@aztec/stdlib/rollup';
 import {
@@ -448,11 +453,24 @@ describe('ValidatorClient', () => {
           Array.isArray(args) &&
           args[0]?.offenseType === OffenseType.ATTESTED_TO_INVALID_CHECKPOINT_PROPOSAL,
       );
+    // AZIP-22 Fast Inbox: an empty-consumption streaming setup. Proposals reference the genesis Inbox bucket, the
+    // parent block's L1-to-L2 leaf count equals its cumulative total (0), so the derived per-block bundle is empty.
+    const genesisInboxBucket: InboxBucket = {
+      seq: 0n,
+      inboxRollingHash: Fr.ZERO,
+      totalMsgCount: 0n,
+      timestamp: 0n,
+      msgCount: 0,
+      lastMessageIndex: 0n,
+      isOpen: false,
+    };
+    const genesisBucketRef = InboxBucketRef.fromBucket(genesisInboxBucket);
+
     beforeEach(async () => {
       const emptyInHash = computeInHashFromL1ToL2Messages([]);
       const blockHeader = makeBlockHeader(1, { blockNumber: BlockNumber(100), slotNumber: SlotNumber(100) });
       blockNumber = BlockNumber(blockHeader.globalVariables.blockNumber);
-      proposal = await makeBlockProposal({ blockHeader, inHash: emptyInHash });
+      proposal = await makeBlockProposal({ blockHeader, inHash: emptyInHash, bucketRef: genesisBucketRef });
       // The proposal targets slot 100, which under pipelining is built during the previous slot. Set the
       // wall clock to the start of that build slot (target_slot_start - S), matching how a pipelined
       // proposer is positioned when validating an inbound block proposal. With S - 2E = 0 in this config
@@ -512,6 +530,7 @@ describe('ValidatorClient', () => {
           getBlockNumber: () => blockNumber - 1,
           getSlot: () => parentSlot,
           globalVariables: blockHeader.globalVariables,
+          state: { l1ToL2MessageTree: { nextAvailableLeafIndex: 0 } },
         },
         archive: new AppendOnlyTreeSnapshot(Fr.random(), blockNumber - 1),
         blockHash: BlockHash.random(),
@@ -525,6 +544,11 @@ describe('ValidatorClient', () => {
       blockSource.getGenesisValues.mockResolvedValue({ genesisArchiveRoot: new Fr(GENESIS_ARCHIVE_ROOT) });
       blockSource.syncImmediate.mockImplementation(() => Promise.resolve());
 
+      // Resolve every Inbox bucket query to the genesis bucket, so streaming checks accept with an empty bundle.
+      l1ToL2MessageSource.getInboxBucket.mockResolvedValue(genesisInboxBucket);
+      l1ToL2MessageSource.getInboxBucketByTotalMsgCount.mockResolvedValue(genesisInboxBucket);
+      l1ToL2MessageSource.getL1ToL2MessagesBetweenBuckets.mockResolvedValue([]);
+
       const clonedBlockHeader = blockHeader.clone();
       blockBuildResult = {
         publicProcessorDuration: 0,
@@ -727,6 +751,7 @@ describe('ValidatorClient', () => {
         archiveRoot: proposal.archive,
         txHashes: proposal.txHashes,
         signer: selfSigner,
+        bucketRef: genesisBucketRef,
       });
 
       epochCache.getProposerAttesterAddressInSlot.mockResolvedValue(selfSigner.address);