@aztec/end-to-end 0.0.1-commit.35158ae7e → 0.0.1-commit.3750d92a7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (62) hide show
  1. package/README.md +27 -0
  2. package/dest/bench/client_flows/client_flows_benchmark.d.ts +1 -1
  3. package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
  4. package/dest/bench/client_flows/client_flows_benchmark.js +11 -26
  5. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts +3 -2
  6. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map +1 -1
  7. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js +1 -1
  8. package/dest/e2e_epochs/epochs_test.js +3 -3
  9. package/dest/e2e_fees/fees_test.js +1 -1
  10. package/dest/e2e_p2p/inactivity_slash_test.js +3 -3
  11. package/dest/e2e_p2p/p2p_network.d.ts +5 -7
  12. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  13. package/dest/e2e_p2p/p2p_network.js +14 -13
  14. package/dest/e2e_p2p/reqresp/utils.js +1 -1
  15. package/dest/e2e_p2p/shared.d.ts +5 -7
  16. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  17. package/dest/e2e_p2p/shared.js +16 -42
  18. package/dest/fixtures/authwit_proxy.d.ts +1 -1
  19. package/dest/fixtures/authwit_proxy.d.ts.map +1 -1
  20. package/dest/fixtures/authwit_proxy.js +4 -0
  21. package/dest/fixtures/e2e_prover_test.d.ts +1 -1
  22. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  23. package/dest/fixtures/e2e_prover_test.js +4 -7
  24. package/dest/fixtures/setup.d.ts +11 -8
  25. package/dest/fixtures/setup.d.ts.map +1 -1
  26. package/dest/fixtures/setup.js +11 -9
  27. package/dest/fixtures/setup_p2p_test.d.ts +6 -6
  28. package/dest/fixtures/setup_p2p_test.d.ts.map +1 -1
  29. package/dest/fixtures/setup_p2p_test.js +8 -8
  30. package/dest/fixtures/token_utils.d.ts +1 -1
  31. package/dest/fixtures/token_utils.d.ts.map +1 -1
  32. package/dest/fixtures/token_utils.js +2 -5
  33. package/dest/legacy-jest-resolver.d.cts +3 -0
  34. package/dest/legacy-jest-resolver.d.cts.map +1 -0
  35. package/dest/spartan/setup_test_wallets.d.ts +1 -1
  36. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  37. package/dest/spartan/setup_test_wallets.js +9 -42
  38. package/dest/test-wallet/test_wallet.d.ts +16 -8
  39. package/dest/test-wallet/test_wallet.d.ts.map +1 -1
  40. package/dest/test-wallet/test_wallet.js +82 -46
  41. package/dest/test-wallet/worker_wallet.d.ts +4 -4
  42. package/dest/test-wallet/worker_wallet.d.ts.map +1 -1
  43. package/dest/test-wallet/worker_wallet_schema.d.ts +7 -2
  44. package/dest/test-wallet/worker_wallet_schema.d.ts.map +1 -1
  45. package/package.json +40 -39
  46. package/src/bench/client_flows/client_flows_benchmark.ts +29 -21
  47. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +3 -6
  48. package/src/e2e_epochs/epochs_test.ts +3 -3
  49. package/src/e2e_fees/fees_test.ts +1 -1
  50. package/src/e2e_p2p/inactivity_slash_test.ts +3 -3
  51. package/src/e2e_p2p/p2p_network.ts +22 -27
  52. package/src/e2e_p2p/reqresp/utils.ts +1 -1
  53. package/src/e2e_p2p/shared.ts +16 -57
  54. package/src/fixtures/authwit_proxy.ts +4 -0
  55. package/src/fixtures/e2e_prover_test.ts +7 -6
  56. package/src/fixtures/setup.ts +21 -17
  57. package/src/fixtures/setup_p2p_test.ts +9 -9
  58. package/src/fixtures/token_utils.ts +1 -4
  59. package/src/legacy-jest-resolver.cjs +135 -0
  60. package/src/spartan/setup_test_wallets.ts +13 -35
  61. package/src/test-wallet/test_wallet.ts +100 -50
  62. package/src/test-wallet/worker_wallet.ts +3 -2
@@ -7,7 +7,7 @@ import { SecretValue } from '@aztec/foundation/config';
7
7
  import { withLoggerBindings } from '@aztec/foundation/log/server';
