@aztec/end-to-end 0.0.1-commit.6d3c34e → 0.0.1-commit.7ac86ea28

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 (226) hide show
  1. package/dest/bench/client_flows/benchmark.d.ts +3 -2
  2. package/dest/bench/client_flows/benchmark.d.ts.map +1 -1
  3. package/dest/bench/client_flows/benchmark.js +21 -1
  4. package/dest/bench/client_flows/client_flows_benchmark.d.ts +13 -14
  5. package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
  6. package/dest/bench/client_flows/client_flows_benchmark.js +120 -135
  7. package/dest/bench/client_flows/config.d.ts +2 -2
  8. package/dest/bench/client_flows/config.d.ts.map +1 -1
  9. package/dest/bench/client_flows/config.js +18 -0
  10. package/dest/bench/utils.d.ts +6 -5
  11. package/dest/bench/utils.d.ts.map +1 -1
  12. package/dest/bench/utils.js +9 -7
  13. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts +9 -8
  14. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map +1 -1
  15. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js +106 -112
  16. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.d.ts +7 -8
  17. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.d.ts.map +1 -1
  18. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.js +68 -64
  19. package/dest/e2e_deploy_contract/deploy_test.d.ts +5 -4
  20. package/dest/e2e_deploy_contract/deploy_test.d.ts.map +1 -1
  21. package/dest/e2e_deploy_contract/deploy_test.js +18 -13
  22. package/dest/e2e_epochs/epochs_test.d.ts +15 -7
  23. package/dest/e2e_epochs/epochs_test.d.ts.map +1 -1
  24. package/dest/e2e_epochs/epochs_test.js +86 -39
  25. package/dest/e2e_fees/bridging_race.notest.js +3 -5
  26. package/dest/e2e_fees/fees_test.d.ts +14 -14
  27. package/dest/e2e_fees/fees_test.d.ts.map +1 -1
  28. package/dest/e2e_fees/fees_test.js +124 -142
  29. package/dest/e2e_l1_publisher/write_json.d.ts +4 -3
  30. package/dest/e2e_l1_publisher/write_json.d.ts.map +1 -1
  31. package/dest/e2e_l1_publisher/write_json.js +14 -13
  32. package/dest/e2e_nested_contract/nested_contract_test.d.ts +6 -9
  33. package/dest/e2e_nested_contract/nested_contract_test.d.ts.map +1 -1
  34. package/dest/e2e_nested_contract/nested_contract_test.js +31 -39
  35. package/dest/e2e_p2p/inactivity_slash_test.d.ts +3 -3
  36. package/dest/e2e_p2p/inactivity_slash_test.d.ts.map +1 -1
  37. package/dest/e2e_p2p/inactivity_slash_test.js +3 -3
  38. package/dest/e2e_p2p/p2p_network.d.ts +8 -7
  39. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  40. package/dest/e2e_p2p/p2p_network.js +126 -105
  41. package/dest/e2e_p2p/reqresp/utils.d.ts +22 -0
  42. package/dest/e2e_p2p/reqresp/utils.d.ts.map +1 -0
  43. package/dest/e2e_p2p/reqresp/utils.js +177 -0
  44. package/dest/e2e_p2p/shared.d.ts +6 -6
  45. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  46. package/dest/e2e_p2p/shared.js +14 -17
  47. package/dest/e2e_storage_proof/fixtures/storage_proof_fetcher.d.ts +2 -0
  48. package/dest/e2e_storage_proof/fixtures/storage_proof_fetcher.d.ts.map +1 -0
  49. package/dest/e2e_storage_proof/fixtures/storage_proof_fetcher.js +184 -0
  50. package/dest/e2e_storage_proof/fixtures/storage_proof_fixture.d.ts +18 -0
  51. package/dest/e2e_storage_proof/fixtures/storage_proof_fixture.d.ts.map +1 -0
  52. package/dest/e2e_storage_proof/fixtures/storage_proof_fixture.js +120 -0
  53. package/dest/e2e_token_contract/token_contract_test.d.ts +19 -10
  54. package/dest/e2e_token_contract/token_contract_test.d.ts.map +1 -1
  55. package/dest/e2e_token_contract/token_contract_test.js +99 -91
  56. package/dest/fixtures/authwit_proxy.d.ts +15 -0
  57. package/dest/fixtures/authwit_proxy.d.ts.map +1 -0
  58. package/dest/fixtures/authwit_proxy.js +30 -0
  59. package/dest/fixtures/e2e_prover_test.d.ts +11 -17
  60. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  61. package/dest/fixtures/e2e_prover_test.js +112 -127
  62. package/dest/fixtures/fixtures.d.ts +5 -1
  63. package/dest/fixtures/fixtures.d.ts.map +1 -1
  64. package/dest/fixtures/fixtures.js +6 -0
  65. package/dest/fixtures/ha_setup.d.ts +71 -0
  66. package/dest/fixtures/ha_setup.d.ts.map +1 -0
  67. package/dest/fixtures/ha_setup.js +116 -0
  68. package/dest/fixtures/index.d.ts +2 -1
  69. package/dest/fixtures/index.d.ts.map +1 -1
  70. package/dest/fixtures/index.js +1 -0
  71. package/dest/fixtures/setup.d.ts +235 -0
  72. package/dest/fixtures/setup.d.ts.map +1 -0
  73. package/dest/fixtures/setup.js +605 -0
  74. package/dest/fixtures/setup_p2p_test.d.ts +16 -9
  75. package/dest/fixtures/setup_p2p_test.d.ts.map +1 -1
  76. package/dest/fixtures/setup_p2p_test.js +40 -29
  77. package/dest/fixtures/token_utils.d.ts +2 -2
  78. package/dest/fixtures/token_utils.d.ts.map +1 -1
  79. package/dest/fixtures/token_utils.js +10 -6
  80. package/dest/fixtures/utils.d.ts +5 -191
  81. package/dest/fixtures/utils.d.ts.map +1 -1
  82. package/dest/fixtures/utils.js +4 -615
  83. package/dest/quality_of_service/prometheus_client.d.ts +38 -0
  84. package/dest/quality_of_service/prometheus_client.d.ts.map +1 -0
  85. package/dest/quality_of_service/prometheus_client.js +67 -0
  86. package/dest/shared/cross_chain_test_harness.d.ts +14 -4
  87. package/dest/shared/cross_chain_test_harness.d.ts.map +1 -1
  88. package/dest/shared/cross_chain_test_harness.js +11 -11
  89. package/dest/shared/gas_portal_test_harness.d.ts +11 -1
  90. package/dest/shared/gas_portal_test_harness.d.ts.map +1 -1
  91. package/dest/shared/gas_portal_test_harness.js +1 -1
  92. package/dest/shared/index.d.ts +2 -1
  93. package/dest/shared/index.d.ts.map +1 -1
  94. package/dest/shared/index.js +1 -0
  95. package/dest/shared/mock_state_view.d.ts +86 -0
  96. package/dest/shared/mock_state_view.d.ts.map +1 -0
  97. package/dest/shared/mock_state_view.js +186 -0
  98. package/dest/shared/submit-transactions.d.ts +4 -4
  99. package/dest/shared/submit-transactions.d.ts.map +1 -1
  100. package/dest/shared/submit-transactions.js +9 -11
  101. package/dest/shared/uniswap_l1_l2.d.ts +1 -1
  102. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  103. package/dest/shared/uniswap_l1_l2.js +20 -18
  104. package/dest/simulators/lending_simulator.d.ts +5 -1
  105. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  106. package/dest/simulators/lending_simulator.js +2 -2
  107. package/dest/simulators/token_simulator.d.ts +1 -1
  108. package/dest/simulators/token_simulator.d.ts.map +1 -1
  109. package/dest/simulators/token_simulator.js +2 -23
  110. package/dest/spartan/setup_test_wallets.d.ts +9 -2
  111. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  112. package/dest/spartan/setup_test_wallets.js +92 -35
  113. package/dest/spartan/tx_metrics.d.ts +51 -2
  114. package/dest/spartan/tx_metrics.d.ts.map +1 -1
  115. package/dest/spartan/tx_metrics.js +333 -6
  116. package/dest/spartan/utils/bot.d.ts +28 -0
  117. package/dest/spartan/utils/bot.d.ts.map +1 -0
  118. package/dest/spartan/utils/bot.js +142 -0
  119. package/dest/spartan/utils/chaos.d.ts +79 -0
  120. package/dest/spartan/utils/chaos.d.ts.map +1 -0
  121. package/dest/spartan/utils/chaos.js +142 -0
  122. package/dest/spartan/utils/clients.d.ts +39 -0
  123. package/dest/spartan/utils/clients.d.ts.map +1 -0
  124. package/dest/spartan/utils/clients.js +90 -0
  125. package/dest/spartan/utils/config.d.ts +39 -0
  126. package/dest/spartan/utils/config.d.ts.map +1 -0
  127. package/dest/spartan/utils/config.js +21 -0
  128. package/dest/spartan/utils/health.d.ts +63 -0
  129. package/dest/spartan/utils/health.d.ts.map +1 -0
  130. package/dest/spartan/utils/health.js +202 -0
  131. package/dest/spartan/utils/helm.d.ts +15 -0
  132. package/dest/spartan/utils/helm.d.ts.map +1 -0
  133. package/dest/spartan/utils/helm.js +47 -0
  134. package/dest/spartan/utils/index.d.ts +9 -0
  135. package/dest/spartan/utils/index.d.ts.map +1 -0
  136. package/dest/spartan/utils/index.js +18 -0
  137. package/dest/spartan/utils/k8s.d.ts +128 -0
  138. package/dest/spartan/utils/k8s.d.ts.map +1 -0
  139. package/dest/spartan/utils/k8s.js +381 -0
  140. package/dest/spartan/utils/nodes.d.ts +41 -0
  141. package/dest/spartan/utils/nodes.d.ts.map +1 -0
  142. package/dest/spartan/utils/nodes.js +461 -0
  143. package/dest/spartan/utils/scripts.d.ts +30 -0
  144. package/dest/spartan/utils/scripts.d.ts.map +1 -0
  145. package/dest/spartan/utils/scripts.js +81 -0
  146. package/dest/spartan/utils.d.ts +2 -253
  147. package/dest/spartan/utils.d.ts.map +1 -1
  148. package/dest/spartan/utils.js +1 -892
  149. package/dest/test-wallet/test_wallet.d.ts +83 -0
  150. package/dest/test-wallet/test_wallet.d.ts.map +1 -0
  151. package/dest/test-wallet/test_wallet.js +214 -0
  152. package/dest/test-wallet/utils.d.ts +41 -0
  153. package/dest/test-wallet/utils.d.ts.map +1 -0
  154. package/dest/test-wallet/utils.js +71 -0
  155. package/dest/test-wallet/wallet_worker_script.d.ts +2 -0
  156. package/dest/test-wallet/wallet_worker_script.d.ts.map +1 -0
  157. package/dest/test-wallet/wallet_worker_script.js +40 -0
  158. package/dest/test-wallet/worker_wallet.d.ts +52 -0
  159. package/dest/test-wallet/worker_wallet.d.ts.map +1 -0
  160. package/dest/test-wallet/worker_wallet.js +103 -0
  161. package/dest/test-wallet/worker_wallet_schema.d.ts +274 -0
  162. package/dest/test-wallet/worker_wallet_schema.d.ts.map +1 -0
  163. package/dest/test-wallet/worker_wallet_schema.js +10 -0
  164. package/package.json +44 -39
  165. package/src/bench/client_flows/benchmark.ts +24 -2
  166. package/src/bench/client_flows/client_flows_benchmark.ts +140 -210
  167. package/src/bench/client_flows/config.ts +9 -1
  168. package/src/bench/utils.ts +10 -8
  169. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +114 -143
  170. package/src/e2e_cross_chain_messaging/cross_chain_messaging_test.ts +103 -111
  171. package/src/e2e_deploy_contract/deploy_test.ts +22 -15
  172. package/src/e2e_epochs/epochs_test.ts +129 -93
  173. package/src/e2e_fees/bridging_race.notest.ts +4 -10
  174. package/src/e2e_fees/fees_test.ts +163 -224
  175. package/src/e2e_l1_publisher/write_json.ts +16 -13
  176. package/src/e2e_nested_contract/nested_contract_test.ts +33 -56
  177. package/src/e2e_p2p/inactivity_slash_test.ts +3 -3
  178. package/src/e2e_p2p/p2p_network.ts +196 -170
  179. package/src/e2e_p2p/reqresp/utils.ts +235 -0
  180. package/src/e2e_p2p/shared.ts +22 -22
  181. package/src/e2e_storage_proof/fixtures/storage_proof.json +915 -0
  182. package/src/e2e_storage_proof/fixtures/storage_proof_fetcher.ts +190 -0
  183. package/src/e2e_storage_proof/fixtures/storage_proof_fixture.ts +173 -0
  184. package/src/e2e_token_contract/token_contract_test.ts +113 -119
  185. package/src/fixtures/authwit_proxy.ts +50 -0
  186. package/src/fixtures/dumps/epoch_proof_result.json +1 -1
  187. package/src/fixtures/e2e_prover_test.ts +127 -171
  188. package/src/fixtures/fixtures.ts +10 -0
  189. package/src/fixtures/ha_setup.ts +186 -0
  190. package/src/fixtures/index.ts +1 -0
  191. package/src/fixtures/setup.ts +896 -0
  192. package/src/fixtures/setup_p2p_test.ts +40 -44
  193. package/src/fixtures/token_utils.ts +7 -5
  194. package/src/fixtures/utils.ts +27 -901
  195. package/src/guides/up_quick_start.sh +3 -3
  196. package/src/quality_of_service/prometheus_client.ts +113 -0
  197. package/src/shared/cross_chain_test_harness.ts +13 -27
  198. package/src/shared/gas_portal_test_harness.ts +1 -1
  199. package/src/shared/index.ts +1 -0
  200. package/src/shared/mock_state_view.ts +188 -0
  201. package/src/shared/submit-transactions.ts +14 -16
  202. package/src/shared/uniswap_l1_l2.ts +21 -30
  203. package/src/simulators/lending_simulator.ts +2 -2
  204. package/src/simulators/token_simulator.ts +1 -29
  205. package/src/spartan/setup_test_wallets.ts +127 -25
  206. package/src/spartan/tx_metrics.ts +255 -9
  207. package/src/spartan/utils/bot.ts +188 -0
  208. package/src/spartan/utils/chaos.ts +253 -0
  209. package/src/spartan/utils/clients.ts +100 -0
  210. package/src/spartan/utils/config.ts +27 -0
  211. package/src/spartan/utils/health.ts +255 -0
  212. package/src/spartan/utils/helm.ts +84 -0
  213. package/src/spartan/utils/index.ts +65 -0
  214. package/src/spartan/utils/k8s.ts +535 -0
  215. package/src/spartan/utils/nodes.ts +538 -0
  216. package/src/spartan/utils/scripts.ts +99 -0
  217. package/src/spartan/utils.ts +1 -1158
  218. package/src/test-wallet/test_wallet.ts +306 -0
  219. package/src/test-wallet/utils.ts +112 -0
  220. package/src/test-wallet/wallet_worker_script.ts +43 -0
  221. package/src/test-wallet/worker_wallet.ts +165 -0
  222. package/src/test-wallet/worker_wallet_schema.ts +13 -0
  223. package/dest/fixtures/snapshot_manager.d.ts +0 -93
  224. package/dest/fixtures/snapshot_manager.d.ts.map +0 -1
  225. package/dest/fixtures/snapshot_manager.js +0 -488
  226. package/src/fixtures/snapshot_manager.ts +0 -646
