@aztec/end-to-end 0.0.1-commit.f504929 → 0.0.1-commit.f5d02921e

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 (73) hide show
  1. package/dest/bench/client_flows/client_flows_benchmark.d.ts +1 -1
  2. package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
  3. package/dest/bench/client_flows/client_flows_benchmark.js +20 -28
  4. package/dest/e2e_epochs/epochs_test.d.ts +3 -1
  5. package/dest/e2e_epochs/epochs_test.d.ts.map +1 -1
  6. package/dest/e2e_epochs/epochs_test.js +5 -2
  7. package/dest/e2e_fees/fees_test.d.ts +1 -1
  8. package/dest/e2e_fees/fees_test.d.ts.map +1 -1
  9. package/dest/e2e_fees/fees_test.js +8 -1
  10. package/dest/e2e_p2p/inactivity_slash_test.js +1 -1
  11. package/dest/e2e_p2p/p2p_network.d.ts +1 -1
  12. package/dest/e2e_p2p/p2p_network.js +1 -1
  13. package/dest/e2e_p2p/reqresp/utils.d.ts +1 -1
  14. package/dest/e2e_p2p/reqresp/utils.d.ts.map +1 -1
  15. package/dest/e2e_p2p/reqresp/utils.js +15 -2
  16. package/dest/e2e_p2p/shared.d.ts +12 -6
  17. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  18. package/dest/e2e_p2p/shared.js +17 -9
  19. package/dest/fixtures/e2e_prover_test.d.ts +4 -3
  20. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  21. package/dest/fixtures/e2e_prover_test.js +5 -10
  22. package/dest/fixtures/get_bb_config.d.ts +1 -1
  23. package/dest/fixtures/get_bb_config.d.ts.map +1 -1
  24. package/dest/fixtures/get_bb_config.js +5 -5
  25. package/dest/fixtures/setup.d.ts +6 -1
  26. package/dest/fixtures/setup.d.ts.map +1 -1
  27. package/dest/fixtures/setup.js +13 -9
  28. package/dest/fixtures/token_utils.d.ts +1 -1
  29. package/dest/fixtures/token_utils.d.ts.map +1 -1
  30. package/dest/fixtures/token_utils.js +2 -5
  31. package/dest/shared/uniswap_l1_l2.d.ts +1 -1
  32. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  33. package/dest/shared/uniswap_l1_l2.js +9 -12
  34. package/dest/simulators/lending_simulator.d.ts +1 -1
  35. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  36. package/dest/simulators/lending_simulator.js +2 -2
  37. package/dest/simulators/token_simulator.js +1 -1
  38. package/dest/spartan/setup_test_wallets.d.ts +4 -2
  39. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  40. package/dest/spartan/setup_test_wallets.js +78 -16
  41. package/dest/spartan/utils/config.d.ts +4 -1
  42. package/dest/spartan/utils/config.d.ts.map +1 -1
  43. package/dest/spartan/utils/config.js +1 -0
  44. package/dest/spartan/utils/index.d.ts +2 -1
  45. package/dest/spartan/utils/index.d.ts.map +1 -1
  46. package/dest/spartan/utils/index.js +2 -0
  47. package/dest/spartan/utils/pod_logs.d.ts +25 -0
  48. package/dest/spartan/utils/pod_logs.d.ts.map +1 -0
  49. package/dest/spartan/utils/pod_logs.js +74 -0
  50. package/dest/test-wallet/test_wallet.d.ts +10 -17
  51. package/dest/test-wallet/test_wallet.d.ts.map +1 -1
  52. package/dest/test-wallet/test_wallet.js +47 -48
  53. package/dest/test-wallet/worker_wallet_schema.d.ts +2 -2
  54. package/package.json +39 -40
  55. package/src/bench/client_flows/client_flows_benchmark.ts +34 -22
  56. package/src/e2e_epochs/epochs_test.ts +14 -2
  57. package/src/e2e_fees/fees_test.ts +4 -1
  58. package/src/e2e_p2p/inactivity_slash_test.ts +1 -1
  59. package/src/e2e_p2p/p2p_network.ts +1 -1
  60. package/src/e2e_p2p/reqresp/utils.ts +23 -2
  61. package/src/e2e_p2p/shared.ts +20 -10
  62. package/src/fixtures/e2e_prover_test.ts +7 -15
  63. package/src/fixtures/get_bb_config.ts +7 -6
  64. package/src/fixtures/setup.ts +17 -8
  65. package/src/fixtures/token_utils.ts +1 -4
  66. package/src/shared/uniswap_l1_l2.ts +29 -24
  67. package/src/simulators/lending_simulator.ts +4 -2
  68. package/src/simulators/token_simulator.ts +1 -1
  69. package/src/spartan/setup_test_wallets.ts +103 -15
  70. package/src/spartan/utils/config.ts +1 -0
  71. package/src/spartan/utils/index.ts +3 -0
  72. package/src/spartan/utils/pod_logs.ts +99 -0
  73. package/src/test-wallet/test_wallet.ts +60 -65