8
8
  import { bufferToHex } from '@aztec/foundation/string';
9
9
  import type { DateProvider } from '@aztec/foundation/timer';
10
- import type { PublicDataTreeLeaf } from '@aztec/stdlib/trees';
10
+ import type { GenesisData } from '@aztec/stdlib/world-state';
11
11
 
12
12
  import getPort from 'get-port';
13
13
 
@@ -40,7 +40,7 @@ export async function createNodes(
40
40
  bootstrapNodeEnr: string,
41
41
  numNodes: number,
42
42
  bootNodePort: number,
43
- prefilledPublicData?: PublicDataTreeLeaf[],
43
+ genesis?: GenesisData,
44
44
  dataDirectory?: string,
45
45
  metricsPort?: number,
46
46
  indexOffset = 0,
@@ -65,7 +65,7 @@ export async function createNodes(
65
65
  port,
66
66
  bootstrapNodeEnr,
67
67
  validatorIndices,
68
- prefilledPublicData,
68
+ genesis,
69
69
  dataDir,
70
70
  metricsPort,
71
71
  );
@@ -97,7 +97,7 @@ export async function createNode(
97
97
  tcpPort: number,
98
98
  bootstrapNode: string | undefined,
99
99
  addressIndex: number | number[],
100
- prefilledPublicData?: PublicDataTreeLeaf[],
100
+ genesis?: GenesisData,
101
101
  dataDirectory?: string,
102
102
  metricsPort?: number,
103
103
  ) {
@@ -108,7 +108,7 @@ export async function createNode(
108
108
  return await AztecNodeService.createAndSync(
109
109
  validatorConfig,
110
110
  { telemetry, dateProvider },
111
- { prefilledPublicData, dontStartSequencer: config.dontStartSequencer },
111
+ { genesis, dontStartSequencer: config.dontStartSequencer },
112
112
  );
113
113
  });
114
114
  }
@@ -119,7 +119,7 @@ export async function createNonValidatorNode(
119
119
  dateProvider: DateProvider,
120
120
  tcpPort: number,
121
121
  bootstrapNode: string | undefined,
122
- prefilledPublicData?: PublicDataTreeLeaf[],
122
+ genesis?: GenesisData,
123
123
  dataDirectory?: string,
124
124
  metricsPort?: number,
125
125
  ) {
@@ -133,7 +133,7 @@ export async function createNonValidatorNode(
133
133
  sequencerPublisherPrivateKeys: [],
134
134
  };
135
135
  const telemetry = await getEndToEndTestTelemetryClient(metricsPort);
136
- return await AztecNodeService.createAndSync(config, { telemetry, dateProvider }, { prefilledPublicData });
136
+ return await AztecNodeService.createAndSync(config, { telemetry, dateProvider }, { genesis });
137
137
  });
138
138
  }
139
139
 
@@ -143,7 +143,7 @@ export async function createProverNode(
143
143
  bootstrapNode: string | undefined,
144
144
  addressIndex: number,
145
145
  deps: { dateProvider: DateProvider },
146
- prefilledPublicData?: PublicDataTreeLeaf[],
146
+ genesis?: GenesisData,
147
147
  dataDirectory?: string,
148
148
  metricsPort?: number,
149
149
  ): Promise<{ proverNode: AztecNodeService }> {
@@ -159,7 +159,7 @@ export async function createProverNode(
159
159
  { ...config, ...p2pConfig },
160
160
  { dataDirectory },
161
161
  { ...deps, telemetry },
162
- { prefilledPublicData: prefilledPublicData ?? [] },
162
+ { genesis },
163
163
  );
164
164
  });
165
165
  }
@@ -6,11 +6,8 @@ import { TokenContract } from '@aztec/noir-contracts.js/Token';
6
6
 