@@ -1,5 +1,5 @@
1
1
  import type { InitialAccountData } from '@aztec/accounts/testing';
2
- import { type Archiver, createArchiver } from '@aztec/archiver';
2
+ import { AztecNodeService } from '@aztec/aztec-node';
3
3
  import { AztecAddress, EthAddress } from '@aztec/aztec.js/addresses';
4
4
  import { type Logger, createLogger } from '@aztec/aztec.js/log';
5
5
  import type { AztecNode } from '@aztec/aztec.js/node';
@@ -11,34 +11,32 @@ import {
11
11
  TestCircuitVerifier,
12
12
  } from '@aztec/bb-prover';
13
13
  import { BackendType, Barretenberg } from '@aztec/bb.js';
14
- import { createBlobClientWithFileStores } from '@aztec/blob-client/client';
15
14
  import type { DeployAztecL1ContractsReturnType } from '@aztec/ethereum/deploy-aztec-l1-contracts';
16
15
  import { Buffer32 } from '@aztec/foundation/buffer';
17
16
  import { SecretValue } from '@aztec/foundation/config';
18
17
  import { FeeAssetHandlerAbi } from '@aztec/l1-artifacts';
19
18
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
20
- import { type ProverNode, type ProverNodeConfig, createProverNode } from '@aztec/prover-node';
21
19
  import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
