More test fixes

AztecProtocol · Aug 16, 2024 · bfec6f6 · bfec6f6
1 parent e323883
commit bfec6f6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/yarn-project/end-to-end/src/e2e_prover/e2e_prover_test.ts b/yarn-project/end-to-end/src/e2e_prover/e2e_prover_test.ts
@@ -39,7 +39,7 @@ import {
   createSnapshotManager,
   publicDeployAccounts,
 } from '../fixtures/snapshot_manager.js';
-import { setupPXEService } from '../fixtures/utils.js';
+import { getPrivateKeyFromIndex, setupPXEService } from '../fixtures/utils.js';
 import { TokenSimulator } from '../simulators/token_simulator.js';
 
 const { E2E_DATA_PATH: dataPath } = process.env;
@@ -234,6 +234,9 @@ export class FullProverTest {
       { blockUntilSync: true },
     );
 
+    // The simulated prover node (now shutdown) used private key index 2
+    const proverNodePrivateKey = getPrivateKeyFromIndex(2);
+
     this.logger.verbose('Starting fully proven prover node');
     const proverConfig: ProverNodeConfig = {
       ...this.context.aztecNodeConfig,
@@ -242,6 +245,7 @@ export class FullProverTest {
       proverId: new Fr(81),
       realProofs: true,
       proverAgentConcurrency: 2,
+      publisherPrivateKey: `0x${proverNodePrivateKey!.toString('hex')}`,
     };
     this.proverNode = await createProverNode(proverConfig, {
       aztecNodeTxProvider: this.aztecNode,

diff --git a/yarn-project/end-to-end/src/fixtures/snapshot_manager.ts b/yarn-project/end-to-end/src/fixtures/snapshot_manager.ts
@@ -266,6 +266,7 @@ async function setupFromFresh(
   const publisherPrivKey = publisherPrivKeyRaw === null ? null : Buffer.from(publisherPrivKeyRaw);
 
   const validatorPrivKey = getPrivateKeyFromIndex(1);
+  const proverNodePrivateKey = getPrivateKeyFromIndex(2);
 
   aztecNodeConfig.publisherPrivateKey = `0x${publisherPrivKey!.toString('hex')}`;
   aztecNodeConfig.validatorPrivateKey = `0x${validatorPrivKey!.toString('hex')}`;
@@ -312,6 +313,7 @@ async function setupFromFresh(
     proverId: new Fr(42),
     realProofs: false,
     proverAgentConcurrency: 2,
+    publisherPrivateKey: `0x${proverNodePrivateKey!.toString('hex')}`,
   };
   const proverNode = await createProverNode(proverConfig, {
     aztecNodeTxProvider: aztecNode,