@aztec/end-to-end 0.0.1-commit.4ad48494d → 0.0.1-commit.4d3c002

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 (152) 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 +18 -25
  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 +2 -2
  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 +16 -16
  10. package/dest/e2e_epochs/epochs_test.d.ts +11 -7
  11. package/dest/e2e_epochs/epochs_test.d.ts.map +1 -1
  12. package/dest/e2e_epochs/epochs_test.js +58 -36
  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 +13 -6
  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/inactivity_slash_test.js +1 -1
  20. package/dest/e2e_p2p/p2p_network.d.ts +4 -3
  21. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  22. package/dest/e2e_p2p/p2p_network.js +24 -7
  23. package/dest/e2e_p2p/reqresp/utils.d.ts +3 -3
  24. package/dest/e2e_p2p/reqresp/utils.d.ts.map +1 -1
  25. package/dest/e2e_p2p/reqresp/utils.js +45 -8
  26. package/dest/e2e_p2p/shared.d.ts +21 -1
  27. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  28. package/dest/e2e_p2p/shared.js +33 -2
  29. package/dest/e2e_token_contract/token_contract_test.d.ts +2 -2
  30. package/dest/e2e_token_contract/token_contract_test.d.ts.map +1 -1
  31. package/dest/e2e_token_contract/token_contract_test.js +13 -13
  32. package/dest/fixtures/authwit_proxy.d.ts +15 -0
  33. package/dest/fixtures/authwit_proxy.d.ts.map +1 -0
  34. package/dest/fixtures/authwit_proxy.js +30 -0
  35. package/dest/fixtures/e2e_prover_test.d.ts +6 -5
  36. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  37. package/dest/fixtures/e2e_prover_test.js +36 -48
  38. package/dest/fixtures/elu_monitor.d.ts +21 -0
  39. package/dest/fixtures/elu_monitor.d.ts.map +1 -0
  40. package/dest/fixtures/elu_monitor.js +102 -0
  41. package/dest/fixtures/fixtures.d.ts +5 -1
  42. package/dest/fixtures/fixtures.d.ts.map +1 -1
  43. package/dest/fixtures/fixtures.js +6 -0
  44. package/dest/fixtures/get_bb_config.d.ts +1 -1
  45. package/dest/fixtures/get_bb_config.d.ts.map +1 -1
  46. package/dest/fixtures/get_bb_config.js +5 -5
  47. package/dest/fixtures/setup.d.ts +30 -9
  48. package/dest/fixtures/setup.d.ts.map +1 -1
  49. package/dest/fixtures/setup.js +62 -89
  50. package/dest/fixtures/setup_p2p_test.d.ts +6 -3
  51. package/dest/fixtures/setup_p2p_test.d.ts.map +1 -1
  52. package/dest/fixtures/setup_p2p_test.js +12 -9
  53. package/dest/fixtures/token_utils.d.ts +2 -2
  54. package/dest/fixtures/token_utils.d.ts.map +1 -1
  55. package/dest/fixtures/token_utils.js +5 -7
  56. package/dest/shared/cross_chain_test_harness.d.ts +1 -1
  57. package/dest/shared/cross_chain_test_harness.d.ts.map +1 -1
  58. package/dest/shared/cross_chain_test_harness.js +13 -13
  59. package/dest/shared/gas_portal_test_harness.js +2 -2
  60. package/dest/shared/index.d.ts +2 -1
  61. package/dest/shared/index.d.ts.map +1 -1
  62. package/dest/shared/index.js +1 -0
  63. package/dest/shared/jest_setup.js +41 -1
  64. package/dest/shared/mock_state_view.d.ts +86 -0
  65. package/dest/shared/mock_state_view.d.ts.map +1 -0
  66. package/dest/shared/mock_state_view.js +186 -0
  67. package/dest/shared/submit-transactions.js +1 -1
  68. package/dest/shared/uniswap_l1_l2.d.ts +1 -1
  69. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  70. package/dest/shared/uniswap_l1_l2.js +14 -17
  71. package/dest/simulators/lending_simulator.d.ts +1 -1
  72. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  73. package/dest/simulators/lending_simulator.js +4 -4
  74. package/dest/simulators/token_simulator.d.ts +1 -1
  75. package/dest/simulators/token_simulator.d.ts.map +1 -1
  76. package/dest/simulators/token_simulator.js +2 -2
  77. package/dest/spartan/setup_test_wallets.d.ts +11 -2
  78. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  79. package/dest/spartan/setup_test_wallets.js +97 -38
  80. package/dest/spartan/tx_metrics.js +1 -1
  81. package/dest/spartan/utils/bot.d.ts +3 -2
  82. package/dest/spartan/utils/bot.d.ts.map +1 -1
  83. package/dest/spartan/utils/bot.js +2 -1
  84. package/dest/spartan/utils/config.d.ts +7 -1
  85. package/dest/spartan/utils/config.d.ts.map +1 -1
  86. package/dest/spartan/utils/config.js +3 -1
  87. package/dest/spartan/utils/index.d.ts +4 -2
  88. package/dest/spartan/utils/index.d.ts.map +1 -1
  89. package/dest/spartan/utils/index.js +5 -1
  90. package/dest/spartan/utils/k8s.d.ts +3 -1
  91. package/dest/spartan/utils/k8s.d.ts.map +1 -1
  92. package/dest/spartan/utils/k8s.js +6 -0
  93. package/dest/spartan/utils/nodes.d.ts +4 -5
  94. package/dest/spartan/utils/nodes.d.ts.map +1 -1
  95. package/dest/spartan/utils/nodes.js +9 -9
  96. package/dest/spartan/utils/pod_logs.d.ts +25 -0
  97. package/dest/spartan/utils/pod_logs.d.ts.map +1 -0
  98. package/dest/spartan/utils/pod_logs.js +74 -0
  99. package/dest/test-wallet/test_wallet.d.ts +11 -17
  100. package/dest/test-wallet/test_wallet.d.ts.map +1 -1
  101. package/dest/test-wallet/test_wallet.js +48 -49
  102. package/dest/test-wallet/wallet_worker_script.d.ts +2 -0
  103. package/dest/test-wallet/wallet_worker_script.d.ts.map +1 -0
  104. package/dest/test-wallet/wallet_worker_script.js +48 -0
  105. package/dest/test-wallet/worker_wallet.d.ts +52 -0
  106. package/dest/test-wallet/worker_wallet.d.ts.map +1 -0
  107. package/dest/test-wallet/worker_wallet.js +151 -0
  108. package/dest/test-wallet/worker_wallet_schema.d.ts +274 -0
  109. package/dest/test-wallet/worker_wallet_schema.d.ts.map +1 -0
  110. package/dest/test-wallet/worker_wallet_schema.js +10 -0
  111. package/package.json +42 -44
  112. package/src/bench/client_flows/client_flows_benchmark.ts +26 -9
  113. package/src/bench/utils.ts +7 -2
  114. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +20 -16
  115. package/src/e2e_epochs/epochs_test.ts +76 -64
  116. package/src/e2e_fees/fees_test.ts +13 -6
  117. package/src/e2e_nested_contract/nested_contract_test.ts +6 -4
  118. package/src/e2e_p2p/inactivity_slash_test.ts +1 -1
  119. package/src/e2e_p2p/p2p_network.ts +31 -5
  120. package/src/e2e_p2p/reqresp/utils.ts +57 -8
  121. package/src/e2e_p2p/shared.ts +55 -2
  122. package/src/e2e_token_contract/token_contract_test.ts +18 -10
  123. package/src/fixtures/authwit_proxy.ts +50 -0
  124. package/src/fixtures/dumps/epoch_proof_result.json +1 -1
  125. package/src/fixtures/e2e_prover_test.ts +39 -51
  126. package/src/fixtures/elu_monitor.ts +126 -0
  127. package/src/fixtures/fixtures.ts +10 -0
  128. package/src/fixtures/get_bb_config.ts +7 -6
  129. package/src/fixtures/setup.ts +85 -118
  130. package/src/fixtures/setup_p2p_test.ts +9 -17
  131. package/src/fixtures/token_utils.ts +3 -3
  132. package/src/shared/cross_chain_test_harness.ts +13 -9
  133. package/src/shared/gas_portal_test_harness.ts +1 -1
  134. package/src/shared/index.ts +1 -0
  135. package/src/shared/jest_setup.ts +51 -1
  136. package/src/shared/mock_state_view.ts +188 -0
  137. package/src/shared/submit-transactions.ts +1 -1
  138. package/src/shared/uniswap_l1_l2.ts +35 -28
  139. package/src/simulators/lending_simulator.ts +8 -4
  140. package/src/simulators/token_simulator.ts +6 -2
  141. package/src/spartan/setup_test_wallets.ts +138 -31
  142. package/src/spartan/tx_metrics.ts +1 -1
  143. package/src/spartan/utils/bot.ts +3 -0
  144. package/src/spartan/utils/config.ts +2 -0
  145. package/src/spartan/utils/index.ts +7 -0
  146. package/src/spartan/utils/k8s.ts +8 -0
  147. package/src/spartan/utils/nodes.ts +15 -10
  148. package/src/spartan/utils/pod_logs.ts +99 -0
  149. package/src/test-wallet/test_wallet.ts +62 -66
  150. package/src/test-wallet/wallet_worker_script.ts +60 -0
  151. package/src/test-wallet/worker_wallet.ts +213 -0
  152. package/src/test-wallet/worker_wallet_schema.ts +13 -0