22
20
  import type { AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
23
- import { TestWallet } from '@aztec/test-wallet/server';
24
21
  import { getGenesisValues } from '@aztec/world-state/testing';
25
22
 
26
23
  import { type Hex, getContract } from 'viem';
27
24
  import { privateKeyToAddress } from 'viem/accounts';
28
25
 
29
26
  import { TokenSimulator } from '../simulators/token_simulator.js';
27
+ import { TestWallet } from '../test-wallet/test_wallet.js';
30
28
  import { getACVMConfig } from './get_acvm_config.js';
31
29
  import { getBBConfig } from './get_bb_config.js';
32
30
  import {
33
- type ISnapshotManager,
34
- type SubsystemsContext,
35
- createSnapshotManager,
31
+ type EndToEndContext,
36
32
  deployAccounts,
33
+ getPrivateKeyFromIndex,
34
+ getSponsoredFPCAddress,
37
35
  publicDeployAccounts,
38
- } from './snapshot_manager.js';
39
- import { getPrivateKeyFromIndex, getSponsoredFPCAddress, setupPXEAndGetWallet } from './utils.js';
40
-
41
- const { E2E_DATA_PATH: dataPath } = process.env;
36
+ setup,
37
+ setupPXEAndGetWallet,
38
+ teardown,
39
+ } from './setup.js';
42
40
 
