@aztec/end-to-end 0.0.1-commit.d3ec352c → 0.0.1-commit.f295ac2

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 (141) 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 +14 -15
  5. package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
  6. package/dest/bench/client_flows/client_flows_benchmark.js +110 -138
  7. package/dest/bench/client_flows/data_extractor.js +3 -1
  8. package/dest/bench/utils.d.ts +6 -6
  9. package/dest/bench/utils.d.ts.map +1 -1
  10. package/dest/bench/utils.js +18 -11
  11. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts +6 -7
  12. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map +1 -1
  13. package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js +98 -113
  14. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.d.ts +19 -13
  15. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.d.ts.map +1 -1
  16. package/dest/e2e_cross_chain_messaging/cross_chain_messaging_test.js +91 -71
  17. package/dest/e2e_deploy_contract/deploy_test.d.ts +4 -3
  18. package/dest/e2e_deploy_contract/deploy_test.d.ts.map +1 -1
  19. package/dest/e2e_deploy_contract/deploy_test.js +18 -13
  20. package/dest/e2e_epochs/epochs_test.d.ts +3 -2
  21. package/dest/e2e_epochs/epochs_test.d.ts.map +1 -1
  22. package/dest/e2e_epochs/epochs_test.js +13 -11
  23. package/dest/e2e_fees/bridging_race.notest.js +3 -5
  24. package/dest/e2e_fees/fees_test.d.ts +18 -15
  25. package/dest/e2e_fees/fees_test.d.ts.map +1 -1
  26. package/dest/e2e_fees/fees_test.js +126 -141
  27. package/dest/e2e_l1_publisher/write_json.d.ts +3 -3
  28. package/dest/e2e_l1_publisher/write_json.d.ts.map +1 -1
  29. package/dest/e2e_l1_publisher/write_json.js +19 -15
  30. package/dest/e2e_multi_validator/utils.js +1 -1
  31. package/dest/e2e_nested_contract/nested_contract_test.d.ts +6 -9
  32. package/dest/e2e_nested_contract/nested_contract_test.d.ts.map +1 -1
  33. package/dest/e2e_nested_contract/nested_contract_test.js +32 -40
  34. package/dest/e2e_p2p/inactivity_slash_test.d.ts +4 -4
  35. package/dest/e2e_p2p/inactivity_slash_test.d.ts.map +1 -1
  36. package/dest/e2e_p2p/inactivity_slash_test.js +6 -9
  37. package/dest/e2e_p2p/p2p_network.d.ts +13 -11
  38. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  39. package/dest/e2e_p2p/p2p_network.js +116 -111
  40. package/dest/e2e_p2p/shared.d.ts +2 -2
  41. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  42. package/dest/e2e_p2p/shared.js +4 -4
  43. package/dest/e2e_token_contract/token_contract_test.d.ts +16 -9
  44. package/dest/e2e_token_contract/token_contract_test.d.ts.map +1 -1
  45. package/dest/e2e_token_contract/token_contract_test.js +90 -92
  46. package/dest/fixtures/e2e_prover_test.d.ts +10 -18
  47. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  48. package/dest/fixtures/e2e_prover_test.js +88 -103
  49. package/dest/fixtures/fixtures.d.ts +2 -3
  50. package/dest/fixtures/fixtures.d.ts.map +1 -1
  51. package/dest/fixtures/fixtures.js +1 -2
  52. package/dest/fixtures/get_acvm_config.js +1 -1
  53. package/dest/fixtures/l1_to_l2_messaging.d.ts +4 -3
  54. package/dest/fixtures/l1_to_l2_messaging.d.ts.map +1 -1
  55. package/dest/fixtures/l1_to_l2_messaging.js +2 -2
  56. package/dest/fixtures/setup.d.ts +216 -0
  57. package/dest/fixtures/setup.d.ts.map +1 -0
  58. package/dest/fixtures/setup.js +684 -0
  59. package/dest/fixtures/setup_p2p_test.js +3 -3
  60. package/dest/fixtures/utils.d.ts +5 -638
  61. package/dest/fixtures/utils.d.ts.map +1 -1
  62. package/dest/fixtures/utils.js +4 -647
  63. package/dest/fixtures/web3signer.js +1 -1
  64. package/dest/fixtures/with_telemetry_utils.d.ts +2 -2
  65. package/dest/fixtures/with_telemetry_utils.d.ts.map +1 -1
  66. package/dest/fixtures/with_telemetry_utils.js +2 -2
  67. package/dest/quality_of_service/grafana_client.d.ts +41 -0
  68. package/dest/quality_of_service/grafana_client.d.ts.map +1 -0
  69. package/dest/quality_of_service/{alert_checker.js → grafana_client.js} +1 -1
  70. package/dest/quality_of_service/prometheus_client.d.ts +38 -0
  71. package/dest/quality_of_service/prometheus_client.d.ts.map +1 -0
  72. package/dest/quality_of_service/prometheus_client.js +67 -0
  73. package/dest/shared/cross_chain_test_harness.d.ts +16 -4
  74. package/dest/shared/cross_chain_test_harness.d.ts.map +1 -1
  75. package/dest/shared/cross_chain_test_harness.js +3 -3
  76. package/dest/shared/gas_portal_test_harness.d.ts +12 -2
  77. package/dest/shared/gas_portal_test_harness.d.ts.map +1 -1
  78. package/dest/shared/index.d.ts +2 -2
  79. package/dest/shared/index.d.ts.map +1 -1
  80. package/dest/shared/uniswap_l1_l2.d.ts +3 -27
  81. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  82. package/dest/shared/uniswap_l1_l2.js +43 -23
  83. package/dest/simulators/lending_simulator.d.ts +6 -2
  84. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  85. package/dest/simulators/lending_simulator.js +1 -1
  86. package/dest/spartan/setup_test_wallets.d.ts +4 -3
  87. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  88. package/dest/spartan/setup_test_wallets.js +2 -1
  89. package/dest/spartan/tx_metrics.d.ts +42 -0
  90. package/dest/spartan/tx_metrics.d.ts.map +1 -0
  91. package/dest/spartan/tx_metrics.js +118 -0
  92. package/dest/spartan/utils.d.ts +51 -12
  93. package/dest/spartan/utils.d.ts.map +1 -1
  94. package/dest/spartan/utils.js +262 -102
  95. package/package.json +40 -39
  96. package/src/bench/client_flows/benchmark.ts +24 -2
  97. package/src/bench/client_flows/client_flows_benchmark.ts +150 -200
  98. package/src/bench/client_flows/data_extractor.ts +1 -1
  99. package/src/bench/utils.ts +22 -14
  100. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +107 -142
  101. package/src/e2e_cross_chain_messaging/cross_chain_messaging_test.ts +140 -125
  102. package/src/e2e_deploy_contract/deploy_test.ts +21 -14
  103. package/src/e2e_epochs/epochs_test.ts +26 -13
  104. package/src/e2e_fees/bridging_race.notest.ts +3 -6
  105. package/src/e2e_fees/fees_test.ts +177 -216
  106. package/src/e2e_l1_publisher/write_json.ts +22 -17
  107. package/src/e2e_multi_validator/utils.ts +1 -1
  108. package/src/e2e_nested_contract/nested_contract_test.ts +35 -56
  109. package/src/e2e_p2p/inactivity_slash_test.ts +9 -12
  110. package/src/e2e_p2p/p2p_network.ts +174 -183
  111. package/src/e2e_p2p/shared.ts +11 -6
  112. package/src/e2e_token_contract/token_contract_test.ts +105 -118
  113. package/src/fixtures/e2e_prover_test.ts +112 -144
  114. package/src/fixtures/fixtures.ts +1 -3
  115. package/src/fixtures/get_acvm_config.ts +1 -1
  116. package/src/fixtures/l1_to_l2_messaging.ts +4 -2
  117. package/src/fixtures/setup.ts +1010 -0
  118. package/src/fixtures/setup_p2p_test.ts +3 -3
  119. package/src/fixtures/utils.ts +27 -966
  120. package/src/fixtures/web3signer.ts +1 -1
  121. package/src/fixtures/with_telemetry_utils.ts +2 -2
  122. package/src/quality_of_service/{alert_checker.ts → grafana_client.ts} +1 -1
  123. package/src/quality_of_service/prometheus_client.ts +113 -0
  124. package/src/shared/cross_chain_test_harness.ts +6 -10
  125. package/src/shared/gas_portal_test_harness.ts +1 -1
  126. package/src/shared/index.ts +1 -1
  127. package/src/shared/uniswap_l1_l2.ts +53 -67
  128. package/src/simulators/lending_simulator.ts +2 -2
  129. package/src/spartan/setup_test_wallets.ts +9 -2
  130. package/src/spartan/tx_metrics.ts +153 -0
  131. package/src/spartan/utils.ts +308 -45
  132. package/dest/fixtures/setup_l1_contracts.d.ts +0 -477
  133. package/dest/fixtures/setup_l1_contracts.d.ts.map +0 -1
  134. package/dest/fixtures/setup_l1_contracts.js +0 -17
  135. package/dest/fixtures/snapshot_manager.d.ts +0 -95
  136. package/dest/fixtures/snapshot_manager.d.ts.map +0 -1
  137. package/dest/fixtures/snapshot_manager.js +0 -505
  138. package/dest/quality_of_service/alert_checker.d.ts +0 -41
  139. package/dest/quality_of_service/alert_checker.d.ts.map +0 -1
  140. package/src/fixtures/setup_l1_contracts.ts +0 -26
  141. package/src/fixtures/snapshot_manager.ts +0 -665