7
7
  export async function deployToken(wallet: Wallet, admin: AztecAddress, initialAdminBalance: bigint, logger: Logger) {
8
8
  logger.info(`Deploying Token contract...`);
9
- const {
10
- receipt: { contract, instance },
11
- } = await TokenContract.deploy(wallet, admin, 'TokenName', 'TokenSymbol', 18).send({
9
+ const { contract, instance } = await TokenContract.deploy(wallet, admin, 'TokenName', 'TokenSymbol', 18).send({
12
10
  from: admin,
13
- wait: { returnReceipt: true },
14
11
  });
15
12
 
16
13
  if (initialAdminBalance > 0n) {
@@ -0,0 +1,135 @@
1
+ // Custom Jest resolver. When CONTRACT_ARTIFACTS_VERSION is set, redirects *only* JSON artifact files under
2
+ // @aztec/noir-contracts.js/artifacts/, @aztec/noir-test-contracts.js/artifacts/, and @aztec/accounts/artifacts/ to a local cache of the pinned
3
+ // legacy versions. TypeScript wrapper classes (e.g. Token.ts) continue to load from the current workspace and use the
4
+ // current @aztec/aztec.js — only the artifact JSON (the deployed-contract ABI / bytecode / notes surface) is swapped.
5
+ //
6
+ // Why JSON-only: the JSON artifact is the actual interchange surface a "deployed contract" exposes. The TS wrapper is
7
+ // generated client-side ergonomics that's tightly coupled to the current @aztec/aztec.js API. Redirecting the wrapper
8
+ // would couple this test to a moving aztec.js surface and break at import time on unrelated breaking changes; we want
9
+ // to fail only on actual artifact-compat regressions.
10
+ //
11
+ // The cache is populated on demand by running `npm install` into .legacy-contracts/<version>/.
12
+ //
13
+ // Activated by env var; passthrough otherwise.
14
+ /* eslint-disable @typescript-eslint/no-require-imports */
15
+
16
+ const path = require('path');
17
+ const fs = require('fs');
18
+ const { execSync } = require('child_process');
19
+
20
+ const version = process.env.CONTRACT_ARTIFACTS_VERSION;
21
+ const REDIRECTED = ['@aztec/noir-contracts.js', '@aztec/noir-test-contracts.js', '@aztec/accounts'];
22
+
23
+ // Jest sets rootDir to <e2e>/src; this file lives there too.
24
+ const e2eRoot = path.resolve(__dirname, '..');
25
+ const cacheRoot = version ? path.join(e2eRoot, '.legacy-contracts', version) : null;
26
+
27
+ function pkgJsonPath(name) {
28
+ return path.join(cacheRoot, 'node_modules', name, 'package.json');
29
+ }
30
+
31
+ function ensureCache() {
32
+ const missing = REDIRECTED.some(p => !fs.existsSync(pkgJsonPath(p)));
33
+ if (!missing) {
34
+ return;
35
+ }
36
+ fs.mkdirSync(cacheRoot, { recursive: true });
37
+ // Seed a standalone package.json so `npm install --prefix` treats cacheRoot as its own project. Without this, npm
38
+ // walks up and finds the yarn-project workspace root, which breaks on `workspace:` protocol deps and risks
39
+ // clobbering the monorepo's node_modules.
40
+ const seed = path.join(cacheRoot, 'package.json');
41
+ if (!fs.existsSync(seed)) {
42
+ fs.writeFileSync(seed, JSON.stringify({ name: 'legacy-contracts-cache', private: true }));
43
+ }
44
+
45
+ const specs = REDIRECTED.map(p => `${p}@${version}`).join(' ');
46
+ process.stderr.write(`[legacy-contracts] installing ${specs} into ${cacheRoot}\n`);
47
+ // --prefix: install into cacheRoot instead of cwd, so the cache is isolated from the monorepo.
48
+ // --no-save: don't write the installed packages back to the seeded package.json.
49
+ // --ignore-scripts: skip lifecycle scripts (preinstall/postinstall) of the legacy packages and their transitive
50
+ // deps; we only want the files on disk, not to run any build steps.
51
+ // --legacy-peer-deps: tolerate peer-dependency mismatches between the pinned legacy @aztec/* graph and whatever
52
+ // current versions npm would otherwise try to reconcile.
53
+ execSync(`npm install --prefix "${cacheRoot}" --no-save --ignore-scripts --legacy-peer-deps ${specs}`, {
54
+ stdio: 'inherit',
55
+ });
56
+
57
+ // Verify versions on disk match the requested version.
58
+ for (const p of REDIRECTED) {
59
+ const onDisk = JSON.parse(fs.readFileSync(pkgJsonPath(p), 'utf8')).version;
60
+ if (onDisk !== version) {
61
+ throw new Error(`[legacy-contracts] ${p} on disk is ${onDisk}, expected ${version}`);
62
+ }
63
+ }
64
+ }
65
+
66
+ if (version) {
67
+ ensureCache();
68
+ }
69
+
70
+ let bannerPrinted = false;
71
+ const seen = new Set();
72
+
73
+ function printBannerOnce() {
74
+ if (bannerPrinted || !version) {
75
+ return;
76
+ }
77
+ bannerPrinted = true;
78
+ const lines = ['='.repeat(60), `[legacy-contracts][jest] CONTRACT_ARTIFACTS_VERSION=${version}`];
79
+ for (const p of REDIRECTED) {
80
+ const v = JSON.parse(fs.readFileSync(pkgJsonPath(p), 'utf8')).version;
81
+ if (v !== version) {
82
+ throw new Error(`[legacy-contracts] ${p} on disk is ${v}, expected ${version}`);
83
+ }
84
+ lines.push(`[legacy-contracts][jest] redirecting ${p}/artifacts/*.json -> .legacy-contracts/${version}/...`);
85
+ }
86
+ lines.push('='.repeat(60));
87
+ process.stderr.write(lines.join('\n') + '\n');
88
+ }
89
+
90
+ // Match a resolved absolute path against the workspace artifacts dirs and return the legacy cache equivalent, or null
91
+ // if it's not an artifact path we should redirect.
92
+ function legacyArtifactPath(resolved) {
93
+ if (!resolved.endsWith('.json')) {
94
+ return null;
95
+ }
96
+ for (const pkg of REDIRECTED) {
97
+ // pkg = '@aztec/noir-contracts.js' -> match '/noir-contracts.js/artifacts/'
98
+ const dirName = pkg.split('/')[1];
99
+ const marker = `/${dirName}/artifacts/`;
100
+ const idx = resolved.indexOf(marker);
101
+ if (idx === -1) {
102
+ continue;
103
+ }
104
+ const basename = resolved.slice(idx + marker.length);
105
+ return path.join(cacheRoot, 'node_modules', pkg, 'artifacts', basename);
106
+ }
107
+ return null;
108
+ }
109
+
110
+ module.exports = function legacyResolver(request, options) {
111
+ // Always run the default resolver first. We only inspect (and possibly rewrite) the *result*; this catches both
112
+ // bare-specifier imports of `@aztec/noir-contracts.js/artifacts/foo.json` and the relative `../artifacts/foo.json`
113
+ // imports inside the workspace TS wrapper classes — both resolve to the same workspace artifact path that we then
114
+ // redirect.
115
+ const resolved = options.defaultResolver(request, options);
116
+ if (!version) {
117
+ return resolved;
118
+ }
119
+ printBannerOnce();
120
+ const legacy = legacyArtifactPath(resolved);
121
+ if (!legacy) {
122
+ return resolved;
123
+ }
124
+ if (!fs.existsSync(legacy)) {
125
+ throw new Error(
126
+ `[legacy-contracts] artifact ${path.basename(legacy)} not present in legacy cache @${version}; ` +
127
+ `the contract may have been added after that release. Pin a newer CONTRACT_ARTIFACTS_VERSION or skip this test.`,
128
+ );
129
+ }
130
+ if (!seen.has(resolved)) {
131
+ seen.add(resolved);
132
+ process.stderr.write(`[legacy-contracts][jest] redirected ${path.basename(legacy)} -> ${legacy}\n`);
133
+ }
134
+ return legacy;
135
+ };
@@ -138,37 +138,11 @@ async function deployAccountWithDiagnostics(
138
138
  estimateGas?: boolean,
139
139
  ): Promise<void> {
140
140
  const deployMethod = await account.getDeployMethod();
141
- let txHash;
142
141
  let gasSettings: any;
143
- try {
144
- if (estimateGas) {
145
- const sim = await deployMethod.simulate({ from: NO_FROM, fee: { paymentMethod } });
146
- gasSettings = sim.estimatedGas;
147
- logger.info(`${accountLabel} estimated gas: DA=${gasSettings.gasLimits.daGas} L2=${gasSettings.gasLimits.l2Gas}`);
148
- }
149
- const deployResult = await deployMethod.send({
150
- from: NO_FROM,
151
- fee: { paymentMethod, gasSettings },
152
- wait: NO_WAIT,
153
- });
154
- txHash = deployResult.txHash;
155
- await waitForTx(aztecNode, txHash, { timeout: 2400 });
156
- logger.info(`${accountLabel} deployed at ${account.address}`);
157
- } catch (error) {
158
- const blockNumber = await aztecNode.getBlockNumber();
159
- let receipt;
160
- try {
161
- receipt = await aztecNode.getTxReceipt(txHash);
162
- } catch {
163
- receipt = 'unavailable';
164
- }
165
- logger.error(`${accountLabel} deployment failed`, {
166
- txHash: txHash.toString(),
167
- receipt: JSON.stringify(receipt),
168
- currentBlockNumber: blockNumber,
169
- error: String(error),
170
- });
171
- throw error;
142
+ if (estimateGas) {
143
+ const sim = await deployMethod.simulate({ from: NO_FROM, fee: { paymentMethod } });
144
+ gasSettings = sim.estimatedGas;
145
+ logger.info(`${accountLabel} estimated gas: DA=${gasSettings.gasLimits.daGas} L2=${gasSettings.gasLimits.l2Gas}`);
172
146
  }
173
147
 
174
148
  // Track the tx hash across retries so we don't re-send when the previous tx is still pending.
@@ -196,7 +170,7 @@ async function deployAccountWithDiagnostics(
196
170
 
197
171
  if (!sentTxHash) {
198
172
  const deployResult = await deployMethod.send({
199
- from: AztecAddress.ZERO,
173
+ from: NO_FROM,
200
174
  fee: { paymentMethod, gasSettings },
201
175
  wait: NO_WAIT,
202
176
  });
@@ -391,14 +365,18 @@ async function deployTokenAndMint(
391
365
  logger: Logger,
392
366
  ) {
393
367
  logger.verbose(`Deploying TokenContract...`);
394
- const {
395
- receipt: { contract: tokenContract },
396
- } = await TokenContract.deploy(wallet, admin, TOKEN_NAME, TOKEN_SYMBOL, TOKEN_DECIMALS).send({
368
+ const { contract: tokenContract } = await TokenContract.deploy(
369
+ wallet,
370
+ admin,
371
+ TOKEN_NAME,
372
+ TOKEN_SYMBOL,
373
+ TOKEN_DECIMALS,
374
+ ).send({
397
375
  from: admin,
398
376
  fee: {
399
377
  paymentMethod,
400
378
  },
401
- wait: { timeout: 600, returnReceipt: true },
379
+ wait: { timeout: 600 },
402
380
  });
403
381
 
404
382
  const tokenAddress = tokenContract.address;
@@ -1,7 +1,9 @@
1
1
  import { EcdsaKAccountContract, EcdsaRAccountContract } from '@aztec/accounts/ecdsa';
2
2
  import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
3
- import { StubAccountContractArtifact, createStubAccount } from '@aztec/accounts/stub';
3
+ import { StubEcdsaAccountContractArtifact, createStubEcdsaAccount } from '@aztec/accounts/stub/ecdsa';
4
+ import { StubSchnorrAccountContractArtifact, createStubSchnorrAccount } from '@aztec/accounts/stub/schnorr';
4
5
  import { type Account, type AccountContract, NO_FROM } from '@aztec/aztec.js/account';
6
+ import type { CompleteAddress } from '@aztec/aztec.js/addresses';
5
7
  import {
6
8
  type CallIntent,
7
9
  type ContractFunctionInteractionCallIntent,
@@ -13,6 +15,7 @@ import {
13
15
  } from '@aztec/aztec.js/authorization';
14
16
  import type { AztecNode } from '@aztec/aztec.js/node';
15
17
  import { AccountManager, type SendOptions } from '@aztec/aztec.js/wallet';
18
+ import { TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
16
19
  import type { DefaultAccountEntrypointOptions } from '@aztec/entrypoints/account';
17
20
  import { DefaultEntrypoint } from '@aztec/entrypoints/default';
18
21
  import { Fq, Fr } from '@aztec/foundation/curves/bn254';
@@ -25,16 +28,17 @@ import { AztecAddress } from '@aztec/stdlib/aztec-address';
25
28
  import { getContractInstanceFromInstantiationParams } from '@aztec/stdlib/contract';
26
29
  import { deriveSigningKey } from '@aztec/stdlib/keys';
27
30
  import type { NoteDao } from '@aztec/stdlib/note';
28
- import type {
29
- BlockHeader,
31
+ import {
32
+ type BlockHeader,
33
+ type ContractOverrides,
30
34
  SimulationOverrides,
31
- TxExecutionRequest,
32
- TxHash,
33
- TxReceipt,
34
- TxSimulationResult,
35
+ type TxExecutionRequest,
36
+ type TxHash,
37
+ type TxReceipt,
35
38
  } from '@aztec/stdlib/tx';
36
39
  import { ExecutionPayload, mergeExecutionPayloads } from '@aztec/stdlib/tx';
37
40
  import { BaseWallet, type SimulateViaEntrypointOptions } from '@aztec/wallet-sdk/base-wallet';
41
+ import type { AccountType } from '@aztec/wallets/embedded';
38
42
 
39
43
  import { AztecNodeProxy, ProvenTx } from './utils.js';
40
44
 
@@ -44,6 +48,7 @@ import { AztecNodeProxy, ProvenTx } from './utils.js';
44
48
  export interface AccountData {
45
49
  secret: Fr;
46
50
  salt: Fr;
51
+ type?: AccountType;
47
52
  contract: AccountContract;
48
53
  }
49
54
 
@@ -84,50 +89,81 @@ export class TestWallet extends BaseWallet {
84
89
 
85
90
  createSchnorrAccount(secret: Fr, salt: Fr, signingKey?: Fq): Promise<AccountManager> {
86
91
  signingKey = signingKey ?? deriveSigningKey(secret);
87
- const accountData = {
88
- secret,
89
- salt,
90
- contract: new SchnorrAccountContract(signingKey),
91
- };
92
- return this.createAccount(accountData);
92
+ return this.createAccount({ secret, salt, type: 'schnorr', contract: new SchnorrAccountContract(signingKey) });
93
93
  }
94
94
 
95
95
  createECDSARAccount(secret: Fr, salt: Fr, signingKey: Buffer): Promise<AccountManager> {
96
- const accountData = {
96
+ return this.createAccount({
97
97
  secret,
98
98
  salt,
99
+ type: 'ecdsasecp256r1',
99
100
  contract: new EcdsaRAccountContract(signingKey),
100
- };
101
- return this.createAccount(accountData);
101
+ });
102
102
  }
103
103
 
104
104
  createECDSAKAccount(secret: Fr, salt: Fr, signingKey: Buffer): Promise<AccountManager> {
105
- const accountData = {
105
+ return this.createAccount({
106
106
  secret,
107
107
  salt,
108
+ type: 'ecdsasecp256k1',
108
109
  contract: new EcdsaKAccountContract(signingKey),
109
- };
110
- return this.createAccount(accountData);
110
+ });
111
111
  }
112
112
 
113
- async getFakeAccountDataFor(address: AztecAddress) {
114
- const originalAccount = await this.getAccountFromAddress(address);
115
- const originalAddress = originalAccount.getCompleteAddress();
116
- const contractInstance = await this.pxe.getContractInstance(originalAddress.address);
117
- if (!contractInstance) {
118
- throw new Error(`No contract instance found for address: ${originalAddress.address}`);
113
+ /**
114
+ * Builds contract overrides for all provided addresses by replacing their account contracts with stub implementations.
115
+ */
116
+ protected async buildAccountOverrides(addresses: AztecAddress[]): Promise<ContractOverrides> {
117
+ const accounts = await this.getAccounts();
118
+ const contracts: ContractOverrides = {};
119
+
120
+ const filtered = accounts.filter(acc => addresses.some(addr => addr.equals(acc.item)));
121
+
122
+ for (const account of filtered) {
123
+ const address = account.item;
124
+ const originalAccount = await this.getAccountFromAddress(address);
125
+ const completeAddress = originalAccount.getCompleteAddress();
126
+ const contractInstance = await this.pxe.getContractInstance(completeAddress.address);
127
+ if (!contractInstance) {
128
+ throw new Error(
129
+ `No contract instance found for address: ${completeAddress.address} during account override building. This is a bug!`,
130
+ );
131
+ }
132
+
133
+ const stubArtifact = this.getStubArtifactFor(address);
134
+ const stubConstructorArgs =
135
+ this.getTypeFor(address) === 'schnorr' ? [Fr.ZERO, Fr.ZERO] : [Buffer.alloc(32), Buffer.alloc(32)];
136
+ const stubInstance = await getContractInstanceFromInstantiationParams(stubArtifact, {
137
+ salt: Fr.random(),
138
+ constructorArgs: stubConstructorArgs,
139
+ });
140
+
141
+ contracts[address.toString()] = {
142
+ instance: stubInstance,
143
+ artifact: stubArtifact,
144
+ };
119
145
  }
120
- const stubAccount = createStubAccount(originalAddress);
121
- const instance = await getContractInstanceFromInstantiationParams(StubAccountContractArtifact, {
122
- salt: Fr.random(),
123
- });
124
- return {
125
- account: stubAccount,
126
- instance,
127
- artifact: StubAccountContractArtifact,
128
- };
146
+
147
+ return contracts;
148
+ }
149
+
150
+ protected accounts: Map<string, { account: Account; type: AccountType }> = new Map();
151
+
152
+ private getTypeFor(address: AztecAddress): AccountType {
153
+ return this.accounts.get(address.toString())?.type ?? 'schnorr';
154
+ }
155
+
156
+ private getStubArtifactFor(address: AztecAddress) {
157
+ return this.getTypeFor(address) === 'schnorr'
158
+ ? StubSchnorrAccountContractArtifact
159
+ : StubEcdsaAccountContractArtifact;
160
+ }
161
+
162
+ private getStubAccountFor(address: AztecAddress, completeAddress: CompleteAddress) {
163
+ return this.getTypeFor(address) === 'schnorr'
164
+ ? createStubSchnorrAccount(completeAddress)
165
+ : createStubEcdsaAccount(completeAddress);
129
166
  }
130
- protected accounts: Map<string, Account> = new Map();
131
167
 
132
168
  /**
133
169
  * Controls how the test wallet simulates transactions:
@@ -146,22 +182,25 @@ export class TestWallet extends BaseWallet {
146
182
  }
147
183
 
148
184
  protected getAccountFromAddress(address: AztecAddress): Promise<Account> {
149
- const account = this.accounts.get(address?.toString() ?? '');
185
+ const entry = this.accounts.get(address?.toString() ?? '');
150
186
 
151
- if (!account) {
187
+ if (!entry) {
152
188
  throw new Error(`Account not found in wallet for address: ${address}`);
153
189
  }
154
190
 
155
- return Promise.resolve(account);
191
+ return Promise.resolve(entry.account);
156
192
  }
157
193
 
158
194
  getAccounts() {
159
- return Promise.resolve(Array.from(this.accounts.values()).map(acc => ({ alias: '', item: acc.getAddress() })));
195
+ return Promise.resolve(
196
+ Array.from(this.accounts.values()).map(entry => ({ alias: '', item: entry.account.getAddress() })),
197
+ );
160
198
  }
161
199
 
162
200
  async createAccount(accountData?: AccountData): Promise<AccountManager> {
163
201
  const secret = accountData?.secret ?? Fr.random();
164
202
  const salt = accountData?.salt ?? Fr.random();
203
+ const type = accountData?.type ?? 'schnorr';
165
204
  const contract = accountData?.contract ?? new SchnorrAccountContract(GrumpkinScalar.random());
166
205
 
167
206
  const accountManager = await AccountManager.create(this, secret, contract, salt);
@@ -171,7 +210,8 @@ export class TestWallet extends BaseWallet {
171
210
 
172
211
  await this.registerContract(instance, artifact, secret);
173
212
 
174
- this.accounts.set(accountManager.address.toString(), await accountManager.getAccount());
213
+ const address = accountManager.address.toString();
214
+ this.accounts.set(address, { account: await accountManager.getAccount(), type });
175
215
 
176
216
  return accountManager;
177
217
  }
@@ -217,9 +257,12 @@ export class TestWallet extends BaseWallet {
217
257
  protected override async simulateViaEntrypoint(
218
258
  executionPayload: ExecutionPayload,
219
259
  opts: SimulateViaEntrypointOptions,
220
- ): Promise<TxSimulationResult> {
221
- const { from, feeOptions, scopes, skipTxValidation, skipFeeEnforcement } = opts;
260
+ ): Promise<TxSimulationResultWithAppOffset> {
261
+ const { from, feeOptions, additionalScopes, skipTxValidation, skipFeeEnforcement } = opts;
262
+ const scopes = this.scopesFrom(from, additionalScopes);
222
263
  const skipKernels = this.simulationMode !== 'full';
264
+ const useOverride = this.simulationMode === 'kernelless-override';
265
+
223
266
  const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
224
267
  const finalExecutionPayload = feeExecutionPayload
225
268
  ? mergeExecutionPayloads([feeExecutionPayload, executionPayload])
@@ -228,18 +271,19 @@ export class TestWallet extends BaseWallet {
228
271
 
229
272
  let overrides: SimulationOverrides | undefined;
230
273
  let txRequest: TxExecutionRequest;
274
+ if (useOverride) {
275
+ const accountOverrides = await this.buildAccountOverrides(scopes);
276
+ overrides = new SimulationOverrides(accountOverrides);
277
+ }
278
+
231
279
  if (from === NO_FROM) {
232
280
  const entrypoint = new DefaultEntrypoint();
233
281
  txRequest = await entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
234
282
  } else {
235
- const useOverride = this.simulationMode === 'kernelless-override';
236
283
  let fromAccount: Account;
237
284
  if (useOverride) {
238
- const { account, instance, artifact } = await this.getFakeAccountDataFor(from);
239
- fromAccount = account;
240
- overrides = {
241
- contracts: { [from.toString()]: { instance, artifact } },
242
- };
285
+ const originalAccount = await this.getAccountFromAddress(from);
286
+ fromAccount = this.getStubAccountFor(from, originalAccount.getCompleteAddress());
243
287
  } else {
244
288
  fromAccount = await this.getAccountFromAddress(from);
245
289
  }
@@ -257,7 +301,7 @@ export class TestWallet extends BaseWallet {
257
301
  );
258
302
  }
259
303
 
260
- return this.pxe.simulateTx(txRequest, {
304
+ const result = await this.pxe.simulateTx(txRequest, {
261
305
  simulatePublic: true,
262
306
  skipKernels,
263
307
  skipFeeEnforcement,
@@ -265,10 +309,16 @@ export class TestWallet extends BaseWallet {
265
309
  overrides,
266
310
  scopes,
267
311
  });
312
+ const appCallOffset = await this.computeAppCallOffset(from, feeOptions);
313
+ return TxSimulationResultWithAppOffset.fromResultAndOffset(result, appCallOffset);
268
314
  }
269
315
 
270
316
  async proveTx(exec: ExecutionPayload, opts: Omit<SendOptions, 'wait'>): Promise<ProvenTx> {
271
- const fee = await this.completeFeeOptions(opts.from, exec.feePayer, opts.fee?.gasSettings);
317
+ const fee = await this.completeFeeOptions({
318
+ from: opts.from,
319
+ feePayer: exec.feePayer,
320
+ gasSettings: opts.fee?.gasSettings,
321
+ });
272
322
  const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(exec, opts.from, fee);
273
323
  const txProvingResult = await this.pxe.proveTx(txRequest, this.scopesFrom(opts.from, opts.additionalScopes));
274
324
  return new ProvenTx(
@@ -13,6 +13,7 @@ import type {
13
13
  ProfileOptions,
14
14
  SendOptions,
15
15
  SimulateOptions,
16
+ TxSimulationResultWithAppOffset,
16
17
  Wallet,
17
18
  WalletCapabilities,
18
19
  } from '@aztec/aztec.js/wallet';
@@ -29,7 +30,7 @@ import type { ContractArtifact, EventMetadataDefinition, FunctionCall } from '@a
29
30
  import type { AuthWitness } from '@aztec/stdlib/auth-witness';
30
31
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
31
32
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
32
- import type { ExecutionPayload, TxProfileResult, TxSimulationResult, UtilityExecutionResult } from '@aztec/stdlib/tx';
33
+ import type { ExecutionPayload, TxProfileResult, UtilityExecutionResult } from '@aztec/stdlib/tx';
33
34
  import { Tx } from '@aztec/stdlib/tx';
34
35
 
35
36
  import { Worker } from 'worker_threads';
@@ -165,7 +166,7 @@ export class WorkerWallet implements Wallet {
165
166
  return this.call('registerContract', instance, artifact, secretKey);
166
167
  }
167
168
 
168
- simulateTx(exec: ExecutionPayload, opts: SimulateOptions): Promise<TxSimulationResult> {
169
+ simulateTx(exec: ExecutionPayload, opts: SimulateOptions): Promise<TxSimulationResultWithAppOffset> {
169
170
  return this.call('simulateTx', exec, opts);
170
171
  }
171
172