43
41
  type ProvenSetup = {
44
42
  wallet: TestWallet;
@@ -56,7 +54,6 @@ export class FullProverTest {
56
54
  static TOKEN_NAME = 'USDC';
57
55
  static TOKEN_SYMBOL = 'USD';
58
56
  static TOKEN_DECIMALS = 18n;
59
- private snapshotManager: ISnapshotManager;
60
57
  logger: Logger;
61
58
  wallet!: TestWallet;
62
59
  provenWallet!: TestWallet;
@@ -73,105 +70,89 @@ export class FullProverTest {
73
70
  private acvmConfigCleanup?: () => Promise<void>;
74
71
  circuitProofVerifier?: ClientProtocolCircuitVerifier;
75
72
  provenAsset!: TokenContract;
76
- private context!: SubsystemsContext;
77
- private proverNode!: ProverNode;
78
- private simulatedProverNode!: ProverNode;
73
+ context!: EndToEndContext;
74
+ private proverAztecNode!: AztecNodeService;
75
+ private simulatedProverAztecNode!: AztecNodeService;
79
76
  public l1Contracts!: DeployAztecL1ContractsReturnType;
80
77
  public proverAddress!: EthAddress;
78
+ private minNumberOfTxsPerBlock: number;
79
+ private coinbase: EthAddress;
80
+ private realProofs: boolean;
81
81
 
82
- constructor(
83
- testName: string,
84
- private minNumberOfTxsPerBlock: number,
85
- coinbase: EthAddress,
86
- private realProofs = true,
87
- ) {
82
+ constructor(testName: string, minNumberOfTxsPerBlock: number, coinbase: EthAddress, realProofs = true) {
88
83
  this.logger = createLogger(`e2e:full_prover_test:${testName}`);
89
- this.snapshotManager = createSnapshotManager(
90
- `full_prover_integration/${testName}`,
91
- dataPath,
92
- { startProverNode: true, coinbase },
93
- {
94
- realVerifier: realProofs,
95
- },
96
- );
84
+ this.minNumberOfTxsPerBlock = minNumberOfTxsPerBlock;
85
+ this.coinbase = coinbase;
86
+ this.realProofs = realProofs;
97
87
  }
98
88
 
99
89
  /**
100
- * Adds two state shifts to snapshot manager.
101
- * 1. Add 2 accounts.
102
- * 2. Publicly deploy accounts, deploy token contract
90
+ * Applies base setup: deploys 2 accounts and token contract.
103
91
  */
104
- async applyBaseSnapshots() {
105
- await this.snapshotManager.snapshot(
106
- '2_accounts',
107
- deployAccounts(2, this.logger),
108
- ({ deployedAccounts }, { wallet }) => {
109
- this.deployedAccounts = deployedAccounts;
110
- this.accounts = deployedAccounts.map(a => a.address);
111
- this.wallet = wallet;
112
- return Promise.resolve();
113
- },
114
- );
115
-
116
- await this.snapshotManager.snapshot(
117
- 'client_prover_integration',
118
- async () => {
119
- // Create the token contract state.
120
- // Move this account thing to addAccounts above?
121
- this.logger.verbose(`Public deploy accounts...`);
122
- await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
123
-
124
- this.logger.verbose(`Deploying TokenContract...`);
125
- const { contract: asset, instance } = await TokenContract.deploy(
126
- this.wallet,
127
- this.accounts[0],
128
- FullProverTest.TOKEN_NAME,
129
- FullProverTest.TOKEN_SYMBOL,
130
- FullProverTest.TOKEN_DECIMALS,
131
- )
132
- .send({ from: this.accounts[0] })
133
- .wait();
134
- this.logger.verbose(`Token deployed to ${asset.address}`);
135
-
136
- return { tokenContractAddress: asset.address, tokenContractInstance: instance };
137
- },
138
- async ({ tokenContractAddress, tokenContractInstance }) => {
139
- // Restore the token contract state.
140
- this.fakeProofsAsset = TokenContract.at(tokenContractAddress, this.wallet);
141
- this.fakeProofsAssetInstance = tokenContractInstance;
142
- this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
143
-
144
- this.tokenSim = new TokenSimulator(
145
- this.fakeProofsAsset,
146
- this.wallet,
147
- this.accounts[0],
148
- this.logger,
149
- this.accounts,
150
- );
151
-
152
- expect(await this.fakeProofsAsset.methods.get_admin().simulate({ from: this.accounts[0] })).toBe(
153
- this.accounts[0].toBigInt(),
154
- );
155
- },
92
+ private async applyBaseSetup() {
93
+ this.logger.info('Applying base setup: deploying accounts');
94
+ const { deployedAccounts } = await deployAccounts(
95
+ 2,
96
+ this.logger,
97
+ )({
98
+ wallet: this.context.wallet,
99
+ initialFundedAccounts: this.context.initialFundedAccounts,
100
+ });
101
+ this.deployedAccounts = deployedAccounts;
102
+ this.accounts = deployedAccounts.map(a => a.address);
103
+ this.wallet = this.context.wallet;
104
+
105
+ this.logger.info('Applying base setup: publicly deploying accounts');
106
+ await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
107
+
108
+ this.logger.info('Applying base setup: deploying token contract');
109
+ const { contract: asset, instance } = await TokenContract.deploy(
110
+ this.wallet,
111
+ this.accounts[0],
112
+ FullProverTest.TOKEN_NAME,
113
+ FullProverTest.TOKEN_SYMBOL,
114
+ FullProverTest.TOKEN_DECIMALS,
115
+ ).send({ from: this.accounts[0], wait: { returnReceipt: true } });
116
+ this.logger.verbose(`Token deployed to ${asset.address}`);
117
+
118
+ this.fakeProofsAsset = asset;
119
+ this.fakeProofsAssetInstance = instance;
120
+ this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
121
+
122
+ this.tokenSim = new TokenSimulator(this.fakeProofsAsset, this.wallet, this.accounts[0], this.logger, this.accounts);
123
+
124
+ expect(await this.fakeProofsAsset.methods.get_admin().simulate({ from: this.accounts[0] })).toBe(
125
+ this.accounts[0].toBigInt(),
156
126
  );
157
127
  }
158
128
 
159
129
  async setup() {
160
- this.context = await this.snapshotManager.setup();
130
+ this.logger.info('Setting up subsystems from fresh');
131
+ this.context = await setup(0, {
132
+ startProverNode: true,
133
+ coinbase: this.coinbase,
134
+ fundSponsoredFPC: true,
135
+ skipAccountDeployment: true,
136
+ l1ContractsArgs: { realVerifier: this.realProofs },
137
+ });
138
+
139
+ await this.applyBaseSetup();
140
+ await this.applyMint();
141
+
142
+ this.logger.info(`Enabling proving`, { realProofs: this.realProofs });
161
143
 
162
144
  // We don't wish to mark as proven automatically, so we set the flag to false
163
145
  this.context.watcher.setIsMarkingAsProven(false);
164
146
 
165
- this.simulatedProverNode = this.context.proverNode!;
147
+ this.simulatedProverAztecNode = this.context.proverNode!;
166
148
  ({
167
149
  aztecNode: this.aztecNode,
168
150
  deployL1ContractsValues: this.l1Contracts,
169
151
  cheatCodes: this.cheatCodes,
170
152
  } = this.context);
171
- this.aztecNodeAdmin = this.context.aztecNode;
153
+ this.aztecNodeAdmin = this.context.aztecNodeService;
172
154
 
173
155
  const config = this.context.aztecNodeConfig;
174
- const blobClient = await createBlobClientWithFileStores(config, this.logger);
175
156
 
176
157
  // Configure a full prover PXE
177
158
  let acvmConfig: Awaited<ReturnType<typeof getACVMConfig>> | undefined;
@@ -214,7 +195,7 @@ export class FullProverTest {
214
195
  this.aztecNode,
215
196
  { proverEnabled: this.realProofs },
216
197
  undefined,
217
- true,
198
+ 'pxe-proven',
218
199
  );
219
200
  this.logger.debug(`Contract address ${this.fakeProofsAsset.address}`);
220
201
  await provenWallet.registerContract(this.fakeProofsAssetInstance, TokenContract.artifact);
@@ -233,20 +214,13 @@ export class FullProverTest {
233
214
  this.provenWallet = provenWallet;
234
215
  this.logger.info(`Full prover PXE started`);
235
216
 
236
- // Shutdown the current, simulated prover node
217
+ // Shutdown the current, simulated prover node (by stopping its hosting aztec node)
237
218
  this.logger.verbose('Shutting down simulated prover node');
238
- await this.simulatedProverNode.stop();
239
-
240
- // Creating temp store and archiver for fully proven prover node
241
- this.logger.verbose('Starting archiver for new prover node');
242
- const archiver = await createArchiver(
243
- { ...this.context.aztecNodeConfig, dataDirectory: undefined },
244
- { blobClient, dateProvider: this.context.dateProvider },
245
- { blockUntilSync: true },
246
- );
219
+ await this.simulatedProverAztecNode.stop();
247
220
 
248
221
  // The simulated prover node (now shutdown) used private key index 2
249
222
  const proverNodePrivateKey = getPrivateKeyFromIndex(2);
223
+ const proverNodePrivateKeyHex = `0x${proverNodePrivateKey!.toString('hex')}` as const;
250
224
  const proverNodeSenderAddress = privateKeyToAddress(new Buffer32(proverNodePrivateKey!).toString());
251
225
  this.proverAddress = EthAddress.fromString(proverNodeSenderAddress);
252
226
 
@@ -254,14 +228,21 @@ export class FullProverTest {
254
228
  await this.mintFeeJuice(proverNodeSenderAddress);
255
229
 
256
230
  this.logger.verbose('Starting prover node');
257
- const proverConfig: ProverNodeConfig = {
258
- ...this.context.aztecNodeConfig,
259
- txCollectionNodeRpcUrls: [],
231
+ const sponsoredFPCAddress = await getSponsoredFPCAddress();
232
+ const { prefilledPublicData } = await getGenesisValues(
233
+ this.context.initialFundedAccounts.map(a => a.address).concat(sponsoredFPCAddress),
234
+ );
235
+
236
+ const proverNodeConfig: Parameters<typeof AztecNodeService.createAndSync>[0] = {
237
+ ...config,
238
+ enableProverNode: true,
239
+ disableValidator: true,
260
240
  dataDirectory: undefined,
241
+ txCollectionNodeRpcUrls: [],
261
242
  proverId: this.proverAddress,
262
243
  realProofs: this.realProofs,
263
244
  proverAgentCount: 2,
264
- publisherPrivateKeys: [new SecretValue(`0x${proverNodePrivateKey!.toString('hex')}` as const)],
245
+ proverPublisherPrivateKeys: [new SecretValue(proverNodePrivateKeyHex)],
265
246
  proverNodeMaxPendingJobs: 100,
266
247
  proverNodeMaxParallelBlocksPerEpoch: 32,
267
248
  proverNodePollingIntervalMs: 100,
@@ -271,22 +252,15 @@ export class FullProverTest {
271
252
  txGatheringTimeoutMs: 24_000,
272
253
  proverNodeFailedEpochStore: undefined,
273
254
  proverNodeEpochProvingDelayMs: undefined,
255
+ validatorPrivateKeys: new SecretValue([]),
274
256
  };
275
- const sponsoredFPCAddress = await getSponsoredFPCAddress();
276
- const { prefilledPublicData } = await getGenesisValues(
277
- this.context.initialFundedAccounts.map(a => a.address).concat(sponsoredFPCAddress),
278
- );
279
- this.proverNode = await createProverNode(
280
- proverConfig,
281
- {
282
- aztecNodeTxProvider: this.aztecNode,
283
- archiver: archiver as Archiver,
284
- },
257
+
258
+ this.proverAztecNode = await AztecNodeService.createAndSync(
259
+ proverNodeConfig,
260
+ { dateProvider: this.context.dateProvider, p2pClientDeps: { rpcTxProviders: [this.aztecNode] } },
285
261
  { prefilledPublicData },
286
262
  );
287
- await this.proverNode.start();
288
-
289
- this.logger.warn(`Proofs are now enabled`);
263
+ this.logger.warn(`Proofs are now enabled`, { realProofs: this.realProofs });
290
264
  return this;
291
265
  }
292
266
 
@@ -299,72 +273,54 @@ export class FullProverTest {
299
273
  await this.context.deployL1ContractsValues.l1Client.waitForTransactionReceipt({ hash });
300
274
  }
301
275
 
302
- snapshot = <T>(
303
- name: string,
304
- apply: (context: SubsystemsContext) => Promise<T>,
305
- restore: (snapshotData: T, context: SubsystemsContext) => Promise<void> = () => Promise.resolve(),
306
- ): Promise<void> => this.snapshotManager.snapshot(name, apply, restore);
307
-
308
276
  async teardown() {
309
- await this.snapshotManager.teardown();
310
-
311
277
  // Cleanup related to the full prover PXEs
312
278
  for (let i = 0; i < this.provenComponents.length; i++) {
313
279
  await this.provenComponents[i].teardown();
314
280
  }
315
281
 
316
- // clean up the full prover node
317
- await this.proverNode.stop();
282
+ // clean up the full prover node (by stopping its hosting aztec node)
283
+ await this.proverAztecNode.stop();
318
284
 
319
285
  await Barretenberg.destroySingleton();
320
286
  await this.bbConfigCleanup?.();
321
287
  await this.acvmConfigCleanup?.();
288
+
289
+ await teardown(this.context);
322
290
  }
323
291
 
324
- async applyMintSnapshot() {
325
- await this.snapshotManager.snapshot(
326
- 'mint',
327
- async () => {
328
- const { fakeProofsAsset: asset, accounts } = this;
329
- const privateAmount = 10000n;
330
- const publicAmount = 10000n;
331
-
332
- this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
333
- await asset.methods
334
- .mint_to_public(accounts[0], privateAmount + publicAmount)
335
- .send({ from: accounts[0] })
336
- .wait();
337
-
338
- this.logger.verbose(`Transferring ${privateAmount} to private...`);
339
- await asset.methods.transfer_to_private(accounts[0], privateAmount).send({ from: accounts[0] }).wait();
340
-
341
- this.logger.verbose(`Minting complete.`);
342
-
343
- return { amount: publicAmount };
344
- },
345
- async ({ amount }) => {
346
- const {
347
- fakeProofsAsset: asset,
348
- accounts: [address],
349
- tokenSim,
350
- } = this;
351
- tokenSim.mintPublic(address, amount);
352
-
353
- const publicBalance = await asset.methods.balance_of_public(address).simulate({ from: address });
354
- this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
355
- expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
356
-
357
- tokenSim.mintPrivate(address, amount);
358
- const privateBalance = await asset.methods.balance_of_private(address).simulate({ from: address });
359
- this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
360
- expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
361
-
362
- const totalSupply = await asset.methods.total_supply().simulate({ from: address });
363
- this.logger.verbose(`Total supply: ${totalSupply}`);
364
- expect(totalSupply).toEqual(tokenSim.totalSupply);
365
-
366
- return Promise.resolve();
367
- },
368
- );
292
+ private async applyMint() {
293
+ this.logger.info('Applying mint setup');
294
+ const { fakeProofsAsset: asset, accounts } = this;
295
+ const privateAmount = 10000n;
296
+ const publicAmount = 10000n;
297
+
298
+ this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
299
+ await asset.methods.mint_to_public(accounts[0], privateAmount + publicAmount).send({ from: accounts[0] });
300
+
301
+ this.logger.verbose(`Transferring ${privateAmount} to private...`);
302
+ await asset.methods.transfer_to_private(accounts[0], privateAmount).send({ from: accounts[0] });
303
+
304
+ this.logger.info(`Minting complete`);
305
+
306
+ const {
307
+ fakeProofsAsset,
308
+ accounts: [address],
309
+ tokenSim,
310
+ } = this;
311
+ tokenSim.mintPublic(address, publicAmount);
312
+
313
+ const publicBalance = await fakeProofsAsset.methods.balance_of_public(address).simulate({ from: address });
314
+ this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
315
+ expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
316
+
317
+ tokenSim.mintPrivate(address, publicAmount);
318
+ const privateBalance = await fakeProofsAsset.methods.balance_of_private(address).simulate({ from: address });
319
+ this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
320
+ expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
321
+
322
+ const totalSupply = await fakeProofsAsset.methods.total_supply().simulate({ from: address });
323
+ this.logger.verbose(`Total supply: ${totalSupply}`);
324
+ expect(totalSupply).toEqual(tokenSim.totalSupply);
369
325
  }
370
326
  }
@@ -7,6 +7,16 @@ export const shouldCollectMetrics = () => {
7
7
  return undefined;
8
8
  };
9
9
 
10
+ /** Returns the boot node UDP port from environment variable or default value. */
11
+ export function getBootNodeUdpPort(): number {
12
+ return process.env.BOOT_NODE_UDP_PORT ? parseInt(process.env.BOOT_NODE_UDP_PORT, 10) : 4500;
13
+ }
14
+
15
+ /** Returns the anvil port from environment variable or default value. */
16
+ export function getAnvilPort(): number {
17
+ return process.env.ANVIL_PORT ? parseInt(process.env.ANVIL_PORT, 10) : 8545;
18
+ }
19
+
10
20
  export const TEST_PEER_CHECK_INTERVAL_MS = 1000;
11
21
  export const TEST_MAX_PENDING_TX_POOL_COUNT = 10_000; // Number of max pending TXs ~ 1.56GB
12
22
 
@@ -0,0 +1,186 @@
1
+ import { EthAddress } from '@aztec/aztec.js/addresses';
2
+ import { Fr } from '@aztec/aztec.js/fields';
3
+ import type { Logger } from '@aztec/aztec.js/log';
4
+ import { SecretValue } from '@aztec/foundation/config';
5
+
6
+ import { Pool } from 'pg';
7
+ import { privateKeyToAccount } from 'viem/accounts';
8
+
9
+ /**
10
+ * Configuration for HA database connection
11
+ */
12
+ export interface HADatabaseConfig {
13
+ /** PostgreSQL connection URL */
14
+ databaseUrl: string;
15
+ /** Node ID for HA coordination */
16
+ nodeId: string;
17
+ /** Enable HA signing */
18
+ haSigningEnabled: boolean;
19
+ /** Polling interval in ms */
20
+ pollingIntervalMs: number;
21
+ /** Signing timeout in ms */
22
+ signingTimeoutMs: number;
23
+ /** Max stuck duties age in ms */
24
+ maxStuckDutiesAgeMs: number;
25
+ }
26
+
27
+ /**
28
+ * Get database configuration from environment variables
29
+ */
30
+ export function createHADatabaseConfig(nodeId: string): HADatabaseConfig {
31
+ const databaseUrl = process.env.DATABASE_URL || 'postgresql://aztec:aztec@localhost:5432/aztec_ha_test';
32
+
33
+ return {
34
+ databaseUrl,
35
+ nodeId,
36
+ haSigningEnabled: true,
37
+ pollingIntervalMs: 100,
38
+ signingTimeoutMs: 3000,
39
+ maxStuckDutiesAgeMs: 72000,
40
+ };
41
+ }
42
+
43
+ /**
44
+ * Setup PostgreSQL database connection pool for HA tests
45
+ *
46
+ * Note: Database migrations should be run separately before starting tests,
47
+ * either via docker-compose entrypoint or manually with: aztec migrate-ha-db up
48
+ */
49
+ export function setupHADatabase(databaseUrl: string, logger?: Logger): Pool {
50
+ try {
51
+ // Create connection pool for test usage
52
+ // Migrations are already run by docker-compose entrypoint before tests start
53
+ const pool = new Pool({ connectionString: databaseUrl });
54
+
55
+ logger?.info('Connected to HA database (migrations should already be applied)');
56
+
57
+ return pool;
58
+ } catch (error) {
59
+ logger?.error(`Failed to connect to HA database: ${error}`);
60
+ throw error;
61
+ }
62
+ }
63
+
64
+ /**
65
+ * Clean up HA database - drop all tables
66
+ * Use this between tests to ensure clean state
67
+ */
68
+ export async function cleanupHADatabase(pool: Pool, logger?: Logger): Promise<void> {
69
+ try {
70
+ // Drop all HA tables
71
+ await pool.query('DROP TABLE IF EXISTS validator_duties CASCADE');
72
+ await pool.query('DROP TABLE IF EXISTS schema_version CASCADE');
73
+ // Drop migration tracking table (node-pg-migrate uses 'pgmigrations' by default)
74
+ // This ensures migrations will run fresh on next startup
75
+ await pool.query('DROP TABLE IF EXISTS pgmigrations CASCADE');
76
+
77
+ logger?.info('HA database cleaned up successfully');
78
+ } catch (error) {
79
+ logger?.error(`Failed to cleanup HA database: ${error}`);
80
+ throw error;
81
+ }
82
+ }
83
+
84
+ /**
85
+ * Query validator duties from the database
86
+ */
87
+ export async function getValidatorDuties(
88
+ pool: Pool,
89
+ slot: bigint,
90
+ dutyType?: 'ATTESTATION' | 'BLOCK_PROPOSAL' | 'GOVERNANCE_VOTE' | 'SLASHING_VOTE',
91
+ ): Promise<
92
+ Array<{
93
+ slot: string;
94
+ dutyType: string;
95
+ validatorAddress: string;
96
+ nodeId: string;
97
+ startedAt: Date;
98
+ completedAt: Date | undefined;
99
+ }>
100
+ > {
101
+ const query = dutyType
102
+ ? 'SELECT slot, duty_type, validator_address, node_id, started_at, completed_at FROM validator_duties WHERE slot = $1 AND duty_type = $2 ORDER BY started_at'
103
+ : 'SELECT slot, duty_type, validator_address, node_id, started_at, completed_at FROM validator_duties WHERE slot = $1 ORDER BY started_at';
104
+
105
+ const params = dutyType ? [slot.toString(), dutyType] : [slot.toString()];
106
+
107
+ const result = await pool.query<{
108
+ slot: string;
109
+ duty_type: string;
110
+ validator_address: string;
111
+ node_id: string;
112
+ started_at: Date;
113
+ completed_at: Date | undefined;
114
+ }>(query, params);
115
+
116
+ return result.rows.map(row => ({
117
+ slot: row.slot,
118
+ dutyType: row.duty_type,
119
+ validatorAddress: row.validator_address,
120
+ nodeId: row.node_id,
121
+ startedAt: row.started_at,
122
+ completedAt: row.completed_at,
123
+ }));
124
+ }
125
+
126
+ /**
127
+ * Convert private keys to Ethereum addresses
128
+ */
129
+ export function getAddressesFromPrivateKeys(privateKeys: `0x${string}`[]): string[] {
130
+ return privateKeys.map(pk => {
131
+ const account = privateKeyToAccount(pk);
132
+ return account.address;
133
+ });
134
+ }
135
+
136
+ /**
137
+ * Create initial validators from private keys for L1 contract deployment
138
+ */
139
+ export function createInitialValidatorsFromPrivateKeys(attesterPrivateKeys: `0x${string}`[]): Array<{
140
+ attester: EthAddress;
141
+ withdrawer: EthAddress;
142
+ privateKey: `0x${string}`;
143
+ bn254SecretKey: SecretValue<bigint>;
144
+ }> {
145
+ return attesterPrivateKeys.map(pk => {
146
+ const account = privateKeyToAccount(pk);
147
+ return {
148
+ attester: EthAddress.fromString(account.address),
149
+ withdrawer: EthAddress.fromString(account.address),
150
+ privateKey: pk,
151
+ bn254SecretKey: new SecretValue(Fr.random().toBigInt()),
152
+ };
153
+ });
154
+ }
155
+
156
+ /**
157
+ * Verify no duplicate attestations per validator (HA coordination check)
158
+ * Groups duties by validator address and verifies each validator attested exactly once
159
+ */
160
+ export function verifyNoDuplicateAttestations(
161
+ attestationDuties: Array<{
162
+ validatorAddress: string;
163
+ nodeId: string;
164
+ completedAt: Date | undefined;
165
+ }>,
166
+ logger?: Logger,
167
+ ): Map<string, typeof attestationDuties> {
168
+ const dutiesByValidator = new Map<string, typeof attestationDuties>();
169
+ for (const duty of attestationDuties) {
170
+ const existing = dutiesByValidator.get(duty.validatorAddress) || [];
171
+ existing.push(duty);
172
+ dutiesByValidator.set(duty.validatorAddress, existing);
173
+ }
174
+
175
+ for (const [validatorAddress, validatorDuties] of dutiesByValidator.entries()) {
176
+ if (validatorDuties.length !== 1) {
177
+ throw new Error(`Validator ${validatorAddress} attested ${validatorDuties.length} times (expected exactly once)`);
178
+ }
179
+ if (!validatorDuties[0].completedAt) {
180
+ throw new Error(`Validator ${validatorAddress} attestation duty not completed`);
181
+ }
182
+ logger?.info(`Validator ${validatorAddress} attested once via node ${validatorDuties[0].nodeId}`);
183
+ }
184
+
185
+ return dutiesByValidator;
186
+ }
@@ -1,4 +1,5 @@
1
1
  export * from './fixtures.js';
2
+ export * from './ha_setup.js';
2
3
  export * from './logging.js';
3
4
  export * from './utils.js';
4
5
  export * from './token_utils.js';