@@ -1,4 +1,5 @@
1
1
  import { generateSchnorrAccounts } from '@aztec/accounts/testing';
2
+ import { NO_FROM } from '@aztec/aztec.js/account';
2
3
  import { AztecAddress } from '@aztec/aztec.js/addresses';
3
4
  import { NO_WAIT } from '@aztec/aztec.js/contracts';
4
5
  import { L1FeeJuicePortalManager } from '@aztec/aztec.js/ethereum';
@@ -10,7 +11,7 @@ import type { Wallet } from '@aztec/aztec.js/wallet';
10
11
  import { createEthereumChain } from '@aztec/ethereum/chain';
11
12
  import { createExtendedL1Client } from '@aztec/ethereum/client';
12
13
  import type { Logger } from '@aztec/foundation/log';
13
- import { retryUntil } from '@aztec/foundation/retry';
14
+ import { makeBackoff, retry, retryUntil } from '@aztec/foundation/retry';
14
15
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
15
16
  import type { AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
16
17
  import { registerInitialLocalNetworkAccountsInWallet } from '@aztec/wallets/testing';
@@ -89,7 +90,7 @@ export async function deploySponsoredTestAccountsWithTokens(
89
90
  const paymentMethod = new SponsoredFeePaymentMethod(await getSponsoredFPCAddress());
90
91
  const recipientDeployMethod = await recipientAccount.getDeployMethod();
91
92
  await recipientDeployMethod.send({
92
- from: AztecAddress.ZERO,
93
+ from: NO_FROM,
93
94
  fee: { paymentMethod },
94
95
  wait: { timeout: 2400 },
95
96
  });
@@ -97,7 +98,7 @@ export async function deploySponsoredTestAccountsWithTokens(
97
98
  fundedAccounts.map(async a => {
98
99
  const deployMethod = await a.getDeployMethod();
99
100
  await deployMethod.send({
100
- from: AztecAddress.ZERO,
101
+ from: NO_FROM,
101
102
  fee: { paymentMethod },
102
103
  wait: { timeout: 2400 },
103
104
  }); // increase timeout on purpose in order to account for two empty epochs
@@ -129,16 +130,27 @@ export async function deploySponsoredTestAccountsWithTokens(
129
130
  }
130
131
 
131
132
  async function deployAccountWithDiagnostics(
132
- account: { getDeployMethod: () => Promise<{ send: (opts: any) => any }>; address: any },
133
+ account: { getDeployMethod: () => Promise<{ simulate: (opts: any) => any; send: (opts: any) => any }>; address: any },
133
134
  paymentMethod: SponsoredFeePaymentMethod,
134
135
  aztecNode: AztecNode,
135
136
  logger: Logger,
136
137
  accountLabel: string,
138
+ estimateGas?: boolean,
137
139
  ): Promise<void> {
138
140
  const deployMethod = await account.getDeployMethod();
139
141
  let txHash;
142
+ let gasSettings: any;
140
143
  try {
141
- const deployResult = await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod }, wait: NO_WAIT });
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
+ });
142
154
  txHash = deployResult.txHash;
143
155
  await waitForTx(aztecNode, txHash, { timeout: 2400 });
144
156
  logger.info(`${accountLabel} deployed at ${account.address}`);
@@ -158,21 +170,77 @@ async function deployAccountWithDiagnostics(
158
170
  });
159
171
  throw error;
160
172
  }