@@ -7,24 +7,19 @@ import type { TestWallet } from '@aztec/test-wallet/server';
7
7
 
8
8
  import { jest } from '@jest/globals';
9
9
 
10
- import {
11
- type ISnapshotManager,
12
- type SubsystemsContext,
13
- createSnapshotManager,
14
- deployAccounts,
15
- publicDeployAccounts,
16
- } from '../fixtures/snapshot_manager.js';
10
+ import { type EndToEndContext, deployAccounts, publicDeployAccounts, setup, teardown } from '../fixtures/setup.js';
17
11
  import { mintTokensToPrivate } from '../fixtures/token_utils.js';
18
12
  import { TokenSimulator } from '../simulators/token_simulator.js';
19
13
 
20
- const { E2E_DATA_PATH: dataPath, METRICS_PORT: metricsPort } = process.env;
14
+ const { METRICS_PORT: metricsPort } = process.env;
21
15
 
22
16
  export class TokenContractTest {
23
17
  static TOKEN_NAME = 'USDC';
24
18
  static TOKEN_SYMBOL = 'USD';
25
19
  static TOKEN_DECIMALS = 18n;
26
- private snapshotManager: ISnapshotManager;
20
+ context!: EndToEndContext;
27
21
  logger: Logger;
22
+ metricsPort?: number;
28
23
  asset!: TokenContract;
29
24
  tokenSim!: TokenSimulator;
30
25
  node!: AztecNode;
@@ -35,133 +30,125 @@ export class TokenContractTest {
35
30
  account1Address!: AztecAddress;
36
31
  account2Address!: AztecAddress;
37
32
 
33
+ private shouldApplyBaseSetup = false;
34
+ private shouldApplyMint = false;
35
+
38
36
  constructor(testName: string) {
39
37
  this.logger = createLogger(`e2e:e2e_token_contract:${testName}`);
40
- this.snapshotManager = createSnapshotManager(`e2e_token_contract/${testName}`, dataPath, {
41
- metricsPort: metricsPort ? parseInt(metricsPort) : undefined,
42
- });
38
+ this.metricsPort = metricsPort ? parseInt(metricsPort) : undefined;
39
+ }
40
+
41
+ /**
42
+ * Registers that base setup should be applied during setup().
43
+ * Call this before setup() to deploy 3 accounts, publicly deploy accounts, token contract and a "bad account".
44
+ */
45
+ applyBaseSnapshots() {
46
+ this.shouldApplyBaseSetup = true;
43
47
  }
44
48
 
45
49
  /**
46
- * Adds two state shifts to snapshot manager.
47
- * 1. Add 3 accounts.
48
- * 2. Publicly deploy accounts, deploy token contract and a "bad account".
50
+ * Registers that mint should be applied during setup().
51
+ * Call this before setup() to mint tokens to the admin account.
49
52
  */
50
- async applyBaseSnapshots() {
53
+ applyMintSnapshot() {
54
+ this.shouldApplyMint = true;
55
+ }
56
+
57
+ /**
58
+ * Applies base setup: deploys 3 accounts, publicly deploys accounts, token contract and a "bad account".
59
+ */
60
+ private async applyBaseSetup() {
51
61
  // Adding a timeout of 2 minutes in here such that it is propagated to the underlying tests
52
62
  jest.setTimeout(120_000);
53
63
 
54
- await this.snapshotManager.snapshot(
55
- '3_accounts',
56
- deployAccounts(3, this.logger),
57
- ({ deployedAccounts }, { wallet, aztecNode }) => {
58
- this.node = aztecNode;
59
- this.wallet = wallet;
60
- [this.adminAddress, this.account1Address, this.account2Address] = deployedAccounts.map(acc => acc.address);
61
- return Promise.resolve();
62
- },
63
- );
64
+ this.logger.info('Applying base setup - deploying 3 accounts');
65
+ const { deployedAccounts } = await deployAccounts(
66
+ 3,
67
+ this.logger,
68
+ )({
69
+ wallet: this.context.wallet,
70
+ initialFundedAccounts: this.context.initialFundedAccounts,
71
+ });
64
72
 
65
- await this.snapshotManager.snapshot(
66
- 'e2e_token_contract',
67
- async () => {
68
- // Create the token contract state.
69
- // Move this account thing to addAccounts above?
70
- this.logger.verbose(`Public deploy accounts...`);
71
- await publicDeployAccounts(this.wallet, [this.adminAddress, this.account1Address]);
72
-
73
- this.logger.verbose(`Deploying TokenContract...`);
74
- const asset = await TokenContract.deploy(
75
- this.wallet,
76
- this.adminAddress,
77
- TokenContractTest.TOKEN_NAME,
78
- TokenContractTest.TOKEN_SYMBOL,
79
- TokenContractTest.TOKEN_DECIMALS,
80
- )
81
- .send({ from: this.adminAddress })
82
- .deployed();
83
- this.logger.verbose(`Token deployed to ${asset.address}`);
84
-
85
- this.logger.verbose(`Deploying bad account...`);
86
- this.badAccount = await InvalidAccountContract.deploy(this.wallet).send({ from: this.adminAddress }).deployed();
87
- this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
88
-
89
- return { tokenContractAddress: asset.address, badAccountAddress: this.badAccount.address };
90
- },
91
- async ({ tokenContractAddress, badAccountAddress }) => {
92
- // Restore the token contract state.
93
- this.asset = TokenContract.at(tokenContractAddress, this.wallet);
94
- this.logger.verbose(`Token contract address: ${this.asset.address}`);
95
-
96
- this.tokenSim = new TokenSimulator(this.asset, this.wallet, this.adminAddress, this.logger, [
97
- this.adminAddress,
98
- this.account1Address,
99
- ]);
100
-
101
- this.badAccount = InvalidAccountContract.at(badAccountAddress, this.wallet);
102
- this.logger.verbose(`Bad account address: ${this.badAccount.address}`);
103
-
104
- expect(await this.asset.methods.get_admin().simulate({ from: this.adminAddress })).toBe(
105
- this.adminAddress.toBigInt(),
106
- );
107
- },
73
+ this.node = this.context.aztecNodeService!;
74
+ this.wallet = this.context.wallet;
75
+ [this.adminAddress, this.account1Address, this.account2Address] = deployedAccounts.map(acc => acc.address);
76
+
77
+ this.logger.info('Applying base setup - deploying token contract');
78
+ this.logger.verbose(`Public deploy accounts...`);
79
+ await publicDeployAccounts(this.wallet, [this.adminAddress, this.account1Address]);
80
+
81
+ this.logger.verbose(`Deploying TokenContract...`);
82
+ this.asset = await TokenContract.deploy(
83
+ this.wallet,
84
+ this.adminAddress,
85
+ TokenContractTest.TOKEN_NAME,
86
+ TokenContractTest.TOKEN_SYMBOL,
87
+ TokenContractTest.TOKEN_DECIMALS,
88
+ )
89
+ .send({ from: this.adminAddress })
90
+ .deployed();
91
+ this.logger.verbose(`Token deployed to ${this.asset.address}`);
92
+
93
+ this.logger.verbose(`Deploying bad account...`);
94
+ this.badAccount = await InvalidAccountContract.deploy(this.wallet).send({ from: this.adminAddress }).deployed();
95
+ this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
96
+
97
+ this.tokenSim = new TokenSimulator(this.asset, this.wallet, this.adminAddress, this.logger, [
98
+ this.adminAddress,
99
+ this.account1Address,
100
+ ]);
101
+
102
+ expect(await this.asset.methods.get_admin().simulate({ from: this.adminAddress })).toBe(
103
+ this.adminAddress.toBigInt(),
108
104
  );
109
-
110
- // TokenContract.artifact.functions.forEach(fn => {
111
- // const sig = decodeFunctionSignature(fn.name, fn.parameters);
112
- // logger.verbose(`Function ${sig} and the selector: ${FunctionSelector.fromNameAndParameters(fn.name, fn.parameters)}`);
113
- // });
114
105
  }
115
106
 
116
107
  async setup() {
117
- await this.snapshotManager.setup();
118
- }
108
+ this.context = await setup(0, {
109
+ metricsPort: this.metricsPort,
110
+ fundSponsoredFPC: true,
111
+ skipAccountDeployment: true,
112
+ });
119
113
 
120
- snapshot = <T>(
121
- name: string,
122
- apply: (context: SubsystemsContext) => Promise<T>,
123
- restore: (snapshotData: T, context: SubsystemsContext) => Promise<void> = () => Promise.resolve(),
124
- ): Promise<void> => this.snapshotManager.snapshot(name, apply, restore);
114
+ if (this.shouldApplyBaseSetup) {
115
+ await this.applyBaseSetup();
116
+ }
117
+
118
+ if (this.shouldApplyMint) {
119
+ await this.applyMint();
120
+ }
121
+ }
125
122
 
126
123
  async teardown() {
127
- await this.snapshotManager.teardown();
124
+ await teardown(this.context);
128
125
  }
129
126
 
130
- async applyMintSnapshot() {
131
- await this.snapshotManager.snapshot(
132
- 'mint',
133
- async () => {
134
- const { asset, adminAddress } = this;
135
- const amount = 10000n;
136
-
137
- this.logger.verbose(`Minting ${amount} publicly...`);
138
- await asset.methods.mint_to_public(adminAddress, amount).send({ from: adminAddress }).wait();
139
-
140
- this.logger.verbose(`Minting ${amount} privately...`);
141
- await mintTokensToPrivate(asset, adminAddress, adminAddress, amount);
142
- this.logger.verbose(`Minting complete.`);
143
-
144
- return { amount };
145
- },
146
- async ({ amount }) => {
147
- const { asset, adminAddress, tokenSim } = this;
148
- tokenSim.mintPublic(adminAddress, amount);
149
-
150
- const publicBalance = await asset.methods.balance_of_public(adminAddress).simulate({ from: adminAddress });
151
- this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
152
- expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(adminAddress));
153
-
154
- tokenSim.mintPrivate(adminAddress, amount);
155
- const privateBalance = await asset.methods.balance_of_private(adminAddress).simulate({ from: adminAddress });
156
- this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
157
- expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(adminAddress));
158
-
159
- const totalSupply = await asset.methods.total_supply().simulate({ from: adminAddress });
160
- this.logger.verbose(`Total supply: ${totalSupply}`);
161
- expect(totalSupply).toEqual(tokenSim.totalSupply);
162
-
163
- return Promise.resolve();
164
- },
165
- );
127
+ private async applyMint() {
128
+ this.logger.info('Applying mint setup');
129
+ const { asset, adminAddress, tokenSim } = this;
130
+ const amount = 10000n;
131
+
132
+ this.logger.verbose(`Minting ${amount} publicly...`);
133
+ await asset.methods.mint_to_public(adminAddress, amount).send({ from: adminAddress }).wait();
134
+ tokenSim.mintPublic(adminAddress, amount);
135
+
136
+ const publicBalance = await asset.methods.balance_of_public(adminAddress).simulate({ from: adminAddress });
137
+ this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
138
+ expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(adminAddress));
139
+
140
+ this.logger.verbose(`Minting ${amount} privately...`);
141
+ await mintTokensToPrivate(asset, adminAddress, adminAddress, amount);
142
+ tokenSim.mintPrivate(adminAddress, amount);
143
+
144
+ const privateBalance = await asset.methods.balance_of_private(adminAddress).simulate({ from: adminAddress });
145
+ this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
146
+ expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(adminAddress));
147
+
148
+ const totalSupply = await asset.methods.total_supply().simulate({ from: adminAddress });
149
+ this.logger.verbose(`Total supply: ${totalSupply}`);
150
+ expect(totalSupply).toEqual(tokenSim.totalSupply);
151
+
152
+ this.logger.verbose(`Minting complete.`);
166
153
  }
