@aztec/end-to-end 0.0.1-commit.c0b82b2 → 0.0.1-commit.c2eed6949

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 (104) 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 +17 -10
  4. package/dest/bench/utils.d.ts +1 -1
  5. package/dest/bench/utils.d.ts.map +1 -1
  6. package/dest/bench/utils.js +6 -3
  7. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts +1 -1
  8. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map +1 -1
  9. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js +14 -14
  10. package/dest/e2e_epochs/epochs_test.d.ts +3 -1
  11. package/dest/e2e_epochs/epochs_test.d.ts.map +1 -1
  12. package/dest/e2e_epochs/epochs_test.js +5 -2
  13. package/dest/e2e_fees/fees_test.d.ts +1 -1
  14. package/dest/e2e_fees/fees_test.d.ts.map +1 -1
  15. package/dest/e2e_fees/fees_test.js +12 -5
  16. package/dest/e2e_nested_contract/nested_contract_test.d.ts +1 -1
  17. package/dest/e2e_nested_contract/nested_contract_test.d.ts.map +1 -1
  18. package/dest/e2e_nested_contract/nested_contract_test.js +4 -6
  19. package/dest/e2e_p2p/p2p_network.d.ts +4 -3
  20. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  21. package/dest/e2e_p2p/p2p_network.js +19 -16
  22. package/dest/e2e_p2p/reqresp/utils.d.ts +1 -1
  23. package/dest/e2e_p2p/reqresp/utils.d.ts.map +1 -1
  24. package/dest/e2e_p2p/reqresp/utils.js +15 -2
  25. package/dest/e2e_p2p/shared.d.ts +21 -1
  26. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  27. package/dest/e2e_p2p/shared.js +33 -2
  28. package/dest/e2e_token_contract/token_contract_test.d.ts +1 -1
  29. package/dest/e2e_token_contract/token_contract_test.d.ts.map +1 -1
  30. package/dest/e2e_token_contract/token_contract_test.js +11 -11
  31. package/dest/fixtures/authwit_proxy.d.ts +3 -3
  32. package/dest/fixtures/authwit_proxy.d.ts.map +1 -1
  33. package/dest/fixtures/e2e_prover_test.d.ts +4 -3
  34. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  35. package/dest/fixtures/e2e_prover_test.js +9 -11
  36. package/dest/fixtures/get_bb_config.d.ts +1 -1
  37. package/dest/fixtures/get_bb_config.d.ts.map +1 -1
  38. package/dest/fixtures/get_bb_config.js +5 -5
  39. package/dest/fixtures/setup.d.ts +6 -1
  40. package/dest/fixtures/setup.d.ts.map +1 -1
  41. package/dest/fixtures/setup.js +16 -12
  42. package/dest/fixtures/token_utils.d.ts +1 -1
  43. package/dest/fixtures/token_utils.d.ts.map +1 -1
  44. package/dest/fixtures/token_utils.js +2 -2
  45. package/dest/shared/cross_chain_test_harness.d.ts +1 -1
  46. package/dest/shared/cross_chain_test_harness.d.ts.map +1 -1
  47. package/dest/shared/cross_chain_test_harness.js +13 -13
  48. package/dest/shared/gas_portal_test_harness.js +2 -2
  49. package/dest/shared/submit-transactions.d.ts +1 -1
  50. package/dest/shared/submit-transactions.d.ts.map +1 -1
  51. package/dest/shared/submit-transactions.js +1 -1
  52. package/dest/shared/uniswap_l1_l2.d.ts +1 -1
  53. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  54. package/dest/shared/uniswap_l1_l2.js +14 -17
  55. package/dest/simulators/lending_simulator.d.ts +1 -1
  56. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  57. package/dest/simulators/lending_simulator.js +4 -4
  58. package/dest/simulators/token_simulator.d.ts +1 -1
  59. package/dest/simulators/token_simulator.d.ts.map +1 -1
  60. package/dest/simulators/token_simulator.js +2 -2
  61. package/dest/spartan/setup_test_wallets.d.ts +4 -2
  62. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  63. package/dest/spartan/setup_test_wallets.js +28 -15
  64. package/dest/spartan/tx_metrics.js +1 -1
  65. package/dest/spartan/utils/config.d.ts +4 -1
  66. package/dest/spartan/utils/config.d.ts.map +1 -1
  67. package/dest/spartan/utils/config.js +1 -0
  68. package/dest/spartan/utils/index.d.ts +2 -1
  69. package/dest/spartan/utils/index.d.ts.map +1 -1
  70. package/dest/spartan/utils/index.js +2 -0
  71. package/dest/spartan/utils/pod_logs.d.ts +25 -0
  72. package/dest/spartan/utils/pod_logs.d.ts.map +1 -0
  73. package/dest/spartan/utils/pod_logs.js +74 -0
  74. package/dest/test-wallet/test_wallet.d.ts +10 -17
  75. package/dest/test-wallet/test_wallet.d.ts.map +1 -1
  76. package/dest/test-wallet/test_wallet.js +47 -48
  77. package/dest/test-wallet/worker_wallet_schema.d.ts +3 -3
  78. package/package.json +39 -40
  79. package/src/bench/client_flows/client_flows_benchmark.ts +39 -31
  80. package/src/bench/utils.ts +7 -2
  81. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +18 -14
  82. package/src/e2e_epochs/epochs_test.ts +14 -2
  83. package/src/e2e_fees/fees_test.ts +12 -5
  84. package/src/e2e_nested_contract/nested_contract_test.ts +6 -4
  85. package/src/e2e_p2p/p2p_network.ts +25 -21
  86. package/src/e2e_p2p/reqresp/utils.ts +23 -2
  87. package/src/e2e_p2p/shared.ts +55 -2
  88. package/src/e2e_token_contract/token_contract_test.ts +16 -8
  89. package/src/fixtures/e2e_prover_test.ts +16 -16
  90. package/src/fixtures/get_bb_config.ts +7 -6
  91. package/src/fixtures/setup.ts +21 -10
  92. package/src/fixtures/token_utils.ts +4 -2
  93. package/src/shared/cross_chain_test_harness.ts +13 -9
  94. package/src/shared/gas_portal_test_harness.ts +1 -1
  95. package/src/shared/submit-transactions.ts +1 -4
  96. package/src/shared/uniswap_l1_l2.ts +35 -28
  97. package/src/simulators/lending_simulator.ts +8 -4
  98. package/src/simulators/token_simulator.ts +6 -2
  99. package/src/spartan/setup_test_wallets.ts +51 -19
  100. package/src/spartan/tx_metrics.ts +1 -1
  101. package/src/spartan/utils/config.ts +1 -0
  102. package/src/spartan/utils/index.ts +3 -0
  103. package/src/spartan/utils/pod_logs.ts +99 -0
  104. 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';