173
+
174
+ // Track the tx hash across retries so we don't re-send when the previous tx is still pending.
175
+ let sentTxHash: { txHash: any } | undefined;
176
+
177
+ await retry(
178
+ async () => {
179
+ // Check if already deployed (handles case where previous attempt succeeded but waitForTx timed out)
180
+ const existing = await aztecNode.getContract(account.address);
181
+ if (existing) {
182
+ logger.info(`${accountLabel} already deployed at ${account.address}, skipping`);
183
+ return;
184
+ }
185
+
186
+ // If we already sent a tx, check if it was dropped before deciding to re-send.
187
+ if (sentTxHash) {
188
+ const prevReceipt = await aztecNode.getTxReceipt(sentTxHash.txHash);
189
+ if (prevReceipt.isDropped()) {
190
+ logger.info(`${accountLabel} previous tx ${sentTxHash.txHash} was dropped, re-sending`);
191
+ sentTxHash = undefined;
192
+ } else {
193
+ logger.info(`${accountLabel} previous tx ${sentTxHash.txHash} still pending, waiting again...`);
194
+ }
195
+ }
196
+
197
+ if (!sentTxHash) {
198
+ const deployResult = await deployMethod.send({
199
+ from: AztecAddress.ZERO,
200
+ fee: { paymentMethod, gasSettings },
201
+ wait: NO_WAIT,
202
+ });
203
+ sentTxHash = { txHash: deployResult.txHash };
204
+ logger.info(`${accountLabel} tx sent`, { txHash: sentTxHash.txHash.toString() });
205
+ }
206
+
207
+ const receipt = await waitForTx(aztecNode, sentTxHash.txHash, { timeout: 600 });
208
+ if (receipt.isDropped()) {
209
+ sentTxHash = undefined;
210
+ throw new Error(`${accountLabel} tx was dropped, retrying...`);
211
+ }
212
+ logger.info(`${accountLabel} deployed at ${account.address}`);
213
+ },
214
+ `deploy ${accountLabel}`,
215
+ makeBackoff([1, 2, 4, 8, 16]),
216
+ logger,
217
+ );
161
218
  }
162
219
 