167
154
  }
@@ -11,9 +11,8 @@ import {
11
11
  TestCircuitVerifier,
12
12
  } from '@aztec/bb-prover';
13
13
  import { BackendType, Barretenberg } from '@aztec/bb.js';
14
- import { createBlobSinkClient } from '@aztec/blob-sink/client';
15
- import type { BlobSinkServer } from '@aztec/blob-sink/server';
16
- import type { DeployL1ContractsReturnType } from '@aztec/ethereum';
14
+ import { createBlobClientWithFileStores } from '@aztec/blob-client/client';
15
+ import type { DeployAztecL1ContractsReturnType } from '@aztec/ethereum/deploy-aztec-l1-contracts';
17
16
  import { Buffer32 } from '@aztec/foundation/buffer';
18
17
  import { SecretValue } from '@aztec/foundation/config';
19
18
  import { FeeAssetHandlerAbi } from '@aztec/l1-artifacts';
@@ -31,15 +30,15 @@ import { TokenSimulator } from '../simulators/token_simulator.js';
31
30
  import { getACVMConfig } from './get_acvm_config.js';
32
31
  import { getBBConfig } from './get_bb_config.js';
33
32
  import {
34
- type ISnapshotManager,
35
- type SubsystemsContext,
36
- createSnapshotManager,
33
+ type EndToEndContext,
37
34
  deployAccounts,
35
+ getPrivateKeyFromIndex,
36
+ getSponsoredFPCAddress,
38
37
  publicDeployAccounts,
39
- } from './snapshot_manager.js';
40
- import { getPrivateKeyFromIndex, getSponsoredFPCAddress, setupPXEAndGetWallet } from './utils.js';
41
-
42
- const { E2E_DATA_PATH: dataPath } = process.env;
38
+ setup,
39
+ setupPXEAndGetWallet,
40
+ teardown,
41
+ } from './setup.js';
43
42
 
