Fix generate_data.ts for Aztec v3 API

- Use TestWallet instead of createPXEClient (removed in v3)
- Use getInitialTestAccountsData and wallet.createSchnorrAccount
- Update imports to use v3 subpaths
- Fix storage slot reference to use user_balances

Co-Authored-By: Claude Opus 4.5 <[email protected]>

Author: critesjosh

note-send-proof/scripts/generate_data.ts

@@ -1,31 +1,35 @@
 import { GettingStartedContract } from '../contract/artifacts/GettingStarted.js';
-import {
-  createPXEClient,
-  waitForPXE,
-  createAztecNodeClient,
-  Fr,
-} from '@aztec/aztec.js';
-import { getInitialTestAccountsWallets } from '@aztec/accounts/testing';
+import { Fr } from '@aztec/aztec.js/fields';
+import { createAztecNodeClient, waitForNode } from '@aztec/aztec.js/node';
+import { getInitialTestAccountsData } from '@aztec/accounts/testing';
+import { TestWallet } from '@aztec/test-wallet/server';
 import { computeNoteHashNonce, computeUniqueNoteHash, deriveStorageSlotInMap, siloNoteHash } from '@aztec/stdlib/hash';
-import { poseidon2HashWithSeparator } from '@aztec/foundation/crypto';
+import { poseidon2HashWithSeparator } from '@aztec/foundation/crypto/poseidon';
 import fs from 'fs';
 import { exit } from 'process';
 
 const NOTE_HASH_SEPARATOR = 1;
-const SANDBOX_URL = 'http://localhost:8080';
+const NODE_URL = 'http://localhost:8080';
 
 async function main() {
-  console.log(`Connecting to PXE at ${SANDBOX_URL}`);
-  const pxe = createPXEClient(SANDBOX_URL);
-  await waitForPXE(pxe);
-  console.log('PXE client connected');
-
-  const wallets = await getInitialTestAccountsWallets(pxe);
-  const deployerWallet = wallets[0];
-  const deployerAddress = deployerWallet.getAddress();
+  console.log(`Connecting to Aztec node at ${NODE_URL}`);
+  const aztecNode = createAztecNodeClient(NODE_URL);
+  await waitForNode(aztecNode);
+  console.log('Aztec node connected');
+
+  const wallet = await TestWallet.create(aztecNode, { dataDirectory: 'pxe-data-gen' });
+  console.log('TestWallet created');
+
+  const accountsData = await getInitialTestAccountsData();
+  const deployerAccount = await wallet.createSchnorrAccount(
+    accountsData[0].secret,
+    accountsData[0].salt,
+    accountsData[0].signingKey
+  );
+  const deployerAddress = deployerAccount.address;
 
   console.log('Deploying GettingStarted contract...');
-  const gettingStarted = await GettingStartedContract.deploy(deployerWallet).send({
+  const gettingStarted = await GettingStartedContract.deploy(wallet, deployerAddress).send({
     from: deployerAddress,
   }).wait();
 
@@ -34,24 +38,20 @@ async function main() {
   const NOTE_VALUE = 69;
 
   console.log('Creating note for user...');
-  const tx = gettingStarted.contract.methods.create_note_for_user(NOTE_VALUE);
-
-  const txExecutionRequest = await tx.create();
-  const txRequestHash = await txExecutionRequest.toTxRequest().hash();
-
-  console.log('TX REQUEST HASH', txRequestHash.toString());
-
-  const sentTx = await tx.send({ from: deployerAddress }).wait();
+  const sentTx = await gettingStarted.contract.methods
+    .create_note_for_user(NOTE_VALUE)
+    .send({ from: deployerAddress })
+    .wait();
   console.log('TX HASH', sentTx.txHash.toString());
 
-  const node = createAztecNodeClient(SANDBOX_URL);
+  const node = createAztecNodeClient(NODE_URL);
   const txEffect = await node.getTxEffect(sentTx.txHash);
 
   if (txEffect === undefined) {
     throw new Error('Cannot find txEffect from tx hash');
   }
 
-  const storageSlot = await deriveStorageSlotInMap(GettingStartedContract.storage.user_private_state.slot, deployerAddress);
+  const storageSlot = await deriveStorageSlotInMap(GettingStartedContract.storage.user_balances.slot, deployerAddress);
 
   const NOTE_RANDOMNESS = new Fr(6969);
 
@@ -61,7 +61,7 @@ async function main() {
 
   const INDEX_OF_NOTE_HASH_IN_TRANSACTION = 0;
 
-  const nonceGenerator = txEffect?.data.nullifiers[0] ?? txRequestHash;
+  const nonceGenerator = txEffect.data.nullifiers[0];
 
   const noteHashNonce = await computeNoteHashNonce(nonceGenerator, INDEX_OF_NOTE_HASH_IN_TRANSACTION);