163
220
  async function deployAccountsInBatches(
164
- accounts: { getDeployMethod: () => Promise<{ send: (opts: any) => any }>; address: any }[],
221
+ accounts: {
222
+ getDeployMethod: () => Promise<{ simulate: (opts: any) => any; send: (opts: any) => any }>;
223
+ address: any;
224
+ }[],
165
225
  paymentMethod: SponsoredFeePaymentMethod,
166
226
  aztecNode: AztecNode,
167
227
  logger: Logger,
168
228
  labelPrefix: string,
169
229
  batchSize = 2,
230
+ estimateGas?: boolean,
170
231
  ): Promise<void> {
171
232
  for (let i = 0; i < accounts.length; i += batchSize) {
172
233
  const batch = accounts.slice(i, i + batchSize);
173
234
  await Promise.all(
174
235
  batch.map((account, idx) =>
175
- deployAccountWithDiagnostics(account, paymentMethod, aztecNode, logger, `${labelPrefix}${i + idx + 1}`),
236
+ deployAccountWithDiagnostics(
237
+ account,
238
+ paymentMethod,
239
+ aztecNode,
240
+ logger,
241
+ `${labelPrefix}${i + idx + 1}`,
242
+ estimateGas,
243
+ ),
176
244
  ),
177
245
  );
178
246
  }
@@ -183,6 +251,7 @@ export async function deploySponsoredTestAccounts(
183
251
  aztecNode: AztecNode,
184
252
  logger: Logger,
185
253
  numberOfFundedWallets = 1,
254
+ opts?: { estimateGas?: boolean },
186
255
  ): Promise<TestAccountsWithoutTokens> {
187
256
  const [recipient, ...funded] = await generateSchnorrAccounts(numberOfFundedWallets + 1);
188
257
  const recipientAccount = await wallet.createSchnorrAccount(recipient.secret, recipient.salt);
@@ -192,8 +261,23 @@ export async function deploySponsoredTestAccounts(
192
261
 
193
262
  const paymentMethod = new SponsoredFeePaymentMethod(await getSponsoredFPCAddress());
194
263
 
195
- await deployAccountWithDiagnostics(recipientAccount, paymentMethod, aztecNode, logger, 'Recipient account');
196
- await deployAccountsInBatches(fundedAccounts, paymentMethod, aztecNode, logger, 'Funded account ', 2);
264
+ await deployAccountWithDiagnostics(
265
+ recipientAccount,
266
+ paymentMethod,
267
+ aztecNode,
268
+ logger,
269
+ 'Recipient account',
270
+ opts?.estimateGas,
271
+ );
272
+ await deployAccountsInBatches(
273
+ fundedAccounts,
274
+ paymentMethod,
275
+ aztecNode,
276
+ logger,
277
+ 'Funded account ',
278
+ 2,
279
+ opts?.estimateGas,
280
+ );
197
281
 
198
282
  return {
199
283
  aztecNode,
@@ -231,7 +315,7 @@ export async function deployTestAccountsWithTokens(
231
315
  fundedAccounts.map(async (a, i) => {
232
316
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(a.address, claims[i]);
233
317
  const deployMethod = await a.getDeployMethod();
234
- await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod } });
318
+ await deployMethod.send({ from: NO_FROM, fee: { paymentMethod } });
235
319
  logger.info(`Account deployed at ${a.address}`);
236
320
  }),
237
321
  );
@@ -275,7 +359,7 @@ async function bridgeL1FeeJuice(
275
359
  const claim = await portal.bridgeTokensPublic(recipient, amount, true /* mint */);
276
360
 
277
361
  const isSynced = async () =>
278
- (await aztecNode.getL1ToL2MessageBlock(Fr.fromHexString(claim.messageHash))) !== undefined;
362
+ (await aztecNode.getL1ToL2MessageCheckpoint(Fr.fromHexString(claim.messageHash))) !== undefined;
279
363
  await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 0.5);
280
364
 
281
365
  log.info(`Created a claim for ${amount} L1 fee juice to ${recipient}.`, claim);
@@ -307,14 +391,18 @@ async function deployTokenAndMint(
307
391
  logger: Logger,
308
392
  ) {
309
393
  logger.verbose(`Deploying TokenContract...`);
310
- const {
311
- receipt: { contract: tokenContract },
312
- } = await TokenContract.deploy(wallet, admin, TOKEN_NAME, TOKEN_SYMBOL, TOKEN_DECIMALS).send({
394
+ const { contract: tokenContract } = await TokenContract.deploy(
395
+ wallet,
396
+ admin,
397
+ TOKEN_NAME,
398
+ TOKEN_SYMBOL,
399
+ TOKEN_DECIMALS,
400
+ ).send({
313
401
  from: admin,
314
402
  fee: {
315
403
  paymentMethod,
316
404
  },
317
- wait: { timeout: 600, returnReceipt: true },
405
+ wait: { timeout: 600 },
318
406
  });
319
407
 
320
408
  const tokenAddress = tokenContract.address;
@@ -8,6 +8,7 @@ const logger = createLogger('e2e:k8s-utils');
8
8
  const testConfigSchema = z.object({
9
9
  NAMESPACE: z.string().default('scenario'),
10
10
  REAL_VERIFIER: schemas.Boolean.optional().default(true),
11
+ DEBUG_FORCE_TX_PROOF_VERIFICATION: schemas.Boolean.optional().default(true),
11
12
  CREATE_ETH_DEVNET: schemas.Boolean.optional().default(false),
12
13
  L1_RPC_URLS_JSON: z.string().optional(),
13
14
  L1_ACCOUNT_MNEMONIC: z.string().optional(),
@@ -66,3 +66,6 @@ export { getPublicViemClient, getL1DeploymentAddresses, getNodeClient } from './
66
66
 
67
67
  // Health checks
68
68
  export { ChainHealth, type ChainHealthSnapshot } from './health.js';
69
+
70
+ // Pod log extraction
71
+ export { type BlockBuiltLogEntry, fetchBlockBuiltLogs } from './pod_logs.js';
@@ -0,0 +1,99 @@
1
+ import type { Logger } from '@aztec/foundation/log';
2
+
3
+ import { exec } from 'child_process';
4
+ import { promisify } from 'util';
5
+
6
+ import { getSequencers } from './nodes.js';
7
+
8
+ const execAsync = promisify(exec);
9
+
10
+ /** Parsed l2-block-built stats from a sequencer pod log line. */
11
+ export type BlockBuiltLogEntry = {
12
+ blockNumber: number;
13
+ txCount: number;
14
+ duration: number;
15
+ publicProcessDuration: number;
16
+ manaPerSec: number;
17
+ privateLogCount: number;
18
+ publicLogCount: number;
19
+ contractClassLogCount: number;
20
+ contractClassLogSize: number;
21
+ };
22
+
23
+ const FIELDS: (keyof BlockBuiltLogEntry)[] = [
24
+ 'blockNumber',
25
+ 'txCount',
26
+ 'duration',
27
+ 'publicProcessDuration',
28
+ 'manaPerSec',
29
+ 'privateLogCount',
30
+ 'publicLogCount',
31
+ 'contractClassLogCount',
32
+ 'contractClassLogSize',
33
+ ];
34
+
35
+ /**
36
+ * Fetches l2-block-built log entries from sequencer pods for given block numbers.
37
+ * Queries all validator pods (only the proposer will have the log for a given block).
38
+ *
39
+ * @param namespace - Kubernetes namespace
40
+ * @param sinceTime - ISO 8601 timestamp to limit log search (e.g., from before block building was re-enabled)
41
+ * @param blockNumbers - Set of block numbers to filter for
42
+ * @param logger - Logger instance
43
+ * @returns Array of parsed BlockBuiltLogEntry, de-duplicated by blockNumber, sorted ascending
44
+ */
45
+ export async function fetchBlockBuiltLogs(
46
+ namespace: string,
47
+ sinceTime: string,
48
+ blockNumbers: Set<number>,
49
+ logger: Logger,
50
+ ): Promise<BlockBuiltLogEntry[]> {
51
+ const pods = await getSequencers(namespace);
52
+ const entriesByBlock = new Map<number, BlockBuiltLogEntry>();
53
+
54
+ // Subtract 60s from sinceTime to account for clock skew between test runner and k8s pods.
55
+ // Block number filtering ensures we only match the right blocks, so extra lines are harmless.
56
+ const sinceDate = new Date(new Date(sinceTime).getTime() - 60_000);
57
+ const sinceFlag = sinceDate.toISOString();
58
+
59
+ for (const pod of pods) {
60
+ try {
61
+ const cmd = `kubectl logs ${pod} -n ${namespace} -c aztec --since-time=${sinceFlag}`;
62
+ logger.info(`Fetching logs: ${cmd}`);
63
+ const { stdout } = await execAsync(cmd, { maxBuffer: 10 * 1024 * 1024 });
64
+
65
+ const lines = stdout.split('\n');
66
+ const matchingLines = lines.filter(l => l.includes('l2-block-built'));
67
+ logger.info(`Pod ${pod}: ${lines.length} log lines, ${matchingLines.length} contain l2-block-built`);
68
+
69
+ for (const line of matchingLines) {
70
+ try {
71
+ const parsed = JSON.parse(line);
72
+ if (parsed.eventName !== 'l2-block-built' || !blockNumbers.has(parsed.blockNumber)) {
73
+ continue;
74
+ }
75
+ if (entriesByBlock.has(parsed.blockNumber)) {
76
+ continue;
77
+ }
78
+ const entry: BlockBuiltLogEntry = {} as BlockBuiltLogEntry;
79
+ for (const field of FIELDS) {
80
+ entry[field] = parsed[field] ?? 0;
81
+ }
82
+ entriesByBlock.set(entry.blockNumber, entry);
83
+ logger.verbose(`Parsed l2-block-built log for block ${entry.blockNumber}`, entry);
84
+ } catch {
85
+ // Not valid JSON, skip
86
+ }
87
+ }
88
+ } catch (err) {
89
+ logger.warn(`Failed to fetch logs from pod ${pod}: ${err}`);
90
+ }
91
+ }
92
+
93
+ if (entriesByBlock.size < blockNumbers.size) {
94
+ const missing = [...blockNumbers].filter(bn => !entriesByBlock.has(bn));
95
+ logger.warn(`Missing l2-block-built logs for block(s): ${missing.join(', ')}`);
96
+ }
97
+
98
+ return [...entriesByBlock.values()].sort((a, b) => a.blockNumber - b.blockNumber);
99
+ }
@@ -1,7 +1,7 @@
1
1
  import { EcdsaKAccountContract, EcdsaRAccountContract } from '@aztec/accounts/ecdsa';
2
2
  import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
3
3
  import { StubAccountContractArtifact, createStubAccount } from '@aztec/accounts/stub';
4
- import { type Account, type AccountContract, SignerlessAccount } from '@aztec/aztec.js/account';
4
+ import { type Account, type AccountContract, NO_FROM } from '@aztec/aztec.js/account';
5
5
  import {
6
6
  type CallIntent,
7
7
  type ContractFunctionInteractionCallIntent,
@@ -14,9 +14,10 @@ import {
14
14
  import type { AztecNode } from '@aztec/aztec.js/node';
15
15
  import { AccountManager, type SendOptions } from '@aztec/aztec.js/wallet';
16
16
  import type { DefaultAccountEntrypointOptions } from '@aztec/entrypoints/account';
17
+ import { DefaultEntrypoint } from '@aztec/entrypoints/default';
17
18
  import { Fq, Fr } from '@aztec/foundation/curves/bn254';
18
19
  import { GrumpkinScalar } from '@aztec/foundation/curves/grumpkin';
19
- import type { AccessScopes, NotesFilter } from '@aztec/pxe/client/lazy';
20
+ import type { NotesFilter } from '@aztec/pxe/client/lazy';
20
21
  import { type PXEConfig, getPXEConfig } from '@aztec/pxe/config';
21
22
  import { PXE, type PXECreationOptions, createPXE } from '@aztec/pxe/server';
22
23
  import { AuthWitness } from '@aztec/stdlib/auth-witness';
@@ -24,9 +25,16 @@ import { AztecAddress } from '@aztec/stdlib/aztec-address';
24
25
  import { getContractInstanceFromInstantiationParams } from '@aztec/stdlib/contract';
25
26
  import { deriveSigningKey } from '@aztec/stdlib/keys';
26
27
  import type { NoteDao } from '@aztec/stdlib/note';
27
- import type { BlockHeader, TxHash, TxReceipt, TxSimulationResult } from '@aztec/stdlib/tx';
28
+ import type {
29
+ BlockHeader,
30
+ SimulationOverrides,
31
+ TxExecutionRequest,
32
+ TxHash,
33
+ TxReceipt,
34
+ TxSimulationResult,
35
+ } from '@aztec/stdlib/tx';
28
36
  import { ExecutionPayload, mergeExecutionPayloads } from '@aztec/stdlib/tx';
29
- import { BaseWallet, type FeeOptions } from '@aztec/wallet-sdk/base-wallet';
37
+ import { BaseWallet, type SimulateViaEntrypointOptions } from '@aztec/wallet-sdk/base-wallet';
30
38
 
31
39
  import { AztecNodeProxy, ProvenTx } from './utils.js';
32
40
 
@@ -104,10 +112,7 @@ export class TestWallet extends BaseWallet {
104
112
 
105
113
  async getFakeAccountDataFor(address: AztecAddress) {
106
114
  const originalAccount = await this.getAccountFromAddress(address);
107
- if (originalAccount instanceof SignerlessAccount) {
108
- throw new Error(`Cannot create fake account data for SignerlessAccount at address: ${address}`);
109
- }
110
- const originalAddress = (originalAccount as Account).getCompleteAddress();
115
+ const originalAddress = originalAccount.getCompleteAddress();
111
116
  const contractInstance = await this.pxe.getContractInstance(originalAddress.address);
112
117
  if (!contractInstance) {
113
118
  throw new Error(`No contract instance found for address: ${originalAddress.address}`);
@@ -125,21 +130,15 @@ export class TestWallet extends BaseWallet {
125
130
  protected accounts: Map<string, Account> = new Map();
126
131
 
127
132
  /**
128
- * Toggle for running "simulated simulations" when calling simulateTx.
129
- *
130
- * When this flag is true, simulateViaEntrypoint constructs a request using a fake account
131
- * (and accepts contract overrides on the input) and the PXE emulates kernel effects without
132
- * generating kernel witnesses. When false, simulateViaEntrypoint defers to the standard
133
- * simulation path via the real account entrypoint.
133
+ * Controls how the test wallet simulates transactions:
134
+ * - `kernelless`: Skips kernel circuits but uses the real account contract. Default.
135
+ * - `kernelless-override`: Skips kernels and replaces the account with a stub that doesn't do authwit validation.
136
+ * - `full`: Uses real kernel circuits and real account contracts. Slow!
134
137
  */
135
- private simulatedSimulations = false;
136
-
137
- enableSimulatedSimulations() {
138
- this.simulatedSimulations = true;
139
- }
138
+ private simulationMode: 'kernelless' | 'kernelless-override' | 'full' = 'kernelless';
140
139
 
141
- disableSimulatedSimulations() {
142
- this.simulatedSimulations = false;
140
+ setSimulationMode(mode: 'kernelless' | 'kernelless-override' | 'full') {
141
+ this.simulationMode = mode;
143
142
  }
144
143
 
145
144
  setMinFeePadding(value?: number) {
@@ -147,12 +146,7 @@ export class TestWallet extends BaseWallet {
147
146
  }
148
147
 
149
148
  protected getAccountFromAddress(address: AztecAddress): Promise<Account> {
150
- let account: Account | undefined;
151
- if (address.equals(AztecAddress.ZERO)) {
152
- account = new SignerlessAccount();
153
- } else {
154
- account = this.accounts.get(address?.toString() ?? '');
155
- }
149
+ const account = this.accounts.get(address?.toString() ?? '');
156
150
 
157
151
  if (!account) {
158
152
  throw new Error(`Account not found in wallet for address: ${address}`);
@@ -220,54 +214,55 @@ export class TestWallet extends BaseWallet {
220
214
  return account.createAuthWit(intentInnerHash, chainInfo);
221
215
  }
222
216
 
223
- /**
224
- * Override simulateViaEntrypoint to use fake accounts for kernelless simulation
225
- * when simulatedSimulations is enabled. Otherwise falls through to the real entrypoint path.
226
- */
227
217
  protected override async simulateViaEntrypoint(
228
218
  executionPayload: ExecutionPayload,
229
- from: AztecAddress,
230
- feeOptions: FeeOptions,
231
- scopes: AccessScopes,
232
- skipTxValidation?: boolean,
233
- skipFeeEnforcement?: boolean,
219
+ opts: SimulateViaEntrypointOptions,
234
220
  ): Promise<TxSimulationResult> {
235
- if (!this.simulatedSimulations) {
236
- return super.simulateViaEntrypoint(
237
- executionPayload,
238
- from,
239
- feeOptions,
240
- scopes,
241
- skipTxValidation,
242
- skipFeeEnforcement,
243
- );
244
- }
245
-
221
+ const { from, feeOptions, scopes, skipTxValidation, skipFeeEnforcement } = opts;
222
+ const skipKernels = this.simulationMode !== 'full';
246
223
  const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
247
- const executionOptions: DefaultAccountEntrypointOptions = {
248
- txNonce: Fr.random(),
249
- cancellable: this.cancellableTransactions,
250
- feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions,
251
- };
252
224
  const finalExecutionPayload = feeExecutionPayload
253
225
  ? mergeExecutionPayloads([feeExecutionPayload, executionPayload])
254
226
  : executionPayload;
255
- const { account: fromAccount, instance, artifact } = await this.getFakeAccountDataFor(from);
256
227
  const chainInfo = await this.getChainInfo();
257
- const txRequest = await fromAccount.createTxExecutionRequest(
258
- finalExecutionPayload,
259
- feeOptions.gasSettings,
260
- chainInfo,
261
- executionOptions,
262
- );
263
- const contractOverrides = {
264
- [from.toString()]: { instance, artifact },
265
- };
228
+
229
+ let overrides: SimulationOverrides | undefined;
230
+ let txRequest: TxExecutionRequest;
231
+ if (from === NO_FROM) {
232
+ const entrypoint = new DefaultEntrypoint();
233
+ txRequest = await entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
234
+ } else {
235
+ const useOverride = this.simulationMode === 'kernelless-override';
236
+ let fromAccount: Account;
237
+ if (useOverride) {
238
+ const { account, instance, artifact } = await this.getFakeAccountDataFor(from);
239
+ fromAccount = account;
240
+ overrides = {
241
+ contracts: { [from.toString()]: { instance, artifact } },
242
+ };
243
+ } else {
244
+ fromAccount = await this.getAccountFromAddress(from);
245
+ }
246
+ const executionOptions: DefaultAccountEntrypointOptions = {
247
+ txNonce: Fr.random(),
248
+ cancellable: this.cancellableTransactions,
249
+ // If from is an address, feeOptions include the way the account contract should handle the fee payment
250
+ feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions!,
251
+ };
252
+ txRequest = await fromAccount.createTxExecutionRequest(
253
+ finalExecutionPayload,
254
+ feeOptions.gasSettings,
255
+ chainInfo,
256
+ executionOptions,
257
+ );
258
+ }
259
+
266
260
  return this.pxe.simulateTx(txRequest, {
267
261
  simulatePublic: true,
268
- skipFeeEnforcement: true,
269
- skipTxValidation: true,
270
- overrides: { contracts: contractOverrides },
262
+ skipKernels,
263
+ skipFeeEnforcement,
264
+ skipTxValidation,
265
+ overrides,
271
266
  scopes,
272
267
  });
273
268
  }