44
43
  type ProvenSetup = {
45
44
  wallet: TestWallet;
@@ -57,7 +56,6 @@ export class FullProverTest {
57
56
  static TOKEN_NAME = 'USDC';
58
57
  static TOKEN_SYMBOL = 'USD';
59
58
  static TOKEN_DECIMALS = 18n;
60
- private snapshotManager: ISnapshotManager;
61
59
  logger: Logger;
62
60
  wallet!: TestWallet;
63
61
  provenWallet!: TestWallet;
@@ -69,111 +67,97 @@ export class FullProverTest {
69
67
  aztecNode!: AztecNode;
70
68
  aztecNodeAdmin!: AztecNodeAdmin;
71
69
  cheatCodes!: CheatCodes;
72
- blobSink!: BlobSinkServer;
73
70
  private provenComponents: ProvenSetup[] = [];
74
71
  private bbConfigCleanup?: () => Promise<void>;
75
72
  private acvmConfigCleanup?: () => Promise<void>;
76
73
  circuitProofVerifier?: ClientProtocolCircuitVerifier;
77
74
  provenAsset!: TokenContract;
78
- private context!: SubsystemsContext;
75
+ context!: EndToEndContext;
79
76
  private proverNode!: ProverNode;
80
77
  private simulatedProverNode!: ProverNode;
81
- public l1Contracts!: DeployL1ContractsReturnType;
78
+ public l1Contracts!: DeployAztecL1ContractsReturnType;
82
79
  public proverAddress!: EthAddress;
80
+ private minNumberOfTxsPerBlock: number;
81
+ private coinbase: EthAddress;
82
+ private realProofs: boolean;
83
83
 
84
- constructor(
85
- testName: string,
86
- private minNumberOfTxsPerBlock: number,
87
- coinbase: EthAddress,
88
- private realProofs = true,
89
- ) {
84
+ constructor(testName: string, minNumberOfTxsPerBlock: number, coinbase: EthAddress, realProofs = true) {
90
85
  this.logger = createLogger(`e2e:full_prover_test:${testName}`);
91
- this.snapshotManager = createSnapshotManager(
92
- `full_prover_integration/${testName}`,
93
- dataPath,
94
- { startProverNode: true, fundRewardDistributor: true, coinbase },
95
- {
96
- realVerifier: realProofs,
97
- },
98
- );
86
+ this.minNumberOfTxsPerBlock = minNumberOfTxsPerBlock;
87
+ this.coinbase = coinbase;
88
+ this.realProofs = realProofs;
99
89
  }
100
90
 
101
91
  /**
102
- * Adds two state shifts to snapshot manager.
103
- * 1. Add 2 accounts.
104
- * 2. Publicly deploy accounts, deploy token contract
92
+ * Applies base setup: deploys 2 accounts and token contract.
105
93
  */
106
- async applyBaseSnapshots() {
107
- await this.snapshotManager.snapshot(
108
- '2_accounts',
109
- deployAccounts(2, this.logger),
110
- ({ deployedAccounts }, { wallet }) => {
111
- this.deployedAccounts = deployedAccounts;
112
- this.accounts = deployedAccounts.map(a => a.address);
113
- this.wallet = wallet;
114
- return Promise.resolve();
115
- },
116
- );
117
-
118
- await this.snapshotManager.snapshot(
119
- 'client_prover_integration',
120
- async () => {
121
- // Create the token contract state.
122
- // Move this account thing to addAccounts above?
123
- this.logger.verbose(`Public deploy accounts...`);
124
- await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
125
-
126
- this.logger.verbose(`Deploying TokenContract...`);
127
- const { contract: asset, instance } = await TokenContract.deploy(
128
- this.wallet,
129
- this.accounts[0],
130
- FullProverTest.TOKEN_NAME,
131
- FullProverTest.TOKEN_SYMBOL,
132
- FullProverTest.TOKEN_DECIMALS,
133
- )
134
- .send({ from: this.accounts[0] })
135
- .wait();
136
- this.logger.verbose(`Token deployed to ${asset.address}`);
137
-
138
- return { tokenContractAddress: asset.address, tokenContractInstance: instance };
139
- },
140
- async ({ tokenContractAddress, tokenContractInstance }) => {
141
- // Restore the token contract state.
142
- this.fakeProofsAsset = TokenContract.at(tokenContractAddress, this.wallet);
143
- this.fakeProofsAssetInstance = tokenContractInstance;
144
- this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
145
-
146
- this.tokenSim = new TokenSimulator(
147
- this.fakeProofsAsset,
148
- this.wallet,
149
- this.accounts[0],
150
- this.logger,
151
- this.accounts,
152
- );
153
-
154
- expect(await this.fakeProofsAsset.methods.get_admin().simulate({ from: this.accounts[0] })).toBe(
155
- this.accounts[0].toBigInt(),
156
- );
157
- },
94
+ private async applyBaseSetup() {
95
+ this.logger.info('Applying base setup: deploying accounts');
96
+ const { deployedAccounts } = await deployAccounts(
97
+ 2,
98
+ this.logger,
99
+ )({
100
+ wallet: this.context.wallet,
101
+ initialFundedAccounts: this.context.initialFundedAccounts,
102
+ });
103
+ this.deployedAccounts = deployedAccounts;
104
+ this.accounts = deployedAccounts.map(a => a.address);
105
+ this.wallet = this.context.wallet;
106
+
107
+ this.logger.info('Applying base setup: publicly deploying accounts');
108
+ await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
109
+
110
+ this.logger.info('Applying base setup: deploying token contract');
111
+ const { contract: asset, instance } = await TokenContract.deploy(
112
+ this.wallet,
113
+ this.accounts[0],
114
+ FullProverTest.TOKEN_NAME,
115
+ FullProverTest.TOKEN_SYMBOL,
116
+ FullProverTest.TOKEN_DECIMALS,
117
+ )
118
+ .send({ from: this.accounts[0] })
119
+ .wait();
120
+ this.logger.verbose(`Token deployed to ${asset.address}`);
121
+
122
+ this.fakeProofsAsset = asset;
123
+ this.fakeProofsAssetInstance = instance;
124
+ this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
125
+
126
+ this.tokenSim = new TokenSimulator(this.fakeProofsAsset, this.wallet, this.accounts[0], this.logger, this.accounts);
127
+
128
+ expect(await this.fakeProofsAsset.methods.get_admin().simulate({ from: this.accounts[0] })).toBe(
129
+ this.accounts[0].toBigInt(),
158
130
  );
159
131
  }
160
132
 
161
133
  async setup() {
162
- this.context = await this.snapshotManager.setup();
134
+ this.logger.info('Setting up subsystems from fresh');
135
+ this.context = await setup(0, {
136
+ startProverNode: true,
137
+ coinbase: this.coinbase,
138
+ fundSponsoredFPC: true,
139
+ skipAccountDeployment: true,
140
+ l1ContractsArgs: { realVerifier: this.realProofs },
141
+ });
142
+
143
+ await this.applyBaseSetup();
144
+ await this.applyMint();
145
+
146
+ this.logger.info(`Enabling proving`, { realProofs: this.realProofs });
163
147
 
164
148
  // We don't wish to mark as proven automatically, so we set the flag to false
165
- this.context.watcher.setIsMarkingAsProven(false);
149
+ this.context.watcher!.setIsMarkingAsProven(false);
166
150
 
167
151
  this.simulatedProverNode = this.context.proverNode!;
168
152
  ({
169
153
  aztecNode: this.aztecNode,
170
154
  deployL1ContractsValues: this.l1Contracts,
171
155
  cheatCodes: this.cheatCodes,
172
- blobSink: this.blobSink,
173
156
  } = this.context);
174
- this.aztecNodeAdmin = this.context.aztecNode;
157
+ this.aztecNodeAdmin = this.context.aztecNodeService!;
175
158
 
176
- const blobSinkClient = createBlobSinkClient({ blobSinkUrl: `http://localhost:${this.blobSink.port}` });
159
+ const config = this.context.aztecNodeConfig;
160
+ const blobClient = await createBlobClientWithFileStores(config, this.logger);
177
161
 
178
162
  // Configure a full prover PXE
179
163
  let acvmConfig: Awaited<ReturnType<typeof getACVMConfig>> | undefined;
@@ -243,7 +227,7 @@ export class FullProverTest {
243
227
  this.logger.verbose('Starting archiver for new prover node');
244
228
  const archiver = await createArchiver(
245
229
  { ...this.context.aztecNodeConfig, dataDirectory: undefined },
246
- { blobSinkClient, dateProvider: this.context.dateProvider },
230
+ { blobClient, dateProvider: this.context.dateProvider! },
247
231
  { blockUntilSync: true },
248
232
  );
249
233
 
@@ -283,13 +267,12 @@ export class FullProverTest {
283
267
  {
284
268
  aztecNodeTxProvider: this.aztecNode,
285
269
  archiver: archiver as Archiver,
286
- blobSinkClient,
287
270
  },
288
271
  { prefilledPublicData },
289
272
  );
290
273
  await this.proverNode.start();
291
274
 
292
- this.logger.warn(`Proofs are now enabled`);
275
+ this.logger.warn(`Proofs are now enabled`, { realProofs: this.realProofs });
293
276
  return this;
294
277
  }
295
278
 
@@ -302,15 +285,7 @@ export class FullProverTest {
302
285
  await this.context.deployL1ContractsValues.l1Client.waitForTransactionReceipt({ hash });
303
286
  }
304
287
 
305
- snapshot = <T>(
306
- name: string,
307
- apply: (context: SubsystemsContext) => Promise<T>,
308
- restore: (snapshotData: T, context: SubsystemsContext) => Promise<void> = () => Promise.resolve(),
309
- ): Promise<void> => this.snapshotManager.snapshot(name, apply, restore);
310
-
311
288
  async teardown() {
312
- await this.snapshotManager.teardown();
313
-
314
289
  // Cleanup related to the full prover PXEs
315
290
  for (let i = 0; i < this.provenComponents.length; i++) {
316
291
  await this.provenComponents[i].teardown();
@@ -322,52 +297,45 @@ export class FullProverTest {
322
297
  await Barretenberg.destroySingleton();
323
298
  await this.bbConfigCleanup?.();
324
299
  await this.acvmConfigCleanup?.();
325
- }
326
-
327
- async applyMintSnapshot() {
328
- await this.snapshotManager.snapshot(
329
- 'mint',
330
- async () => {
331
- const { fakeProofsAsset: asset, accounts } = this;
332
- const privateAmount = 10000n;
333
- const publicAmount = 10000n;
334
-
335
- this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
336
- await asset.methods
337
- .mint_to_public(accounts[0], privateAmount + publicAmount)
338
- .send({ from: accounts[0] })
339
- .wait();
340
300
 
341
- this.logger.verbose(`Transferring ${privateAmount} to private...`);
342
- await asset.methods.transfer_to_private(accounts[0], privateAmount).send({ from: accounts[0] }).wait();
343
-
344
- this.logger.verbose(`Minting complete.`);
301
+ await teardown(this.context);
302
+ }
345
303
 
346
- return { amount: publicAmount };
347
- },
348
- async ({ amount }) => {
349
- const {
350
- fakeProofsAsset: asset,
351
- accounts: [address],
352
- tokenSim,
353
- } = this;
354
- tokenSim.mintPublic(address, amount);
355
-
356
- const publicBalance = await asset.methods.balance_of_public(address).simulate({ from: address });
357
- this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
358
- expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
359
-
360
- tokenSim.mintPrivate(address, amount);
361
- const privateBalance = await asset.methods.balance_of_private(address).simulate({ from: address });
362
- this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
363
- expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
364
-
365
- const totalSupply = await asset.methods.total_supply().simulate({ from: address });
366
- this.logger.verbose(`Total supply: ${totalSupply}`);
367
- expect(totalSupply).toEqual(tokenSim.totalSupply);
368
-
369
- return Promise.resolve();
370
- },
371
- );
304
+ private async applyMint() {
305
+ this.logger.info('Applying mint setup');
306
+ const { fakeProofsAsset: asset, accounts } = this;
307
+ const privateAmount = 10000n;
308
+ const publicAmount = 10000n;
309
+
310
+ this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
311
+ await asset.methods
312
+ .mint_to_public(accounts[0], privateAmount + publicAmount)
313
+ .send({ from: accounts[0] })
314
+ .wait();
315
+
316
+ this.logger.verbose(`Transferring ${privateAmount} to private...`);
317
+ await asset.methods.transfer_to_private(accounts[0], privateAmount).send({ from: accounts[0] }).wait();
318
+
319
+ this.logger.info(`Minting complete`);
320
+
321
+ const {
322
+ fakeProofsAsset,
323
+ accounts: [address],
324
+ tokenSim,
325
+ } = this;
326
+ tokenSim.mintPublic(address, publicAmount);
327
+
328
+ const publicBalance = await fakeProofsAsset.methods.balance_of_public(address).simulate({ from: address });
329
+ this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
330
+ expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
331
+
332
+ tokenSim.mintPrivate(address, publicAmount);
333
+ const privateBalance = await fakeProofsAsset.methods.balance_of_private(address).simulate({ from: address });
334
+ this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
335
+ expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
336
+
337
+ const totalSupply = await fakeProofsAsset.methods.total_supply().simulate({ from: address });
338
+ this.logger.verbose(`Total supply: ${totalSupply}`);
339
+ expect(totalSupply).toEqual(tokenSim.totalSupply);
372
340
  }
373
341
  }
@@ -8,7 +8,7 @@ export const shouldCollectMetrics = () => {
8
8
  };
9
9
 
10
10
  export const TEST_PEER_CHECK_INTERVAL_MS = 1000;
11
- export const TEST_MAX_TX_POOL_SIZE = 2 * 1024 * 1024 * 1024; // 2GB
11
+ export const TEST_MAX_PENDING_TX_POOL_COUNT = 10_000; // Number of max pending TXs ~ 1.56GB
12
12
 
13
13
  export const MNEMONIC = 'test test test test test test test test test test test junk';
14
14
  export const privateKey = Buffer.from('ac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80', 'hex');
@@ -24,5 +24,3 @@ export const NO_L1_TO_L2_MSG_ERROR =
24
24
  /No non-nullified L1 to L2 message found for message hash|Tried to consume nonexistent L1-to-L2 message/;
25
25
  export const STATIC_CALL_STATE_MODIFICATION_ERROR = /Static call cannot update the state.*/;
26
26
  export const STATIC_CONTEXT_ASSERTION_ERROR = /Assertion failed: Function .* can only be called statically.*/;
27
-
28
- export const DEFAULT_BLOB_SINK_PORT = '5052';
@@ -1,6 +1,6 @@
1
1
  import type { Logger } from '@aztec/aztec.js/log';
2
2
  import { parseBooleanEnv } from '@aztec/foundation/config';
3
- import { randomBytes } from '@aztec/foundation/crypto';
3
+ import { randomBytes } from '@aztec/foundation/crypto/random';
4
4
  import { tryRmDir } from '@aztec/foundation/fs';
5
5
 
6
6
  import { promises as fs } from 'fs';
@@ -1,5 +1,7 @@
1
- import { type ExtendedViemWalletClient, type L1ContractAddresses, RollupContract } from '@aztec/ethereum';
2
- import { Fr } from '@aztec/foundation/fields';
1
+ import { RollupContract } from '@aztec/ethereum/contracts';
2
+ import type { L1ContractAddresses } from '@aztec/ethereum/l1-contract-addresses';
3
+ import type { ExtendedViemWalletClient } from '@aztec/ethereum/types';
4
+ import { Fr } from '@aztec/foundation/curves/bn254';
3
5
  import { tryJsonStringify } from '@aztec/foundation/json-rpc';
4
6
  import { InboxAbi } from '@aztec/l1-artifacts';
5
7
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';