@@ -0,0 +1,188 @@
1
+ import { EthAddress } from '@aztec/aztec.js/addresses';
2
+ import { type Logger, createLogger } from '@aztec/aztec.js/log';
3
+ import type { EthCheatCodes } from '@aztec/aztec/testing';
4
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
5
+
6
+ import { getContract } from 'viem';
7
+
8
+ /**
9
+ * Mock StateView contract for testing the Uniswap price oracle.
10
+ *
11
+ * Wraps a compiled Solidity contract that mimics the Uniswap V4 StateView's getSlot0 function.
12
+ * The mock allows setting return values dynamically for testing different price scenarios.
13
+ *
14
+ * Solidity source:
15
+ * ```solidity
16
+ * // SPDX-License-Identifier: Apache-2.0
17
+ * pragma solidity >=0.8.27;
18
+ *
19
+ * contract MockStateView {
20
+ * uint160 public sqrtPriceX96;
21
+ * int24 public tick;
22
+ * uint24 public protocolFee;
23
+ * uint24 public lpFee;
24
+ *
25
+ * function setReturnValues(
26
+ * uint160 _sqrtPriceX96,
27
+ * int24 _tick,
28
+ * uint24 _protocolFee,
29
+ * uint24 _lpFee
30
+ * ) external {
31
+ * sqrtPriceX96 = _sqrtPriceX96;
32
+ * tick = _tick;
33
+ * protocolFee = _protocolFee;
34
+ * lpFee = _lpFee;
35
+ * }
36
+ *
37
+ * function getSlot0(bytes32 poolId) external view returns (uint160, int24, uint24, uint24) {
38
+ * return (sqrtPriceX96, tick, protocolFee, lpFee);
39
+ * }
40
+ * }
41
+ * ```
42
+ */
43
+ export class MockStateView {
44
+ private static readonly BYTECODE: `0x${string}` =
45
+ '0x608060405234801561000f575f5ffd5b5060043610610060575f3560e01c80633eaf5d9f14610064578063704ce43e146100825780638db791d2146100a0578063b0e21e8a146100be578063b52e4bdd146100dc578063c815641c146100f8575b5f5ffd5b61006c61012b565b60405161007991906102ab565b60405180910390f35b61008a61013d565b60405161009791906102e1565b60405180910390f35b6100a8610151565b6040516100b59190610328565b60405180910390f35b6100c6610175565b6040516100d391906102e1565b60405180910390f35b6100f660048036038101906100f191906103c3565b610189565b005b610112600480360381019061010d919061045a565b61022b565b6040516101229493929190610485565b60405180910390f35b5f60149054906101000a900460020b81565b5f601a9054906101000a900462ffffff1681565b5f5f9054906101000a900473ffffffffffffffffffffffffffffffffffffffff1681565b5f60179054906101000a900462ffffff1681565b835f5f6101000a81548173ffffffffffffffffffffffffffffffffffffffff021916908373ffffffffffffffffffffffffffffffffffffffff160217905550825f60146101000a81548162ffffff021916908360020b62ffffff160217905550815f60176101000a81548162ffffff021916908362ffffff160217905550805f601a6101000a81548162ffffff021916908362ffffff16021790555050505050565b5f5f5f5f5f5f9054906101000a900473ffffffffffffffffffffffffffffffffffffffff165f60149054906101000a900460020b5f60179054906101000a900462ffffff165f601a9054906101000a900462ffffff1693509350935093509193509193565b5f8160020b9050919050565b6102a581610290565b82525050565b5f6020820190506102be5f83018461029c565b92915050565b5f62ffffff82169050919050565b6102db816102c4565b82525050565b5f6020820190506102f45f8301846102d2565b92915050565b5f73ffffffffffffffffffffffffffffffffffffffff82169050919050565b610322816102fa565b82525050565b5f60208201905061033b5f830184610319565b92915050565b5f5ffd5b61034e816102fa565b8114610358575f5ffd5b50565b5f8135905061036981610345565b92915050565b61037881610290565b8114610382575f5ffd5b50565b5f813590506103938161036f565b92915050565b6103a2816102c4565b81146103ac575f5ffd5b50565b5f813590506103bd81610399565b92915050565b5f5f5f5f608085870312156103db576103da610341565b5b5f6103e88782880161035b565b94505060206103f987828801610385565b935050604061040a878288016103af565b925050606061041b878288016103af565b91505092959194509250565b5f819050919050565b61043981610427565b8114610443575f5ffd5b50565b5f8135905061045481610430565b92915050565b5f6020828403121561046f5761046e610341565b5b5f61047c84828501610446565b91505092915050565b5f6080820190506104985f830187610319565b6104a5602083018661029c565b6104b260408301856102d2565b6104bf60608301846102d2565b9594505050505056fea2646970667358221220f8b1bfff284535bc078368ed34bd5e78981644845f3c9c1f5a4b8448c976805464736f6c634300081f0033';
46
+ private static readonly ABI = [
47
+ {
48
+ type: 'function',
49
+ name: 'setReturnValues',
50
+ inputs: [
51
+ { name: '_sqrtPriceX96', type: 'uint160' },
52
+ { name: '_tick', type: 'int24' },
53
+ { name: '_protocolFee', type: 'uint24' },
54
+ { name: '_lpFee', type: 'uint24' },
55
+ ],
56
+ outputs: [],
57
+ stateMutability: 'nonpayable',
58
+ },
59
+ {
60
+ inputs: [
61
+ {
62
+ internalType: 'bytes32',
63
+ name: 'poolId',
64
+ type: 'bytes32',
65
+ },
66
+ ],
67
+ name: 'getSlot0',
68
+ outputs: [
69
+ {
70
+ internalType: 'uint160',
71
+ name: '',
72
+ type: 'uint160',
73
+ },
74
+ {
75
+ internalType: 'int24',
76
+ name: '',
77
+ type: 'int24',
78
+ },
79
+ {
80
+ internalType: 'uint24',
81
+ name: '',
82
+ type: 'uint24',
83
+ },
84
+ {
85
+ internalType: 'uint24',
86
+ name: '',
87
+ type: 'uint24',
88
+ },
89
+ ],
90
+ stateMutability: 'view',
91
+ type: 'function',
92
+ },
93
+ ] as const;
94
+
95
+ private constructor(
96
+ private readonly address: EthAddress,
97
+ private readonly walletClient: ExtendedViemWalletClient,
98
+ private readonly log: Logger = createLogger('mock-state-view'),
99
+ ) {}
100
+
101
+ /**
102
+ * Deploys the mock StateView contract at the specified address using etch.
103
+ * @param ethCheatCodes - Cheat codes for etching bytecode
104
+ * @param walletClient - Wallet client for sending transactions
105
+ * @param address - Address to deploy the mock at (typically the real StateView address)
106
+ */
107
+ static async deploy(
108
+ ethCheatCodes: EthCheatCodes,
109
+ walletClient: ExtendedViemWalletClient,
110
+ address: EthAddress,
111
+ ): Promise<MockStateView> {
112
+ await ethCheatCodes.etch(address, MockStateView.BYTECODE);
113
+ return new MockStateView(address, walletClient);
114
+ }
115
+
116
+ /**
117
+ * Sets the price using the ethPerFeeAssetE12 format (same as rollup contract).
118
+ * Computes the corresponding sqrtPriceX96 internally.
119
+ *
120
+ * Math (from fee_asset_price_oracle.ts):
121
+ * ethPerFeeAssetE12 = 1e12 * 2^192 / sqrtPriceX96^2
122
+ *
123
+ * Inverted:
124
+ * sqrtPriceX96^2 = 1e12 * 2^192 / ethPerFeeAssetE12
125
+ * sqrtPriceX96 = sqrt(1e12 * 2^192 / ethPerFeeAssetE12)
126
+ *
127
+ * @param ethPerFeeAssetE12 - The price in ETH per fee asset, scaled by 1e12
128
+ */
129
+ async setEthPerFeeAsset(ethPerFeeAssetE12: bigint) {
130
+ const sqrtPriceX96 = this.ethPerFeeAssetE12ToSqrtPriceX96(ethPerFeeAssetE12);
131
+ return await this.setSqrtPriceX96(sqrtPriceX96);
132
+ }
133
+
134
+ /**
135
+ * Sets the sqrtPriceX96 value directly (Uniswap's price encoding).
136
+ * @param sqrtPriceX96 - The sqrtPriceX96 value
137
+ * @param tick - The tick value (default 10)
138
+ * @param protocolFee - The protocol fee (default 0)
139
+ * @param lpFee - The LP fee (default 500)
140
+ */
141
+ async setSqrtPriceX96(sqrtPriceX96: bigint, tick: number = 10, protocolFee: number = 0, lpFee: number = 500) {
142
+ const contract = getContract({
143
+ address: this.address.toString() as `0x${string}`,
144
+ abi: MockStateView.ABI,
145
+ client: this.walletClient,
146
+ });
147
+
148
+ const hash = await contract.write.setReturnValues([sqrtPriceX96, tick, protocolFee, lpFee]);
149
+ this.log.info(`Set sqrtPriceX96 to ${sqrtPriceX96}`);
150
+ return await this.walletClient.waitForTransactionReceipt({ hash });
151
+ }
152
+
153
+ /**
154
+ * Converts ethPerFeeAssetE12 to sqrtPriceX96 (inverse of sqrtPriceX96ToEthPerFeeAssetE12).
155
+ *
156
+ * Math:
157
+ * sqrtPriceX96 = sqrt(1e12 * 2^192 / ethPerFeeAssetE12)
158
+ */
159
+ ethPerFeeAssetE12ToSqrtPriceX96(ethPerFeeAssetE12: bigint): bigint {
160
+ if (ethPerFeeAssetE12 === 0n) {
161
+ throw new Error('Cannot convert zero ethPerFeeAssetE12');
162
+ }
163
+ const Q192 = 2n ** 192n;
164
+ const sqrtPriceSquared = (10n ** 12n * Q192) / ethPerFeeAssetE12;
165
+ return this.bigintSqrt(sqrtPriceSquared);
166
+ }
167
+
168
+ /** Integer square root using Newton's method */
169
+ bigintSqrt(n: bigint): bigint {
170
+ if (n < 0n) {
171
+ throw new Error('Cannot compute sqrt of negative number');
172
+ }
173
+ if (n === 0n) {
174
+ return 0n;
175
+ }
176
+ let x = n;
177
+ let y = (x + 1n) / 2n;
178
+ while (y < x) {
179
+ x = y;
180
+ y = (x + n / x) / 2n;
181
+ }
182
+ return x;
183
+ }
184
+ }
185
+
186
+ export function diffInBps(a: bigint, b: bigint): bigint {
187
+ return ((a - b) * 10000n) / b;
188
+ }
@@ -19,7 +19,7 @@ export const submitTxsTo = async (
19
19
  times(numTxs, async () => {
20
20
  const accountManager = await wallet.createSchnorrAccount(Fr.random(), Fr.random(), GrumpkinScalar.random());
21
21
  const deployMethod = await accountManager.getDeployMethod();
22
- const txHash = await deployMethod.send({ from: submitter, wait: NO_WAIT });
22
+ const { txHash } = await deployMethod.send({ from: submitter, wait: NO_WAIT });
23
23
 
24
24
  logger.info(`Tx sent with hash ${txHash}`);
25
25
  const receipt: TxReceipt = await wallet.getTxReceipt(txHash);
@@ -130,7 +130,9 @@ export const uniswapL1L2TestSuite = (
130
130
  client: l1Client,
131
131
  });
132
132
  // deploy l2 uniswap contract and attach to portal
133
- uniswapL2Contract = await UniswapContract.deploy(wallet, uniswapPortalAddress).send({ from: ownerAddress });
133
+ ({ contract: uniswapL2Contract } = await UniswapContract.deploy(wallet, uniswapPortalAddress).send({
134
+ from: ownerAddress,
135
+ }));
134
136
 
135
137
  const registryAddress = (await aztecNode.getNodeInfo()).l1ContractAddresses.registryAddress;
136
138
 
@@ -195,7 +197,7 @@ export const uniswapL1L2TestSuite = (
195
197
  logger.info('Withdrawing weth to L1 and sending message to swap to dai');
196
198
  const [secretForDepositingSwappedDai, secretHashForDepositingSwappedDai] = await generateClaimSecret();
197
199
 
198
- const l2UniswapInteractionReceipt = await uniswapL2Contract.methods
200
+ const { receipt: l2UniswapInteractionReceipt } = await uniswapL2Contract.methods
199
201
  .swap_private(
200
202
  wethCrossChainHarness.l2Token.address,
201
203
  wethCrossChainHarness.l2Bridge.address,
@@ -250,8 +252,12 @@ export const uniswapL1L2TestSuite = (
250
252
  await wethCrossChainHarness.expectPublicBalanceOnL2(uniswapL2Contract.address, 0n);
251
253
 
252
254
  // Since the outbox is only consumable when the epoch is proven, we need to advance to the next epoch.
253
- const block = await aztecNode.getBlock(l2UniswapInteractionReceipt.blockNumber!);
254
- const epoch = await rollup.getEpochNumberForCheckpoint(block!.checkpointNumber);
255
+ const swapResult = (await computeL2ToL1MembershipWitness(
256
+ aztecNode,
257
+ swapPrivateLeaf,
258
+ l2UniswapInteractionReceipt.txHash,
259
+ ))!;
260
+ const { epochNumber: epoch } = swapResult;
255
261
  await cheatCodes.rollup.advanceToEpoch(EpochNumber(epoch + 1));
256
262
  await waitForProven(aztecNode, l2UniswapInteractionReceipt, { provenTimeout: 300 });
257
263
 
@@ -260,14 +266,17 @@ export const uniswapL1L2TestSuite = (
260
266
  const daiL1BalanceOfPortalBeforeSwap = await daiCrossChainHarness.getL1BalanceOf(
261
267
  daiCrossChainHarness.tokenPortalAddress,
262
268
  );
263
- const swapResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, swapPrivateLeaf);
264
- const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, withdrawLeaf);
269
+ const withdrawResult = (await computeL2ToL1MembershipWitness(
270
+ aztecNode,
271
+ withdrawLeaf,
272
+ l2UniswapInteractionReceipt.txHash,
273
+ ))!;
265
274
 
266
- const swapPrivateL2MessageIndex = swapResult!.leafIndex;
267
- const swapPrivateSiblingPath = swapResult!.siblingPath;
275
+ const swapPrivateL2MessageIndex = swapResult.leafIndex;
276
+ const swapPrivateSiblingPath = swapResult.siblingPath;
268
277
 
269
- const withdrawL2MessageIndex = withdrawResult!.leafIndex;
270
- const withdrawSiblingPath = withdrawResult!.siblingPath;
278
+ const withdrawL2MessageIndex = withdrawResult.leafIndex;
279
+ const withdrawSiblingPath = withdrawResult.siblingPath;
271
280
 
272
281
  const withdrawMessageMetadata = {
273
282
  _epoch: BigInt(epoch),
@@ -787,7 +796,7 @@ export const uniswapL1L2TestSuite = (
787
796
  logger.info('Withdrawing weth to L1 and sending message to swap to dai');
788
797
 
789
798
  const [, secretHashForDepositingSwappedDai] = await generateClaimSecret();
790
- const withdrawReceipt = await uniswapL2Contract.methods
799
+ const { receipt: withdrawReceipt } = await uniswapL2Contract.methods
791
800
  .swap_private(
792
801
  wethCrossChainHarness.l2Token.address,
793
802
  wethCrossChainHarness.l2Bridge.address,
@@ -838,16 +847,15 @@ export const uniswapL1L2TestSuite = (
838
847
  chainId: new Fr(l1Client.chain.id),
839
848
  });
840
849
 
841
- const block = await aztecNode.getBlock(withdrawReceipt.blockNumber!);
842
- const epoch = await rollup.getEpochNumberForCheckpoint(block!.checkpointNumber);
843
- const swapResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, swapPrivateLeaf);
844
- const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, withdrawLeaf);
850
+ const swapResult = (await computeL2ToL1MembershipWitness(aztecNode, swapPrivateLeaf, withdrawReceipt.txHash))!;
851
+ const { epochNumber: epoch } = swapResult;
852
+ const withdrawResult = (await computeL2ToL1MembershipWitness(aztecNode, withdrawLeaf, withdrawReceipt.txHash))!;
845
853
 
846
- const swapPrivateL2MessageIndex = swapResult!.leafIndex;
847
- const swapPrivateSiblingPath = swapResult!.siblingPath;
854
+ const swapPrivateL2MessageIndex = swapResult.leafIndex;
855
+ const swapPrivateSiblingPath = swapResult.siblingPath;
848
856
 
849
- const withdrawL2MessageIndex = withdrawResult!.leafIndex;
850
- const withdrawSiblingPath = withdrawResult!.siblingPath;
857
+ const withdrawL2MessageIndex = withdrawResult.leafIndex;
858
+ const withdrawSiblingPath = withdrawResult.siblingPath;
851
859
 
852
860
  const withdrawMessageMetadata = {
853
861
  _epoch: BigInt(epoch),
@@ -915,7 +923,7 @@ export const uniswapL1L2TestSuite = (
915
923
 
916
924
  // Call swap_public on L2
917
925
  const secretHashForDepositingSwappedDai = Fr.random();
918
- const withdrawReceipt = await uniswapL2Contract.methods
926
+ const { receipt: withdrawReceipt } = await uniswapL2Contract.methods
919
927
  .swap_public(
920
928
  ownerAddress,
921
929
  wethCrossChainHarness.l2Bridge.address,
@@ -971,16 +979,15 @@ export const uniswapL1L2TestSuite = (
971
979
  chainId: new Fr(l1Client.chain.id),
972
980
  });
973
981
 
974
- const block = await aztecNode.getBlock(withdrawReceipt.blockNumber!);
975
- const epoch = await rollup.getEpochNumberForCheckpoint(block!.checkpointNumber);
976
- const swapResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, swapPublicLeaf);
977
- const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, epoch, withdrawLeaf);
982
+ const swapResult = (await computeL2ToL1MembershipWitness(aztecNode, swapPublicLeaf, withdrawReceipt.txHash))!;
983
+ const { epochNumber: epoch } = swapResult;
984
+ const withdrawResult = (await computeL2ToL1MembershipWitness(aztecNode, withdrawLeaf, withdrawReceipt.txHash))!;
978
985
 
979
- const swapPublicL2MessageIndex = swapResult!.leafIndex;
980
- const swapPublicSiblingPath = swapResult!.siblingPath;
986
+ const swapPublicL2MessageIndex = swapResult.leafIndex;
987
+ const swapPublicSiblingPath = swapResult.siblingPath;
981
988
 
982
- const withdrawL2MessageIndex = withdrawResult!.leafIndex;
983
- const withdrawSiblingPath = withdrawResult!.siblingPath;
989
+ const withdrawL2MessageIndex = withdrawResult.leafIndex;
990
+ const withdrawSiblingPath = withdrawResult.siblingPath;
984
991
 
985
992
  const withdrawMessageMetadata = {
986
993
  _epoch: BigInt(epoch),
@@ -94,7 +94,9 @@ export class LendingSimulator {
94
94
 
95
95
  async prepare() {
96
96
  this.accumulator = BASE;
97
- const slot = await this.rollup.getSlotAt(BigInt(await this.cc.eth.timestamp()) + BigInt(this.ethereumSlotDuration));
97
+ const slot = await this.rollup.getSlotAt(
98
+ BigInt(await this.cc.eth.lastBlockTimestamp()) + BigInt(this.ethereumSlotDuration),
99
+ );
98
100
  this.time = Number(await this.rollup.getTimestampForSlot(slot));
99
101
  }
100
102
 
@@ -103,7 +105,7 @@ export class LendingSimulator {
103
105
  return;
104
106
  }
105
107
 
106
- const slot = await this.rollup.getSlotAt(BigInt(await this.cc.eth.timestamp()));
108
+ const slot = await this.rollup.getSlotAt(BigInt(await this.cc.eth.lastBlockTimestamp()));
107
109
  const targetSlot = SlotNumber(slot + diff);
108
110
  const ts = Number(await this.rollup.getTimestampForSlot(targetSlot));
109
111
  const timeDiff = ts - this.time;
@@ -186,14 +188,16 @@ export class LendingSimulator {
186
188
 
187
189
  expect(this.borrowed).toEqual(this.stableCoin.totalSupply - this.mintedOutside);
188
190
 
189
- const asset = await this.lendingContract.methods.get_asset(0).simulate({ from: this.account.address });
191
+ const { result: asset } = await this.lendingContract.methods.get_asset(0).simulate({ from: this.account.address });
190
192
 
191
193
  const interestAccumulator = asset['interest_accumulator'];
192
194
  expect(interestAccumulator).toEqual(this.accumulator);
193
195
  expect(asset['last_updated_ts']).toEqual(BigInt(this.time));
194
196
 
195
197
  for (const key of [this.account.address, AztecAddress.fromField(await this.account.key())]) {
196
- const privatePos = await this.lendingContract.methods.get_position(key).simulate({ from: this.account.address });
198
+ const { result: privatePos } = await this.lendingContract.methods
199
+ .get_position(key)
200
+ .simulate({ from: this.account.address });
197
201
  expect(new Fr(privatePos['collateral'])).toEqual(this.collateral[key.toString()] ?? Fr.ZERO);
198
202
  expect(new Fr(privatePos['static_debt'])).toEqual(this.staticDebt[key.toString()] ?? Fr.ZERO);
199
203
  expect(privatePos['debt']).toEqual(
@@ -109,7 +109,9 @@ export class TokenSimulator {
109
109
  await Promise.all(
110
110
  chunk(calls, 5).map(batch => new BatchCall(this.defaultWallet, batch).simulate({ from: this.defaultAddress })),
111
111
  )
112
- ).flat();
112
+ )
113
+ .flatMap(r => r.result)
114
+ .map(r => r.result);
113
115
  expect(results[0]).toEqual(this.totalSupply);
114
116
 
115
117
  // Check that all our balances match
@@ -123,7 +125,9 @@ export class TokenSimulator {
123
125
  const wallet = this.lookupProvider.get(address.toString());
124
126
  const asset = wallet ? this.token.withWallet(wallet) : this.token;
125
127
 
126
- const actualPrivateBalance = await asset.methods.balance_of_private(address).simulate({ from: address });
128
+ const { result: actualPrivateBalance } = await asset.methods
129
+ .balance_of_private(address)
130
+ .simulate({ from: address });
127
131
  expect(actualPrivateBalance).toEqual(this.balanceOfPrivate(address));
128
132
  }
129
133
  }
@@ -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';
@@ -20,6 +21,7 @@ import { getBBConfig } from '../fixtures/get_bb_config.js';
20
21
  import { getSponsoredFPCAddress, registerSponsoredFPC } from '../fixtures/utils.js';
21
22
  import { TestWallet } from '../test-wallet/test_wallet.js';
22
23
  import { proveInteraction } from '../test-wallet/utils.js';
24
+ import { WorkerWallet } from '../test-wallet/worker_wallet.js';
23
25
 
24
26
  export interface TestAccounts {
25
27
  aztecNode: AztecNode;
@@ -87,11 +89,19 @@ export async function deploySponsoredTestAccountsWithTokens(
87
89
 
88
90
  const paymentMethod = new SponsoredFeePaymentMethod(await getSponsoredFPCAddress());
89
91
  const recipientDeployMethod = await recipientAccount.getDeployMethod();
90
- await recipientDeployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod }, wait: { timeout: 2400 } });
92
+ await recipientDeployMethod.send({
93
+ from: NO_FROM,
94
+ fee: { paymentMethod },
95
+ wait: { timeout: 2400 },
96
+ });
91
97
  await Promise.all(
92
98
  fundedAccounts.map(async a => {
93
99
  const deployMethod = await a.getDeployMethod();
94
- await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod }, wait: { timeout: 2400 } }); // increase timeout on purpose in order to account for two empty epochs
100
+ await deployMethod.send({
101
+ from: NO_FROM,
102
+ fee: { paymentMethod },
103
+ wait: { timeout: 2400 },
104
+ }); // increase timeout on purpose in order to account for two empty epochs
95
105
  logger.info(`Account deployed at ${a.address}`);
96
106
  }),
97
107
  );
@@ -120,49 +130,91 @@ export async function deploySponsoredTestAccountsWithTokens(
120
130
  }
121
131
 
122
132
  async function deployAccountWithDiagnostics(
123
- account: { getDeployMethod: () => Promise<{ send: (opts: any) => any }>; address: any },
133
+ account: { getDeployMethod: () => Promise<{ simulate: (opts: any) => any; send: (opts: any) => any }>; address: any },
124
134
  paymentMethod: SponsoredFeePaymentMethod,
125
135
  aztecNode: AztecNode,
126
136
  logger: Logger,
127
137
  accountLabel: string,
138
+ estimateGas?: boolean,
128
139
  ): Promise<void> {
129
140
  const deployMethod = await account.getDeployMethod();
130
- let txHash;
131
- try {
132
- txHash = await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod }, wait: NO_WAIT });
133
- await waitForTx(aztecNode, txHash, { timeout: 2400 });
134
- logger.info(`${accountLabel} deployed at ${account.address}`);
135
- } catch (error) {
136
- const blockNumber = await aztecNode.getBlockNumber();
137
- let receipt;
138
- try {
139
- receipt = await aztecNode.getTxReceipt(txHash);
140
- } catch {
141
- receipt = 'unavailable';
142
- }
143
- logger.error(`${accountLabel} deployment failed`, {
144
- txHash: txHash.toString(),
145
- receipt: JSON.stringify(receipt),
146
- currentBlockNumber: blockNumber,
147
- error: String(error),
148
- });
149
- throw error;
141
+ let gasSettings: any;
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}`);
150
146
  }
147
+
148
+ // Track the tx hash across retries so we don't re-send when the previous tx is still pending.
149
+ let sentTxHash: { txHash: any } | undefined;
150
+
151
+ await retry(
152
+ async () => {
153
+ // Check if already deployed (handles case where previous attempt succeeded but waitForTx timed out)
154
+ const existing = await aztecNode.getContract(account.address);
155
+ if (existing) {
156
+ logger.info(`${accountLabel} already deployed at ${account.address}, skipping`);
157
+ return;
158
+ }
159
+
160
+ // If we already sent a tx, check if it was dropped before deciding to re-send.
161
+ if (sentTxHash) {
162
+ const prevReceipt = await aztecNode.getTxReceipt(sentTxHash.txHash);
163
+ if (prevReceipt.isDropped()) {
164
+ logger.info(`${accountLabel} previous tx ${sentTxHash.txHash} was dropped, re-sending`);
165
+ sentTxHash = undefined;
166
+ } else {
167
+ logger.info(`${accountLabel} previous tx ${sentTxHash.txHash} still pending, waiting again...`);
168
+ }
169
+ }
170
+
171
+ if (!sentTxHash) {
172
+ const deployResult = await deployMethod.send({
173
+ from: NO_FROM,
174
+ fee: { paymentMethod, gasSettings },
175
+ wait: NO_WAIT,
176
+ });
177
+ sentTxHash = { txHash: deployResult.txHash };
178
+ logger.info(`${accountLabel} tx sent`, { txHash: sentTxHash.txHash.toString() });
179
+ }
180
+
181
+ const receipt = await waitForTx(aztecNode, sentTxHash.txHash, { timeout: 600 });
182
+ if (receipt.isDropped()) {
183
+ sentTxHash = undefined;
184
+ throw new Error(`${accountLabel} tx was dropped, retrying...`);
185
+ }
186
+ logger.info(`${accountLabel} deployed at ${account.address}`);
187
+ },
188
+ `deploy ${accountLabel}`,
189
+ makeBackoff([1, 2, 4, 8, 16]),
190
+ logger,
191
+ );
151
192
  }
152
193
 
153
194
  async function deployAccountsInBatches(
154
- accounts: { getDeployMethod: () => Promise<{ send: (opts: any) => any }>; address: any }[],
195
+ accounts: {
196
+ getDeployMethod: () => Promise<{ simulate: (opts: any) => any; send: (opts: any) => any }>;
197
+ address: any;
198
+ }[],
155
199
  paymentMethod: SponsoredFeePaymentMethod,
156
200
  aztecNode: AztecNode,
157
201
  logger: Logger,
158
202
  labelPrefix: string,
159
203
  batchSize = 2,
204
+ estimateGas?: boolean,
160
205
  ): Promise<void> {
161
206
  for (let i = 0; i < accounts.length; i += batchSize) {
162
207
  const batch = accounts.slice(i, i + batchSize);
163
208
  await Promise.all(
164
209
  batch.map((account, idx) =>
165
- deployAccountWithDiagnostics(account, paymentMethod, aztecNode, logger, `${labelPrefix}${i + idx + 1}`),
210
+ deployAccountWithDiagnostics(
211
+ account,
212
+ paymentMethod,
213
+ aztecNode,
214
+ logger,
215
+ `${labelPrefix}${i + idx + 1}`,
216
+ estimateGas,
217
+ ),
166
218
  ),
167
219
  );
168
220
  }
@@ -173,6 +225,7 @@ export async function deploySponsoredTestAccounts(
173
225
  aztecNode: AztecNode,
174
226
  logger: Logger,
175
227
  numberOfFundedWallets = 1,
228
+ opts?: { estimateGas?: boolean },
176
229
  ): Promise<TestAccountsWithoutTokens> {
177
230
  const [recipient, ...funded] = await generateSchnorrAccounts(numberOfFundedWallets + 1);
178
231
  const recipientAccount = await wallet.createSchnorrAccount(recipient.secret, recipient.salt);
@@ -182,8 +235,23 @@ export async function deploySponsoredTestAccounts(
182
235
 
183
236
  const paymentMethod = new SponsoredFeePaymentMethod(await getSponsoredFPCAddress());
184
237
 
185
- await deployAccountWithDiagnostics(recipientAccount, paymentMethod, aztecNode, logger, 'Recipient account');
186
- await deployAccountsInBatches(fundedAccounts, paymentMethod, aztecNode, logger, 'Funded account ', 2);
238
+ await deployAccountWithDiagnostics(
239
+ recipientAccount,
240
+ paymentMethod,
241
+ aztecNode,
242
+ logger,
243
+ 'Recipient account',
244
+ opts?.estimateGas,
245
+ );
246
+ await deployAccountsInBatches(
247
+ fundedAccounts,
248
+ paymentMethod,
249
+ aztecNode,
250
+ logger,
251
+ 'Funded account ',
252
+ 2,
253
+ opts?.estimateGas,
254
+ );
187
255
 
188
256
  return {
189
257
  aztecNode,
@@ -221,7 +289,7 @@ export async function deployTestAccountsWithTokens(
221
289
  fundedAccounts.map(async (a, i) => {
222
290
  const paymentMethod = new FeeJuicePaymentMethodWithClaim(a.address, claims[i]);
223
291
  const deployMethod = await a.getDeployMethod();
224
- await deployMethod.send({ from: AztecAddress.ZERO, fee: { paymentMethod } });
292
+ await deployMethod.send({ from: NO_FROM, fee: { paymentMethod } });
225
293
  logger.info(`Account deployed at ${a.address}`);
226
294
  }),
227
295
  );
@@ -265,7 +333,7 @@ async function bridgeL1FeeJuice(
265
333
  const claim = await portal.bridgeTokensPublic(recipient, amount, true /* mint */);
266
334
 
267
335
  const isSynced = async () =>
268
- (await aztecNode.getL1ToL2MessageBlock(Fr.fromHexString(claim.messageHash))) !== undefined;
336
+ (await aztecNode.getL1ToL2MessageCheckpoint(Fr.fromHexString(claim.messageHash))) !== undefined;
269
337
  await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 0.5);
270
338
 
271
339
  log.info(`Created a claim for ${amount} L1 fee juice to ${recipient}.`, claim);
@@ -308,7 +376,7 @@ async function deployTokenAndMint(
308
376
  fee: {
309
377
  paymentMethod,
310
378
  },
311
- wait: { timeout: 600, returnReceipt: true },
379
+ wait: { timeout: 600 },
312
380
  });
313
381
 
314
382
  const tokenAddress = tokenContract.address;
@@ -397,3 +465,42 @@ export async function createWalletAndAztecNodeClient(
397
465
  },
398
466
  };
399
467
  }
468
+
469
+ export type WorkerWalletWrapper = {
470
+ wallet: WorkerWallet;
471
+ aztecNode: AztecNode;
472
+ cleanup: () => Promise<void>;
473
+ };
474
+
475
+ export async function createWorkerWalletClient(
476
+ nodeUrl: string,
477
+ proverEnabled: boolean,
478
+ logger: Logger,
479
+ ): Promise<WorkerWalletWrapper> {
480
+ const aztecNode = createAztecNodeClient(nodeUrl);
481
+ const [bbConfig, acvmConfig] = await Promise.all([getBBConfig(logger), getACVMConfig(logger)]);
482
+
483
+ // Strip cleanup functions — they can't be structured-cloned for worker transfer
484
+ const { cleanup: bbCleanup, ...bbPaths } = bbConfig ?? {};
485
+ const { cleanup: acvmCleanup, ...acvmPaths } = acvmConfig ?? {};
486
+
487
+ const pxeConfig = {
488
+ dataDirectory: undefined,
489
+ dataStoreMapSizeKb: 1024 * 1024,
490
+ ...bbPaths,
491
+ ...acvmPaths,
492
+ proverEnabled,
493
+ };
494
+
495
+ const wallet = await WorkerWallet.create(nodeUrl, pxeConfig);
496
+
497
+ return {
498
+ wallet,
499
+ aztecNode,
500
+ async cleanup() {
501
+ await wallet.stop();
502
+ await bbCleanup?.();
503
+ await acvmCleanup?.();
504
+ },
505
+ };
506
+ }
@@ -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
  },
@@ -35,6 +35,7 @@ export async function installTransferBot({
35
35
  replicas = 1,
36
36
  txIntervalSeconds = 10,
37
37
  followChain = 'CHECKPOINTED',
38
+ pxeSyncChainTip = 'proposed',
38
39
  mnemonic = process.env.LABS_INFRA_MNEMONIC ?? 'test test test test test test test test test test test junk',
39
40
  mnemonicStartIndex,
40
41
  botPrivateKey = process.env.BOT_TRANSFERS_L2_PRIVATE_KEY ?? '0xcafe01',
@@ -49,6 +50,7 @@ export async function installTransferBot({
49
50
  replicas?: number;
50
51
  txIntervalSeconds?: number;
51
52
  followChain?: string;
53
+ pxeSyncChainTip?: string;
52
54
  mnemonic?: string;
53
55
  mnemonicStartIndex?: number | string;
54
56
  botPrivateKey?: string;
@@ -67,6 +69,7 @@ export async function installTransferBot({
67
69
  'bot.replicaCount': replicas,
68
70
  'bot.txIntervalSeconds': txIntervalSeconds,
69
71
  'bot.followChain': followChain,
72
+ 'bot.pxeSyncChainTip': pxeSyncChainTip,
70
73
  'bot.botPrivateKey': botPrivateKey,
71
74
  'bot.nodeUrl': resolvedNodeUrl,
72
75
  'bot.mnemonic': mnemonic,