@@ -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,20 +130,28 @@ 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;
140
142
  try {
141
- txHash = await deployMethod.send({
142
- from: AztecAddress.ZERO,
143
- fee: { paymentMethod },
143
+ let gasSettings;
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 },
144
152
  wait: NO_WAIT,
145
153
  });
154
+ txHash = deployResult.txHash;
146
155
  await waitForTx(aztecNode, txHash, { timeout: 2400 });
147
156
  logger.info(`${accountLabel} deployed at ${account.address}`);
148
157
  } catch (error) {
@@ -164,18 +173,29 @@ async function deployAccountWithDiagnostics(
164
173
  }
165
174
 
166
175
  async function deployAccountsInBatches(
167
- accounts: { getDeployMethod: () => Promise<{ send: (opts: any) => any }>; address: any }[],
176
+ accounts: {
177
+ getDeployMethod: () => Promise<{ simulate: (opts: any) => any; send: (opts: any) => any }>;
178
+ address: any;
179
+ }[],
168
180
  paymentMethod: SponsoredFeePaymentMethod,
169
181
  aztecNode: AztecNode,
170
182
  logger: Logger,
171
183
  labelPrefix: string,
172
184
  batchSize = 2,
185
+ estimateGas?: boolean,
173
186
  ): Promise<void> {
174
187
  for (let i = 0; i < accounts.length; i += batchSize) {
175
188
  const batch = accounts.slice(i, i + batchSize);
176
189
  await Promise.all(
177
190
  batch.map((account, idx) =>
178
- deployAccountWithDiagnostics(account, paymentMethod, aztecNode, logger, `${labelPrefix}${i + idx + 1}`),
191
+ deployAccountWithDiagnostics(
192
+ account,
193
+ paymentMethod,
194
+ aztecNode,
195
+ logger,
196
+ `${labelPrefix}${i + idx + 1}`,
197
+ estimateGas,
198
+ ),
179
199
  ),
180
200
  );
181
201
  }
@@ -186,6 +206,7 @@ export async function deploySponsoredTestAccounts(
186
206
  aztecNode: AztecNode,
187
207
  logger: Logger,
188
208
  numberOfFundedWallets = 1,
209
+ opts?: { estimateGas?: boolean },
189
210
  ): Promise<TestAccountsWithoutTokens> {
190
211
  const [recipient, ...funded] = await generateSchnorrAccounts(numberOfFundedWallets + 1);
191
212
  const recipientAccount = await wallet.createSchnorrAccount(recipient.secret, recipient.salt);
@@ -195,8 +216,23 @@ export async function deploySponsoredTestAccounts(
195
216
 
196
217
  const paymentMethod = new SponsoredFeePaymentMethod(await getSponsoredFPCAddress());
197
218
 
198
- await deployAccountWithDiagnostics(recipientAccount, paymentMethod, aztecNode, logger, 'Recipient account');
199
- await deployAccountsInBatches(fundedAccounts, paymentMethod, aztecNode, logger, 'Funded account ', 2);
219
+ await deployAccountWithDiagnostics(
220
+ recipientAccount,
221
+ paymentMethod,
222
+ aztecNode,
223
+ logger,
224
+ 'Recipient account',
225
+ opts?.estimateGas,
226
+ );
227
+ await deployAccountsInBatches(
228
+ fundedAccounts,
229
+ paymentMethod,
230
+ aztecNode,
231
+ logger,
232
+ 'Funded account ',
233
+ 2,
234
+ opts?.estimateGas,
235
+ );
200
236
 
201
237
  return {
202
238
  aztecNode,
@@ -234,7 +270,7 @@ export async function deployTestAccountsWithTokens(
234
270
  fundedAccounts.map(async (a, i) => {
235
271
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(a.address, claims[i]);
236
272
  const deployMethod = await a.getDeployMethod();
237
- await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod } });
273
+ await deployMethod.send({ from: NO_FROM, fee: { paymentMethod } });
238
274
  logger.info(`Account deployed at ${a.address}`);
239
275
  }),
240
276
  );
@@ -278,7 +314,7 @@ async function bridgeL1FeeJuice(
278
314
  const claim = await portal.bridgeTokensPublic(recipient, amount, true /* mint */);
279
315
 
280
316
  const isSynced = async () =>
281
- (await aztecNode.getL1ToL2MessageBlock(Fr.fromHexString(claim.messageHash))) !== undefined;
317
+ (await aztecNode.getL1ToL2MessageCheckpoint(Fr.fromHexString(claim.messageHash))) !== undefined;
282
318
  await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 0.5);
283
319
 
284
320
  log.info(`Created a claim for ${amount} L1 fee juice to ${recipient}.`, claim);
@@ -310,13 +346,9 @@ async function deployTokenAndMint(
310
346
  logger: Logger,
311
347
  ) {
312
348
  logger.verbose(`Deploying TokenContract...`);
313
- const { contract: tokenContract } = await TokenContract.deploy(
314
- wallet,
315
- admin,
316
- TOKEN_NAME,
317
- TOKEN_SYMBOL,
318
- TOKEN_DECIMALS,
319
- ).send({
349
+ const {
350
+ receipt: { contract: tokenContract },
351
+ } = await TokenContract.deploy(wallet, admin, TOKEN_NAME, TOKEN_SYMBOL, TOKEN_DECIMALS).send({
320
352
  from: admin,
321
353
  fee: {
322
354
  paymentMethod,
@@ -296,7 +296,7 @@ export class TxInclusionMetrics {
296
296
  value: stats.mean,
297
297
  },
298
298
  {
299
- name: `${group}/median_inclusion`,
299
+ name: `${group}/p50_inclusion`,
300
300
  unit: 's',
301
301
  value: stats.median,
302
302
  },
@@ -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
  }