@aztec/end-to-end 0.0.1-commit.c7c42ec → 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 (111) 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 +12 -13
  5. package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
  6. package/dest/bench/client_flows/client_flows_benchmark.js +105 -135
  7. package/dest/bench/client_flows/data_extractor.js +3 -1
  8. package/dest/bench/utils.d.ts +5 -5
  9. package/dest/bench/utils.d.ts.map +1 -1
  10. package/dest/bench/utils.js +8 -5
  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 +15 -10
  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 +89 -70
  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.js +2 -2
  21. package/dest/e2e_fees/bridging_race.notest.js +2 -4
  22. package/dest/e2e_fees/fees_test.d.ts +13 -13
  23. package/dest/e2e_fees/fees_test.d.ts.map +1 -1
  24. package/dest/e2e_fees/fees_test.js +123 -141
  25. package/dest/e2e_l1_publisher/write_json.d.ts +3 -3
  26. package/dest/e2e_l1_publisher/write_json.d.ts.map +1 -1
  27. package/dest/e2e_l1_publisher/write_json.js +19 -15
  28. package/dest/e2e_nested_contract/nested_contract_test.d.ts +6 -9
  29. package/dest/e2e_nested_contract/nested_contract_test.d.ts.map +1 -1
  30. package/dest/e2e_nested_contract/nested_contract_test.js +32 -40
  31. package/dest/e2e_p2p/inactivity_slash_test.d.ts +3 -3
  32. package/dest/e2e_p2p/inactivity_slash_test.d.ts.map +1 -1
  33. package/dest/e2e_p2p/inactivity_slash_test.js +3 -3
  34. package/dest/e2e_p2p/p2p_network.d.ts +7 -6
  35. package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
  36. package/dest/e2e_p2p/p2p_network.js +107 -104
  37. package/dest/e2e_p2p/shared.d.ts +1 -1
  38. package/dest/e2e_p2p/shared.d.ts.map +1 -1
  39. package/dest/e2e_p2p/shared.js +4 -4
  40. package/dest/e2e_token_contract/token_contract_test.d.ts +16 -9
  41. package/dest/e2e_token_contract/token_contract_test.d.ts.map +1 -1
  42. package/dest/e2e_token_contract/token_contract_test.js +90 -92
  43. package/dest/fixtures/e2e_prover_test.d.ts +8 -14
  44. package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
  45. package/dest/fixtures/e2e_prover_test.js +83 -95
  46. package/dest/fixtures/setup.d.ts +216 -0
  47. package/dest/fixtures/setup.d.ts.map +1 -0
  48. package/dest/fixtures/setup.js +684 -0
  49. package/dest/fixtures/utils.d.ts +5 -194
  50. package/dest/fixtures/utils.d.ts.map +1 -1
  51. package/dest/fixtures/utils.js +4 -619
  52. package/dest/quality_of_service/grafana_client.d.ts +41 -0
  53. package/dest/quality_of_service/grafana_client.d.ts.map +1 -0
  54. package/dest/quality_of_service/{alert_checker.js → grafana_client.js} +1 -1
  55. package/dest/quality_of_service/prometheus_client.d.ts +38 -0
  56. package/dest/quality_of_service/prometheus_client.d.ts.map +1 -0
  57. package/dest/quality_of_service/prometheus_client.js +67 -0
  58. package/dest/shared/cross_chain_test_harness.d.ts +14 -3
  59. package/dest/shared/cross_chain_test_harness.d.ts.map +1 -1
  60. package/dest/shared/cross_chain_test_harness.js +2 -2
  61. package/dest/shared/gas_portal_test_harness.d.ts +11 -1
  62. package/dest/shared/gas_portal_test_harness.d.ts.map +1 -1
  63. package/dest/shared/index.d.ts +2 -2
  64. package/dest/shared/index.d.ts.map +1 -1
  65. package/dest/shared/uniswap_l1_l2.d.ts +3 -28
  66. package/dest/shared/uniswap_l1_l2.d.ts.map +1 -1
  67. package/dest/shared/uniswap_l1_l2.js +39 -21
  68. package/dest/simulators/lending_simulator.d.ts +5 -1
  69. package/dest/simulators/lending_simulator.d.ts.map +1 -1
  70. package/dest/spartan/setup_test_wallets.d.ts +4 -3
  71. package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
  72. package/dest/spartan/tx_metrics.d.ts +4 -1
  73. package/dest/spartan/tx_metrics.d.ts.map +1 -1
  74. package/dest/spartan/tx_metrics.js +24 -1
  75. package/dest/spartan/utils.d.ts +12 -5
  76. package/dest/spartan/utils.d.ts.map +1 -1
  77. package/dest/spartan/utils.js +123 -73
  78. package/package.json +40 -39
  79. package/src/bench/client_flows/benchmark.ts +24 -2
  80. package/src/bench/client_flows/client_flows_benchmark.ts +143 -196
  81. package/src/bench/client_flows/data_extractor.ts +1 -1
  82. package/src/bench/utils.ts +11 -7
  83. package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +107 -142
  84. package/src/e2e_cross_chain_messaging/cross_chain_messaging_test.ts +127 -116
  85. package/src/e2e_deploy_contract/deploy_test.ts +21 -14
  86. package/src/e2e_epochs/epochs_test.ts +2 -2
  87. package/src/e2e_fees/bridging_race.notest.ts +2 -5
  88. package/src/e2e_fees/fees_test.ts +172 -216
  89. package/src/e2e_l1_publisher/write_json.ts +22 -17
  90. package/src/e2e_nested_contract/nested_contract_test.ts +35 -56
  91. package/src/e2e_p2p/inactivity_slash_test.ts +5 -5
  92. package/src/e2e_p2p/p2p_network.ts +166 -168
  93. package/src/e2e_p2p/shared.ts +6 -5
  94. package/src/e2e_token_contract/token_contract_test.ts +105 -118
  95. package/src/fixtures/e2e_prover_test.ts +107 -137
  96. package/src/fixtures/setup.ts +1010 -0
  97. package/src/fixtures/utils.ts +27 -907
  98. package/src/quality_of_service/{alert_checker.ts → grafana_client.ts} +1 -1
  99. package/src/quality_of_service/prometheus_client.ts +113 -0
  100. package/src/shared/cross_chain_test_harness.ts +3 -9
  101. package/src/shared/index.ts +1 -1
  102. package/src/shared/uniswap_l1_l2.ts +46 -58
  103. package/src/spartan/setup_test_wallets.ts +7 -1
  104. package/src/spartan/tx_metrics.ts +27 -4
  105. package/src/spartan/utils.ts +112 -24
  106. package/dest/fixtures/snapshot_manager.d.ts +0 -93
  107. package/dest/fixtures/snapshot_manager.d.ts.map +0 -1
  108. package/dest/fixtures/snapshot_manager.js +0 -493
  109. package/dest/quality_of_service/alert_checker.d.ts +0 -41
  110. package/dest/quality_of_service/alert_checker.d.ts.map +0 -1
  111. package/src/fixtures/snapshot_manager.ts +0 -651
@@ -101,7 +101,7 @@ export async function awaitCommitteeExists({ rollup, logger }) {
101
101
  committee = await rollup.getCurrentEpochCommittee();
102
102
  return committee && committee.length > 0;
103
103
  }, 'non-empty committee', 60);
104
- return committee;
104
+ return committee.map((c)=>c.toString());
105
105
  }
106
106
  export async function awaitOffenseDetected({ logger, nodeAdmin, slashingRoundSize, epochDuration, waitUntilOffenseCount, timeoutSeconds = 120 }) {
107
107
  const targetOffenseCount = waitUntilOffenseCount ?? 1;
@@ -130,9 +130,9 @@ export async function awaitOffenseDetected({ logger, nodeAdmin, slashingRoundSiz
130
130
  await cheatCodes.debugRollup();
131
131
  if (slashingProposer.type === 'empire') {
132
132
  // Await for the slash payload to be created if empire (no payload is created on tally until execution time)
133
- const targetEpoch = BigInt(await cheatCodes.getEpoch()) + await rollup.getLagInEpochsForValidatorSet() + 1n;
133
+ const targetEpoch = EpochNumber(await cheatCodes.getEpoch() + await rollup.getLagInEpochsForValidatorSet() + 1);
134
134
  logger.info(`Advancing to epoch ${targetEpoch} so we start slashing`);
135
- await cheatCodes.advanceToEpoch(EpochNumber.fromBigInt(targetEpoch));
135
+ await cheatCodes.advanceToEpoch(targetEpoch);
136
136
  const slashPayloadEvents = await retryUntil(async ()=>{
137
137
  const events = await slashFactory.getSlashPayloadCreatedEvents();
138
138
  return events.length > 0 ? events : undefined;
@@ -173,7 +173,7 @@ export async function awaitOffenseDetected({ logger, nodeAdmin, slashingRoundSiz
173
173
  }
174
174
  logger.info(`Advancing to check current committee`);
175
175
  await cheatCodes.debugRollup();
176
- await cheatCodes.advanceToEpoch(EpochNumber.fromBigInt(BigInt(await cheatCodes.getEpoch()) + await rollup.getLagInEpochsForValidatorSet() + 1n));
176
+ await cheatCodes.advanceToEpoch(EpochNumber(await cheatCodes.getEpoch() + await rollup.getLagInEpochsForValidatorSet() + 1));
177
177
  await cheatCodes.debugRollup();
178
178
  const committeeNextEpoch = await rollup.getCurrentEpochCommittee();
179
179
  // The committee should be undefined, since the validator set is empty
@@ -4,14 +4,15 @@ import type { AztecNode } from '@aztec/aztec.js/node';
4
4
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
5
5
  import { InvalidAccountContract } from '@aztec/noir-test-contracts.js/InvalidAccount';
6
6
  import type { TestWallet } from '@aztec/test-wallet/server';
7
- import { type SubsystemsContext } from '../fixtures/snapshot_manager.js';
7
+ import { type EndToEndContext } from '../fixtures/setup.js';
8
8
  import { TokenSimulator } from '../simulators/token_simulator.js';
9
9
  export declare class TokenContractTest {
10
10
  static TOKEN_NAME: string;
11
11
  static TOKEN_SYMBOL: string;
12
12
  static TOKEN_DECIMALS: bigint;
13
- private snapshotManager;
13
+ context: EndToEndContext;
14
14
  logger: Logger;
15
+ metricsPort?: number;
15
16
  asset: TokenContract;
16
17
  tokenSim: TokenSimulator;
17
18
  node: AztecNode;
@@ -20,16 +21,22 @@ export declare class TokenContractTest {
20
21
  adminAddress: AztecAddress;
21
22
  account1Address: AztecAddress;
22
23
  account2Address: AztecAddress;
24
+ private shouldApplyBaseSetup;
25
+ private shouldApplyMint;
23
26
  constructor(testName: string);
24
27
  /**
25
- * Adds two state shifts to snapshot manager.
26
- * 1. Add 3 accounts.
27
- * 2. Publicly deploy accounts, deploy token contract and a "bad account".
28
+ * Registers that base setup should be applied during setup().
29
+ * Call this before setup() to deploy 3 accounts, publicly deploy accounts, token contract and a "bad account".
28
30
  */
29
- applyBaseSnapshots(): Promise<void>;
31
+ applyBaseSnapshots(): void;
32
+ /**
33
+ * Registers that mint should be applied during setup().
34
+ * Call this before setup() to mint tokens to the admin account.
35
+ */
36
+ applyMintSnapshot(): void;
37
+ private applyBaseSetup;
30
38
  setup(): Promise<void>;
31
- snapshot: <T>(name: string, apply: (context: SubsystemsContext) => Promise<T>, restore?: (snapshotData: T, context: SubsystemsContext) => Promise<void>) => Promise<void>;
32
39
  teardown(): Promise<void>;
33
- applyMintSnapshot(): Promise<void>;
40
+ private applyMint;
34
41
  }
35
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidG9rZW5fY29udHJhY3RfdGVzdC5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL2UyZV90b2tlbl9jb250cmFjdC90b2tlbl9jb250cmFjdF90ZXN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUN6RCxPQUFPLEVBQUUsS0FBSyxNQUFNLEVBQWdCLE1BQU0scUJBQXFCLENBQUM7QUFDaEUsT0FBTyxLQUFLLEVBQUUsU0FBUyxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDdEQsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLGdDQUFnQyxDQUFDO0FBQy9ELE9BQU8sRUFBRSxzQkFBc0IsRUFBRSxNQUFNLDhDQUE4QyxDQUFDO0FBQ3RGLE9BQU8sS0FBSyxFQUFFLFVBQVUsRUFBRSxNQUFNLDJCQUEyQixDQUFDO0FBSTVELE9BQU8sRUFFTCxLQUFLLGlCQUFpQixFQUl2QixNQUFNLGlDQUFpQyxDQUFDO0FBRXpDLE9BQU8sRUFBRSxjQUFjLEVBQUUsTUFBTSxrQ0FBa0MsQ0FBQztBQUlsRSxxQkFBYSxpQkFBaUI7SUFDNUIsTUFBTSxDQUFDLFVBQVUsU0FBVTtJQUMzQixNQUFNLENBQUMsWUFBWSxTQUFTO0lBQzVCLE1BQU0sQ0FBQyxjQUFjLFNBQU87SUFDNUIsT0FBTyxDQUFDLGVBQWUsQ0FBbUI7SUFDMUMsTUFBTSxFQUFFLE1BQU0sQ0FBQztJQUNmLEtBQUssRUFBRyxhQUFhLENBQUM7SUFDdEIsUUFBUSxFQUFHLGNBQWMsQ0FBQztJQUMxQixJQUFJLEVBQUcsU0FBUyxDQUFDO0lBRWpCLFVBQVUsRUFBRyxzQkFBc0IsQ0FBQztJQUNwQyxNQUFNLEVBQUcsVUFBVSxDQUFDO0lBQ3BCLFlBQVksRUFBRyxZQUFZLENBQUM7SUFDNUIsZUFBZSxFQUFHLFlBQVksQ0FBQztJQUMvQixlQUFlLEVBQUcsWUFBWSxDQUFDO0lBRS9CLFlBQVksUUFBUSxFQUFFLE1BQU0sRUFLM0I7SUFFRDs7OztPQUlHO0lBQ0csa0JBQWtCLGtCQWdFdkI7SUFFSyxLQUFLLGtCQUVWO0lBRUQsUUFBUSxHQUFJLENBQUMsOEpBSTJEO0lBRWxFLFFBQVEsa0JBRWI7SUFFSyxpQkFBaUIsa0JBb0N0QjtDQUNGIn0=
42
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidG9rZW5fY29udHJhY3RfdGVzdC5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL2UyZV90b2tlbl9jb250cmFjdC90b2tlbl9jb250cmFjdF90ZXN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUN6RCxPQUFPLEVBQUUsS0FBSyxNQUFNLEVBQWdCLE1BQU0scUJBQXFCLENBQUM7QUFDaEUsT0FBTyxLQUFLLEVBQUUsU0FBUyxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDdEQsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLGdDQUFnQyxDQUFDO0FBQy9ELE9BQU8sRUFBRSxzQkFBc0IsRUFBRSxNQUFNLDhDQUE4QyxDQUFDO0FBQ3RGLE9BQU8sS0FBSyxFQUFFLFVBQVUsRUFBRSxNQUFNLDJCQUEyQixDQUFDO0FBSTVELE9BQU8sRUFBRSxLQUFLLGVBQWUsRUFBeUQsTUFBTSxzQkFBc0IsQ0FBQztBQUVuSCxPQUFPLEVBQUUsY0FBYyxFQUFFLE1BQU0sa0NBQWtDLENBQUM7QUFJbEUscUJBQWEsaUJBQWlCO0lBQzVCLE1BQU0sQ0FBQyxVQUFVLFNBQVU7SUFDM0IsTUFBTSxDQUFDLFlBQVksU0FBUztJQUM1QixNQUFNLENBQUMsY0FBYyxTQUFPO0lBQzVCLE9BQU8sRUFBRyxlQUFlLENBQUM7SUFDMUIsTUFBTSxFQUFFLE1BQU0sQ0FBQztJQUNmLFdBQVcsQ0FBQyxFQUFFLE1BQU0sQ0FBQztJQUNyQixLQUFLLEVBQUcsYUFBYSxDQUFDO0lBQ3RCLFFBQVEsRUFBRyxjQUFjLENBQUM7SUFDMUIsSUFBSSxFQUFHLFNBQVMsQ0FBQztJQUVqQixVQUFVLEVBQUcsc0JBQXNCLENBQUM7SUFDcEMsTUFBTSxFQUFHLFVBQVUsQ0FBQztJQUNwQixZQUFZLEVBQUcsWUFBWSxDQUFDO0lBQzVCLGVBQWUsRUFBRyxZQUFZLENBQUM7SUFDL0IsZUFBZSxFQUFHLFlBQVksQ0FBQztJQUUvQixPQUFPLENBQUMsb0JBQW9CLENBQVM7SUFDckMsT0FBTyxDQUFDLGVBQWUsQ0FBUztJQUVoQyxZQUFZLFFBQVEsRUFBRSxNQUFNLEVBRzNCO0lBRUQ7OztPQUdHO0lBQ0gsa0JBQWtCLFNBRWpCO0lBRUQ7OztPQUdHO0lBQ0gsaUJBQWlCLFNBRWhCO1lBS2EsY0FBYztJQStDdEIsS0FBSyxrQkFjVjtJQUVLLFFBQVEsa0JBRWI7WUFFYSxTQUFTO0NBMkJ4QiJ9
@@ -1 +1 @@
1
- {"version":3,"file":"token_contract_test.d.ts","sourceRoot":"","sources":["../../src/e2e_token_contract/token_contract_test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,qBAAqB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,8CAA8C,CAAC;AACtF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAI5D,OAAO,EAEL,KAAK,iBAAiB,EAIvB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAIlE,qBAAa,iBAAiB;IAC5B,MAAM,CAAC,UAAU,SAAU;IAC3B,MAAM,CAAC,YAAY,SAAS;IAC5B,MAAM,CAAC,cAAc,SAAO;IAC5B,OAAO,CAAC,eAAe,CAAmB;IAC1C,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAG,aAAa,CAAC;IACtB,QAAQ,EAAG,cAAc,CAAC;IAC1B,IAAI,EAAG,SAAS,CAAC;IAEjB,UAAU,EAAG,sBAAsB,CAAC;IACpC,MAAM,EAAG,UAAU,CAAC;IACpB,YAAY,EAAG,YAAY,CAAC;IAC5B,eAAe,EAAG,YAAY,CAAC;IAC/B,eAAe,EAAG,YAAY,CAAC;IAE/B,YAAY,QAAQ,EAAE,MAAM,EAK3B;IAED;;;;OAIG;IACG,kBAAkB,kBAgEvB;IAEK,KAAK,kBAEV;IAED,QAAQ,GAAI,CAAC,8JAI2D;IAElE,QAAQ,kBAEb;IAEK,iBAAiB,kBAoCtB;CACF"}
1
+ {"version":3,"file":"token_contract_test.d.ts","sourceRoot":"","sources":["../../src/e2e_token_contract/token_contract_test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,qBAAqB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,8CAA8C,CAAC;AACtF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAI5D,OAAO,EAAE,KAAK,eAAe,EAAyD,MAAM,sBAAsB,CAAC;AAEnH,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAIlE,qBAAa,iBAAiB;IAC5B,MAAM,CAAC,UAAU,SAAU;IAC3B,MAAM,CAAC,YAAY,SAAS;IAC5B,MAAM,CAAC,cAAc,SAAO;IAC5B,OAAO,EAAG,eAAe,CAAC;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,EAAG,aAAa,CAAC;IACtB,QAAQ,EAAG,cAAc,CAAC;IAC1B,IAAI,EAAG,SAAS,CAAC;IAEjB,UAAU,EAAG,sBAAsB,CAAC;IACpC,MAAM,EAAG,UAAU,CAAC;IACpB,YAAY,EAAG,YAAY,CAAC;IAC5B,eAAe,EAAG,YAAY,CAAC;IAC/B,eAAe,EAAG,YAAY,CAAC;IAE/B,OAAO,CAAC,oBAAoB,CAAS;IACrC,OAAO,CAAC,eAAe,CAAS;IAEhC,YAAY,QAAQ,EAAE,MAAM,EAG3B;IAED;;;OAGG;IACH,kBAAkB,SAEjB;IAED;;;OAGG;IACH,iBAAiB,SAEhB;YAKa,cAAc;IA+CtB,KAAK,kBAcV;IAEK,QAAQ,kBAEb;YAEa,SAAS;CA2BxB"}
@@ -2,16 +2,17 @@ import { createLogger } from '@aztec/aztec.js/log';
2
2
  import { TokenContract } from '@aztec/noir-contracts.js/Token';
3
3
  import { InvalidAccountContract } from '@aztec/noir-test-contracts.js/InvalidAccount';
4
4
  import { jest } from '@jest/globals';
5
- import { createSnapshotManager, deployAccounts, publicDeployAccounts } from '../fixtures/snapshot_manager.js';
5
+ import { deployAccounts, publicDeployAccounts, setup, teardown } from '../fixtures/setup.js';
6
6
  import { mintTokensToPrivate } from '../fixtures/token_utils.js';
7
7
  import { TokenSimulator } from '../simulators/token_simulator.js';
8
- const { E2E_DATA_PATH: dataPath, METRICS_PORT: metricsPort } = process.env;
8
+ const { METRICS_PORT: metricsPort } = process.env;
9
9
  export class TokenContractTest {
10
10
  static TOKEN_NAME = 'USDC';
11
11
  static TOKEN_SYMBOL = 'USD';
12
12
  static TOKEN_DECIMALS = 18n;
13
- snapshotManager;
13
+ context;
14
14
  logger;
15
+ metricsPort;
15
16
  asset;
16
17
  tokenSim;
17
18
  node;
@@ -20,107 +21,104 @@ export class TokenContractTest {
20
21
  adminAddress;
21
22
  account1Address;
22
23
  account2Address;
24
+ shouldApplyBaseSetup = false;
25
+ shouldApplyMint = false;
23
26
  constructor(testName){
24
27
  this.logger = createLogger(`e2e:e2e_token_contract:${testName}`);
25
- this.snapshotManager = createSnapshotManager(`e2e_token_contract/${testName}`, dataPath, {
26
- metricsPort: metricsPort ? parseInt(metricsPort) : undefined
27
- });
28
+ this.metricsPort = metricsPort ? parseInt(metricsPort) : undefined;
29
+ }
30
+ /**
31
+ * Registers that base setup should be applied during setup().
32
+ * Call this before setup() to deploy 3 accounts, publicly deploy accounts, token contract and a "bad account".
33
+ */ applyBaseSnapshots() {
34
+ this.shouldApplyBaseSetup = true;
35
+ }
36
+ /**
37
+ * Registers that mint should be applied during setup().
38
+ * Call this before setup() to mint tokens to the admin account.
39
+ */ applyMintSnapshot() {
40
+ this.shouldApplyMint = true;
28
41
  }
29
42
  /**
30
- * Adds two state shifts to snapshot manager.
31
- * 1. Add 3 accounts.
32
- * 2. Publicly deploy accounts, deploy token contract and a "bad account".
33
- */ async applyBaseSnapshots() {
43
+ * Applies base setup: deploys 3 accounts, publicly deploys accounts, token contract and a "bad account".
44
+ */ async applyBaseSetup() {
34
45
  // Adding a timeout of 2 minutes in here such that it is propagated to the underlying tests
35
46
  jest.setTimeout(120_000);
36
- await this.snapshotManager.snapshot('3_accounts', deployAccounts(3, this.logger), ({ deployedAccounts }, { wallet, aztecNode })=>{
37
- this.node = aztecNode;
38
- this.wallet = wallet;
39
- [this.adminAddress, this.account1Address, this.account2Address] = deployedAccounts.map((acc)=>acc.address);
40
- return Promise.resolve();
47
+ this.logger.info('Applying base setup - deploying 3 accounts');
48
+ const { deployedAccounts } = await deployAccounts(3, this.logger)({
49
+ wallet: this.context.wallet,
50
+ initialFundedAccounts: this.context.initialFundedAccounts
41
51
  });
42
- await this.snapshotManager.snapshot('e2e_token_contract', async ()=>{
43
- // Create the token contract state.
44
- // Move this account thing to addAccounts above?
45
- this.logger.verbose(`Public deploy accounts...`);
46
- await publicDeployAccounts(this.wallet, [
47
- this.adminAddress,
48
- this.account1Address
49
- ]);
50
- this.logger.verbose(`Deploying TokenContract...`);
51
- const asset = await TokenContract.deploy(this.wallet, this.adminAddress, TokenContractTest.TOKEN_NAME, TokenContractTest.TOKEN_SYMBOL, TokenContractTest.TOKEN_DECIMALS).send({
52
- from: this.adminAddress
53
- }).deployed();
54
- this.logger.verbose(`Token deployed to ${asset.address}`);
55
- this.logger.verbose(`Deploying bad account...`);
56
- this.badAccount = await InvalidAccountContract.deploy(this.wallet).send({
57
- from: this.adminAddress
58
- }).deployed();
59
- this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
60
- return {
61
- tokenContractAddress: asset.address,
62
- badAccountAddress: this.badAccount.address
63
- };
64
- }, async ({ tokenContractAddress, badAccountAddress })=>{
65
- // Restore the token contract state.
66
- this.asset = TokenContract.at(tokenContractAddress, this.wallet);
67
- this.logger.verbose(`Token contract address: ${this.asset.address}`);
68
- this.tokenSim = new TokenSimulator(this.asset, this.wallet, this.adminAddress, this.logger, [
69
- this.adminAddress,
70
- this.account1Address
71
- ]);
72
- this.badAccount = InvalidAccountContract.at(badAccountAddress, this.wallet);
73
- this.logger.verbose(`Bad account address: ${this.badAccount.address}`);
74
- expect(await this.asset.methods.get_admin().simulate({
75
- from: this.adminAddress
76
- })).toBe(this.adminAddress.toBigInt());
77
- });
78
- // TokenContract.artifact.functions.forEach(fn => {
79
- // const sig = decodeFunctionSignature(fn.name, fn.parameters);
80
- // logger.verbose(`Function ${sig} and the selector: ${FunctionSelector.fromNameAndParameters(fn.name, fn.parameters)}`);
81
- // });
52
+ this.node = this.context.aztecNodeService;
53
+ this.wallet = this.context.wallet;
54
+ [this.adminAddress, this.account1Address, this.account2Address] = deployedAccounts.map((acc)=>acc.address);
55
+ this.logger.info('Applying base setup - deploying token contract');
56
+ this.logger.verbose(`Public deploy accounts...`);
57
+ await publicDeployAccounts(this.wallet, [
58
+ this.adminAddress,
59
+ this.account1Address
60
+ ]);
61
+ this.logger.verbose(`Deploying TokenContract...`);
62
+ this.asset = await TokenContract.deploy(this.wallet, this.adminAddress, TokenContractTest.TOKEN_NAME, TokenContractTest.TOKEN_SYMBOL, TokenContractTest.TOKEN_DECIMALS).send({
63
+ from: this.adminAddress
64
+ }).deployed();
65
+ this.logger.verbose(`Token deployed to ${this.asset.address}`);
66
+ this.logger.verbose(`Deploying bad account...`);
67
+ this.badAccount = await InvalidAccountContract.deploy(this.wallet).send({
68
+ from: this.adminAddress
69
+ }).deployed();
70
+ this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
71
+ this.tokenSim = new TokenSimulator(this.asset, this.wallet, this.adminAddress, this.logger, [
72
+ this.adminAddress,
73
+ this.account1Address
74
+ ]);
75
+ expect(await this.asset.methods.get_admin().simulate({
76
+ from: this.adminAddress
77
+ })).toBe(this.adminAddress.toBigInt());
82
78
  }
83
79
  async setup() {
84
- await this.snapshotManager.setup();
80
+ this.context = await setup(0, {
81
+ metricsPort: this.metricsPort,
82
+ fundSponsoredFPC: true,
83
+ skipAccountDeployment: true
84
+ });
85
+ if (this.shouldApplyBaseSetup) {
86
+ await this.applyBaseSetup();
87
+ }
88
+ if (this.shouldApplyMint) {
89
+ await this.applyMint();
90
+ }
85
91
  }
86
- snapshot = (name, apply, restore = ()=>Promise.resolve())=>this.snapshotManager.snapshot(name, apply, restore);
87
92
  async teardown() {
88
- await this.snapshotManager.teardown();
93
+ await teardown(this.context);
89
94
  }
90
- async applyMintSnapshot() {
91
- await this.snapshotManager.snapshot('mint', async ()=>{
92
- const { asset, adminAddress } = this;
93
- const amount = 10000n;
94
- this.logger.verbose(`Minting ${amount} publicly...`);
95
- await asset.methods.mint_to_public(adminAddress, amount).send({
96
- from: adminAddress
97
- }).wait();
98
- this.logger.verbose(`Minting ${amount} privately...`);
99
- await mintTokensToPrivate(asset, adminAddress, adminAddress, amount);
100
- this.logger.verbose(`Minting complete.`);
101
- return {
102
- amount
103
- };
104
- }, async ({ amount })=>{
105
- const { asset, adminAddress, tokenSim } = this;
106
- tokenSim.mintPublic(adminAddress, amount);
107
- const publicBalance = await asset.methods.balance_of_public(adminAddress).simulate({
108
- from: adminAddress
109
- });
110
- this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
111
- expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(adminAddress));
112
- tokenSim.mintPrivate(adminAddress, amount);
113
- const privateBalance = await asset.methods.balance_of_private(adminAddress).simulate({
114
- from: adminAddress
115
- });
116
- this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
117
- expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(adminAddress));
118
- const totalSupply = await asset.methods.total_supply().simulate({
119
- from: adminAddress
120
- });
121
- this.logger.verbose(`Total supply: ${totalSupply}`);
122
- expect(totalSupply).toEqual(tokenSim.totalSupply);
123
- return Promise.resolve();
95
+ async applyMint() {
96
+ this.logger.info('Applying mint setup');
97
+ const { asset, adminAddress, tokenSim } = this;
98
+ const amount = 10000n;
99
+ this.logger.verbose(`Minting ${amount} publicly...`);
100
+ await asset.methods.mint_to_public(adminAddress, amount).send({
101
+ from: adminAddress
102
+ }).wait();
103
+ tokenSim.mintPublic(adminAddress, amount);
104
+ const publicBalance = await asset.methods.balance_of_public(adminAddress).simulate({
105
+ from: adminAddress
106
+ });
107
+ this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
108
+ expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(adminAddress));
109
+ this.logger.verbose(`Minting ${amount} privately...`);
110
+ await mintTokensToPrivate(asset, adminAddress, adminAddress, amount);
111
+ tokenSim.mintPrivate(adminAddress, amount);
112
+ const privateBalance = await asset.methods.balance_of_private(adminAddress).simulate({
113
+ from: adminAddress
114
+ });
115
+ this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
116
+ expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(adminAddress));
117
+ const totalSupply = await asset.methods.total_supply().simulate({
118
+ from: adminAddress
124
119
  });
120
+ this.logger.verbose(`Total supply: ${totalSupply}`);
121
+ expect(totalSupply).toEqual(tokenSim.totalSupply);
122
+ this.logger.verbose(`Minting complete.`);
125
123
  }
126
124
  }
@@ -10,7 +10,7 @@ import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
10
10
  import type { AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
11
11
  import { TestWallet } from '@aztec/test-wallet/server';
12
12
  import { TokenSimulator } from '../simulators/token_simulator.js';
13
- import { type SubsystemsContext } from './snapshot_manager.js';
13
+ import { type EndToEndContext } from './setup.js';
14
14
  /**
15
15
  * Largely taken from the e2e_token_contract test file. We deploy 2 accounts and a token contract.
16
16
  * However, we then setup a second PXE with a full prover instance.
@@ -18,12 +18,9 @@ import { type SubsystemsContext } from './snapshot_manager.js';
18
18
  * We then prove and verify transactions created via this full prover PXE.
19
19
  */
20
20
  export declare class FullProverTest {
21
- private minNumberOfTxsPerBlock;
22
- private realProofs;
23
21
  static TOKEN_NAME: string;
24
22
  static TOKEN_SYMBOL: string;
25
23
  static TOKEN_DECIMALS: bigint;
26
- private snapshotManager;
27
24
  logger: Logger;
28
25
  wallet: TestWallet;
29
26
  provenWallet: TestWallet;
@@ -40,22 +37,19 @@ export declare class FullProverTest {
40
37
  private acvmConfigCleanup?;
41
38
  circuitProofVerifier?: ClientProtocolCircuitVerifier;
42
39
  provenAsset: TokenContract;
43
- private context;
40
+ context: EndToEndContext;
44
41
  private proverNode;
45
42
  private simulatedProverNode;
46
43
  l1Contracts: DeployAztecL1ContractsReturnType;
47
44
  proverAddress: EthAddress;
45
+ private minNumberOfTxsPerBlock;
46
+ private coinbase;
47
+ private realProofs;
48
48
  constructor(testName: string, minNumberOfTxsPerBlock: number, coinbase: EthAddress, realProofs?: boolean);
49
- /**
50
- * Adds two state shifts to snapshot manager.
51
- * 1. Add 2 accounts.
52
- * 2. Publicly deploy accounts, deploy token contract
53
- */
54
- applyBaseSnapshots(): Promise<void>;
49
+ private applyBaseSetup;
55
50
  setup(): Promise<this>;
56
51
  private mintFeeJuice;
57
- snapshot: <T>(name: string, apply: (context: SubsystemsContext) => Promise<T>, restore?: (snapshotData: T, context: SubsystemsContext) => Promise<void>) => Promise<void>;
58
52
  teardown(): Promise<void>;
59
- applyMintSnapshot(): Promise<void>;
53
+ private applyMint;
60
54
  }
61
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZTJlX3Byb3Zlcl90ZXN0LmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvZml4dHVyZXMvZTJlX3Byb3Zlcl90ZXN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBSyxFQUFFLGtCQUFrQixFQUFFLE1BQU0seUJBQXlCLENBQUM7QUFFbEUsT0FBTyxFQUFFLFlBQVksRUFBRSxVQUFVLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUNyRSxPQUFPLEVBQUUsS0FBSyxNQUFNLEVBQWdCLE1BQU0scUJBQXFCLENBQUM7QUFDaEUsT0FBTyxLQUFLLEVBQUUsU0FBUyxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDdEQsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ2xELE9BQU8sRUFFTCxLQUFLLDZCQUE2QixFQUduQyxNQUFNLGtCQUFrQixDQUFDO0FBRzFCLE9BQU8sS0FBSyxFQUFFLGdDQUFnQyxFQUFFLE1BQU0sMkNBQTJDLENBQUM7QUFJbEcsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLGdDQUFnQyxDQUFDO0FBRS9ELE9BQU8sS0FBSyxFQUFFLDJCQUEyQixFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFDMUUsT0FBTyxLQUFLLEVBQUUsY0FBYyxFQUFFLE1BQU0saUNBQWlDLENBQUM7QUFDdEUsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLDJCQUEyQixDQUFDO0FBTXZELE9BQU8sRUFBRSxjQUFjLEVBQUUsTUFBTSxrQ0FBa0MsQ0FBQztBQUdsRSxPQUFPLEVBRUwsS0FBSyxpQkFBaUIsRUFJdkIsTUFBTSx1QkFBdUIsQ0FBQztBQVUvQjs7Ozs7R0FLRztBQUVILHFCQUFhLGNBQWM7SUE2QnZCLE9BQU8sQ0FBQyxzQkFBc0I7SUFFOUIsT0FBTyxDQUFDLFVBQVU7SUE5QnBCLE1BQU0sQ0FBQyxVQUFVLFNBQVU7SUFDM0IsTUFBTSxDQUFDLFlBQVksU0FBUztJQUM1QixNQUFNLENBQUMsY0FBYyxTQUFPO0lBQzVCLE9BQU8sQ0FBQyxlQUFlLENBQW1CO0lBQzFDLE1BQU0sRUFBRSxNQUFNLENBQUM7SUFDZixNQUFNLEVBQUcsVUFBVSxDQUFDO0lBQ3BCLFlBQVksRUFBRyxVQUFVLENBQUM7SUFDMUIsUUFBUSxFQUFFLFlBQVksRUFBRSxDQUFNO0lBQzlCLGdCQUFnQixFQUFHLGtCQUFrQixFQUFFLENBQUM7SUFDeEMsZUFBZSxFQUFHLGFBQWEsQ0FBQztJQUNoQyx1QkFBdUIsRUFBRywyQkFBMkIsQ0FBQztJQUN0RCxRQUFRLEVBQUcsY0FBYyxDQUFDO0lBQzFCLFNBQVMsRUFBRyxTQUFTLENBQUM7SUFDdEIsY0FBYyxFQUFHLGNBQWMsQ0FBQztJQUNoQyxVQUFVLEVBQUcsVUFBVSxDQUFDO0lBQ3hCLE9BQU8sQ0FBQyxnQkFBZ0IsQ0FBcUI7SUFDN0MsT0FBTyxDQUFDLGVBQWUsQ0FBQyxDQUFzQjtJQUM5QyxPQUFPLENBQUMsaUJBQWlCLENBQUMsQ0FBc0I7SUFDaEQsb0JBQW9CLENBQUMsRUFBRSw2QkFBNkIsQ0FBQztJQUNyRCxXQUFXLEVBQUcsYUFBYSxDQUFDO0lBQzVCLE9BQU8sQ0FBQyxPQUFPLENBQXFCO0lBQ3BDLE9BQU8sQ0FBQyxVQUFVLENBQWM7SUFDaEMsT0FBTyxDQUFDLG1CQUFtQixDQUFjO0lBQ2xDLFdBQVcsRUFBRyxnQ0FBZ0MsQ0FBQztJQUMvQyxhQUFhLEVBQUcsVUFBVSxDQUFDO0lBRWxDLFlBQ0UsUUFBUSxFQUFFLE1BQU0sRUFDUixzQkFBc0IsRUFBRSxNQUFNLEVBQ3RDLFFBQVEsRUFBRSxVQUFVLEVBQ1osVUFBVSxVQUFPLEVBVzFCO0lBRUQ7Ozs7T0FJRztJQUNHLGtCQUFrQixrQkFxRHZCO0lBRUssS0FBSyxrQkFxSVY7WUFFYSxZQUFZO0lBUzFCLFFBQVEsR0FBSSxDQUFDLDhKQUkyRDtJQUVsRSxRQUFRLGtCQWNiO0lBRUssaUJBQWlCLGtCQTZDdEI7Q0FDRiJ9
55
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZTJlX3Byb3Zlcl90ZXN0LmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvZml4dHVyZXMvZTJlX3Byb3Zlcl90ZXN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sS0FBSyxFQUFFLGtCQUFrQixFQUFFLE1BQU0seUJBQXlCLENBQUM7QUFFbEUsT0FBTyxFQUFFLFlBQVksRUFBRSxVQUFVLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUNyRSxPQUFPLEVBQUUsS0FBSyxNQUFNLEVBQWdCLE1BQU0scUJBQXFCLENBQUM7QUFDaEUsT0FBTyxLQUFLLEVBQUUsU0FBUyxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDdEQsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ2xELE9BQU8sRUFFTCxLQUFLLDZCQUE2QixFQUduQyxNQUFNLGtCQUFrQixDQUFDO0FBRzFCLE9BQU8sS0FBSyxFQUFFLGdDQUFnQyxFQUFFLE1BQU0sMkNBQTJDLENBQUM7QUFJbEcsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLGdDQUFnQyxDQUFDO0FBRS9ELE9BQU8sS0FBSyxFQUFFLDJCQUEyQixFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFDMUUsT0FBTyxLQUFLLEVBQUUsY0FBYyxFQUFFLE1BQU0saUNBQWlDLENBQUM7QUFDdEUsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLDJCQUEyQixDQUFDO0FBTXZELE9BQU8sRUFBRSxjQUFjLEVBQUUsTUFBTSxrQ0FBa0MsQ0FBQztBQUdsRSxPQUFPLEVBQ0wsS0FBSyxlQUFlLEVBUXJCLE1BQU0sWUFBWSxDQUFDO0FBT3BCOzs7OztHQUtHO0FBRUgscUJBQWEsY0FBYztJQUN6QixNQUFNLENBQUMsVUFBVSxTQUFVO0lBQzNCLE1BQU0sQ0FBQyxZQUFZLFNBQVM7SUFDNUIsTUFBTSxDQUFDLGNBQWMsU0FBTztJQUM1QixNQUFNLEVBQUUsTUFBTSxDQUFDO0lBQ2YsTUFBTSxFQUFHLFVBQVUsQ0FBQztJQUNwQixZQUFZLEVBQUcsVUFBVSxDQUFDO0lBQzFCLFFBQVEsRUFBRSxZQUFZLEVBQUUsQ0FBTTtJQUM5QixnQkFBZ0IsRUFBRyxrQkFBa0IsRUFBRSxDQUFDO0lBQ3hDLGVBQWUsRUFBRyxhQUFhLENBQUM7SUFDaEMsdUJBQXVCLEVBQUcsMkJBQTJCLENBQUM7SUFDdEQsUUFBUSxFQUFHLGNBQWMsQ0FBQztJQUMxQixTQUFTLEVBQUcsU0FBUyxDQUFDO0lBQ3RCLGNBQWMsRUFBRyxjQUFjLENBQUM7SUFDaEMsVUFBVSxFQUFHLFVBQVUsQ0FBQztJQUN4QixPQUFPLENBQUMsZ0JBQWdCLENBQXFCO0lBQzdDLE9BQU8sQ0FBQyxlQUFlLENBQUMsQ0FBc0I7SUFDOUMsT0FBTyxDQUFDLGlCQUFpQixDQUFDLENBQXNCO0lBQ2hELG9CQUFvQixDQUFDLEVBQUUsNkJBQTZCLENBQUM7SUFDckQsV0FBVyxFQUFHLGFBQWEsQ0FBQztJQUM1QixPQUFPLEVBQUcsZUFBZSxDQUFDO0lBQzFCLE9BQU8sQ0FBQyxVQUFVLENBQWM7SUFDaEMsT0FBTyxDQUFDLG1CQUFtQixDQUFjO0lBQ2xDLFdBQVcsRUFBRyxnQ0FBZ0MsQ0FBQztJQUMvQyxhQUFhLEVBQUcsVUFBVSxDQUFDO0lBQ2xDLE9BQU8sQ0FBQyxzQkFBc0IsQ0FBUztJQUN2QyxPQUFPLENBQUMsUUFBUSxDQUFhO0lBQzdCLE9BQU8sQ0FBQyxVQUFVLENBQVU7SUFFNUIsWUFBWSxRQUFRLEVBQUUsTUFBTSxFQUFFLHNCQUFzQixFQUFFLE1BQU0sRUFBRSxRQUFRLEVBQUUsVUFBVSxFQUFFLFVBQVUsVUFBTyxFQUtwRztZQUthLGNBQWM7SUF1Q3RCLEtBQUssa0JBZ0pWO1lBRWEsWUFBWTtJQVNwQixRQUFRLGtCQWNiO1lBRWEsU0FBUztDQXFDeEIifQ==
@@ -1 +1 @@
1
- {"version":3,"file":"e2e_prover_test.d.ts","sourceRoot":"","sources":["../../src/fixtures/e2e_prover_test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAElE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACrE,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,qBAAqB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAEL,KAAK,6BAA6B,EAGnC,MAAM,kBAAkB,CAAC;AAG1B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,2CAA2C,CAAC;AAIlG,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAE/D,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAMvD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAGlE,OAAO,EAEL,KAAK,iBAAiB,EAIvB,MAAM,uBAAuB,CAAC;AAU/B;;;;;GAKG;AAEH,qBAAa,cAAc;IA6BvB,OAAO,CAAC,sBAAsB;IAE9B,OAAO,CAAC,UAAU;IA9BpB,MAAM,CAAC,UAAU,SAAU;IAC3B,MAAM,CAAC,YAAY,SAAS;IAC5B,MAAM,CAAC,cAAc,SAAO;IAC5B,OAAO,CAAC,eAAe,CAAmB;IAC1C,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAG,UAAU,CAAC;IACpB,YAAY,EAAG,UAAU,CAAC;IAC1B,QAAQ,EAAE,YAAY,EAAE,CAAM;IAC9B,gBAAgB,EAAG,kBAAkB,EAAE,CAAC;IACxC,eAAe,EAAG,aAAa,CAAC;IAChC,uBAAuB,EAAG,2BAA2B,CAAC;IACtD,QAAQ,EAAG,cAAc,CAAC;IAC1B,SAAS,EAAG,SAAS,CAAC;IACtB,cAAc,EAAG,cAAc,CAAC;IAChC,UAAU,EAAG,UAAU,CAAC;IACxB,OAAO,CAAC,gBAAgB,CAAqB;IAC7C,OAAO,CAAC,eAAe,CAAC,CAAsB;IAC9C,OAAO,CAAC,iBAAiB,CAAC,CAAsB;IAChD,oBAAoB,CAAC,EAAE,6BAA6B,CAAC;IACrD,WAAW,EAAG,aAAa,CAAC;IAC5B,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,UAAU,CAAc;IAChC,OAAO,CAAC,mBAAmB,CAAc;IAClC,WAAW,EAAG,gCAAgC,CAAC;IAC/C,aAAa,EAAG,UAAU,CAAC;IAElC,YACE,QAAQ,EAAE,MAAM,EACR,sBAAsB,EAAE,MAAM,EACtC,QAAQ,EAAE,UAAU,EACZ,UAAU,UAAO,EAW1B;IAED;;;;OAIG;IACG,kBAAkB,kBAqDvB;IAEK,KAAK,kBAqIV;YAEa,YAAY;IAS1B,QAAQ,GAAI,CAAC,8JAI2D;IAElE,QAAQ,kBAcb;IAEK,iBAAiB,kBA6CtB;CACF"}
1
+ {"version":3,"file":"e2e_prover_test.d.ts","sourceRoot":"","sources":["../../src/fixtures/e2e_prover_test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAElE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACrE,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,qBAAqB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAEL,KAAK,6BAA6B,EAGnC,MAAM,kBAAkB,CAAC;AAG1B,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,2CAA2C,CAAC;AAIlG,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAE/D,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAMvD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAGlE,OAAO,EACL,KAAK,eAAe,EAQrB,MAAM,YAAY,CAAC;AAOpB;;;;;GAKG;AAEH,qBAAa,cAAc;IACzB,MAAM,CAAC,UAAU,SAAU;IAC3B,MAAM,CAAC,YAAY,SAAS;IAC5B,MAAM,CAAC,cAAc,SAAO;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAG,UAAU,CAAC;IACpB,YAAY,EAAG,UAAU,CAAC;IAC1B,QAAQ,EAAE,YAAY,EAAE,CAAM;IAC9B,gBAAgB,EAAG,kBAAkB,EAAE,CAAC;IACxC,eAAe,EAAG,aAAa,CAAC;IAChC,uBAAuB,EAAG,2BAA2B,CAAC;IACtD,QAAQ,EAAG,cAAc,CAAC;IAC1B,SAAS,EAAG,SAAS,CAAC;IACtB,cAAc,EAAG,cAAc,CAAC;IAChC,UAAU,EAAG,UAAU,CAAC;IACxB,OAAO,CAAC,gBAAgB,CAAqB;IAC7C,OAAO,CAAC,eAAe,CAAC,CAAsB;IAC9C,OAAO,CAAC,iBAAiB,CAAC,CAAsB;IAChD,oBAAoB,CAAC,EAAE,6BAA6B,CAAC;IACrD,WAAW,EAAG,aAAa,CAAC;IAC5B,OAAO,EAAG,eAAe,CAAC;IAC1B,OAAO,CAAC,UAAU,CAAc;IAChC,OAAO,CAAC,mBAAmB,CAAc;IAClC,WAAW,EAAG,gCAAgC,CAAC;IAC/C,aAAa,EAAG,UAAU,CAAC;IAClC,OAAO,CAAC,sBAAsB,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAa;IAC7B,OAAO,CAAC,UAAU,CAAU;IAE5B,YAAY,QAAQ,EAAE,MAAM,EAAE,sBAAsB,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,UAAO,EAKpG;YAKa,cAAc;IAuCtB,KAAK,kBAgJV;YAEa,YAAY;IASpB,QAAQ,kBAcb;YAEa,SAAS;CAqCxB"}
@@ -15,25 +15,20 @@ import { privateKeyToAddress } from 'viem/accounts';
15
15
  import { TokenSimulator } from '../simulators/token_simulator.js';
16
16
  import { getACVMConfig } from './get_acvm_config.js';
17
17
  import { getBBConfig } from './get_bb_config.js';
18
- import { createSnapshotManager, deployAccounts, publicDeployAccounts } from './snapshot_manager.js';
19
- import { getPrivateKeyFromIndex, getSponsoredFPCAddress, setupPXEAndGetWallet } from './utils.js';
20
- const { E2E_DATA_PATH: dataPath } = process.env;
18
+ import { deployAccounts, getPrivateKeyFromIndex, getSponsoredFPCAddress, publicDeployAccounts, setup, setupPXEAndGetWallet, teardown } from './setup.js';
21
19
  /**
22
20
  * Largely taken from the e2e_token_contract test file. We deploy 2 accounts and a token contract.
23
21
  * However, we then setup a second PXE with a full prover instance.
24
22
  * We configure this instance with all of the accounts and contracts.
25
23
  * We then prove and verify transactions created via this full prover PXE.
26
24
  */ export class FullProverTest {
27
- minNumberOfTxsPerBlock;
28
- realProofs;
29
25
  static TOKEN_NAME = 'USDC';
30
26
  static TOKEN_SYMBOL = 'USD';
31
27
  static TOKEN_DECIMALS = 18n;
32
- snapshotManager;
33
28
  logger;
34
29
  wallet;
35
30
  provenWallet;
36
- accounts;
31
+ accounts = [];
37
32
  deployedAccounts;
38
33
  fakeProofsAsset;
39
34
  fakeProofsAssetInstance;
@@ -41,7 +36,7 @@ const { E2E_DATA_PATH: dataPath } = process.env;
41
36
  aztecNode;
42
37
  aztecNodeAdmin;
43
38
  cheatCodes;
44
- provenComponents;
39
+ provenComponents = [];
45
40
  bbConfigCleanup;
46
41
  acvmConfigCleanup;
47
42
  circuitProofVerifier;
@@ -51,63 +46,62 @@ const { E2E_DATA_PATH: dataPath } = process.env;
51
46
  simulatedProverNode;
52
47
  l1Contracts;
53
48
  proverAddress;
49
+ minNumberOfTxsPerBlock;
50
+ coinbase;
51
+ realProofs;
54
52
  constructor(testName, minNumberOfTxsPerBlock, coinbase, realProofs = true){
53
+ this.logger = createLogger(`e2e:full_prover_test:${testName}`);
55
54
  this.minNumberOfTxsPerBlock = minNumberOfTxsPerBlock;
55
+ this.coinbase = coinbase;
56
56
  this.realProofs = realProofs;
57
- this.accounts = [];
58
- this.provenComponents = [];
59
- this.snapshot = (name, apply, restore = ()=>Promise.resolve())=>this.snapshotManager.snapshot(name, apply, restore);
60
- this.logger = createLogger(`e2e:full_prover_test:${testName}`);
61
- this.snapshotManager = createSnapshotManager(`full_prover_integration/${testName}`, dataPath, {
62
- startProverNode: true,
63
- coinbase
64
- }, {
65
- realVerifier: realProofs
66
- });
67
57
  }
68
58
  /**
69
- * Adds two state shifts to snapshot manager.
70
- * 1. Add 2 accounts.
71
- * 2. Publicly deploy accounts, deploy token contract
72
- */ async applyBaseSnapshots() {
73
- await this.snapshotManager.snapshot('2_accounts', deployAccounts(2, this.logger), ({ deployedAccounts }, { wallet })=>{
74
- this.deployedAccounts = deployedAccounts;
75
- this.accounts = deployedAccounts.map((a)=>a.address);
76
- this.wallet = wallet;
77
- return Promise.resolve();
78
- });
79
- await this.snapshotManager.snapshot('client_prover_integration', async ()=>{
80
- // Create the token contract state.
81
- // Move this account thing to addAccounts above?
82
- this.logger.verbose(`Public deploy accounts...`);
83
- await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
84
- this.logger.verbose(`Deploying TokenContract...`);
85
- const { contract: asset, instance } = await TokenContract.deploy(this.wallet, this.accounts[0], FullProverTest.TOKEN_NAME, FullProverTest.TOKEN_SYMBOL, FullProverTest.TOKEN_DECIMALS).send({
86
- from: this.accounts[0]
87
- }).wait();
88
- this.logger.verbose(`Token deployed to ${asset.address}`);
89
- return {
90
- tokenContractAddress: asset.address,
91
- tokenContractInstance: instance
92
- };
93
- }, async ({ tokenContractAddress, tokenContractInstance })=>{
94
- // Restore the token contract state.
95
- this.fakeProofsAsset = TokenContract.at(tokenContractAddress, this.wallet);
96
- this.fakeProofsAssetInstance = tokenContractInstance;
97
- this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
98
- this.tokenSim = new TokenSimulator(this.fakeProofsAsset, this.wallet, this.accounts[0], this.logger, this.accounts);
99
- expect(await this.fakeProofsAsset.methods.get_admin().simulate({
100
- from: this.accounts[0]
101
- })).toBe(this.accounts[0].toBigInt());
59
+ * Applies base setup: deploys 2 accounts and token contract.
60
+ */ async applyBaseSetup() {
61
+ this.logger.info('Applying base setup: deploying accounts');
62
+ const { deployedAccounts } = await deployAccounts(2, this.logger)({
63
+ wallet: this.context.wallet,
64
+ initialFundedAccounts: this.context.initialFundedAccounts
102
65
  });
66
+ this.deployedAccounts = deployedAccounts;
67
+ this.accounts = deployedAccounts.map((a)=>a.address);
68
+ this.wallet = this.context.wallet;
69
+ this.logger.info('Applying base setup: publicly deploying accounts');
70
+ await publicDeployAccounts(this.wallet, this.accounts.slice(0, 2));
71
+ this.logger.info('Applying base setup: deploying token contract');
72
+ const { contract: asset, instance } = await TokenContract.deploy(this.wallet, this.accounts[0], FullProverTest.TOKEN_NAME, FullProverTest.TOKEN_SYMBOL, FullProverTest.TOKEN_DECIMALS).send({
73
+ from: this.accounts[0]
74
+ }).wait();
75
+ this.logger.verbose(`Token deployed to ${asset.address}`);
76
+ this.fakeProofsAsset = asset;
77
+ this.fakeProofsAssetInstance = instance;
78
+ this.logger.verbose(`Token contract address: ${this.fakeProofsAsset.address}`);
79
+ this.tokenSim = new TokenSimulator(this.fakeProofsAsset, this.wallet, this.accounts[0], this.logger, this.accounts);
80
+ expect(await this.fakeProofsAsset.methods.get_admin().simulate({
81
+ from: this.accounts[0]
82
+ })).toBe(this.accounts[0].toBigInt());
103
83
  }
104
84
  async setup() {
105
- this.context = await this.snapshotManager.setup();
85
+ this.logger.info('Setting up subsystems from fresh');
86
+ this.context = await setup(0, {
87
+ startProverNode: true,
88
+ coinbase: this.coinbase,
89
+ fundSponsoredFPC: true,
90
+ skipAccountDeployment: true,
91
+ l1ContractsArgs: {
92
+ realVerifier: this.realProofs
93
+ }
94
+ });
95
+ await this.applyBaseSetup();
96
+ await this.applyMint();
97
+ this.logger.info(`Enabling proving`, {
98
+ realProofs: this.realProofs
99
+ });
106
100
  // We don't wish to mark as proven automatically, so we set the flag to false
107
101
  this.context.watcher.setIsMarkingAsProven(false);
108
102
  this.simulatedProverNode = this.context.proverNode;
109
103
  ({ aztecNode: this.aztecNode, deployL1ContractsValues: this.l1Contracts, cheatCodes: this.cheatCodes } = this.context);
110
- this.aztecNodeAdmin = this.context.aztecNode;
104
+ this.aztecNodeAdmin = this.context.aztecNodeService;
111
105
  const config = this.context.aztecNodeConfig;
112
106
  const blobClient = await createBlobClientWithFileStores(config, this.logger);
113
107
  // Configure a full prover PXE
@@ -207,13 +201,14 @@ const { E2E_DATA_PATH: dataPath } = process.env;
207
201
  const { prefilledPublicData } = await getGenesisValues(this.context.initialFundedAccounts.map((a)=>a.address).concat(sponsoredFPCAddress));
208
202
  this.proverNode = await createProverNode(proverConfig, {
209
203
  aztecNodeTxProvider: this.aztecNode,
210
- archiver: archiver,
211
- blobClient
204
+ archiver: archiver
212
205
  }, {
213
206
  prefilledPublicData
214
207
  });
215
208
  await this.proverNode.start();
216
- this.logger.warn(`Proofs are now enabled`);
209
+ this.logger.warn(`Proofs are now enabled`, {
210
+ realProofs: this.realProofs
211
+ });
217
212
  return this;
218
213
  }
219
214
  async mintFeeJuice(recipient) {
@@ -232,9 +227,7 @@ const { E2E_DATA_PATH: dataPath } = process.env;
232
227
  hash
233
228
  });
234
229
  }
235
- snapshot;
236
230
  async teardown() {
237
- await this.snapshotManager.teardown();
238
231
  // Cleanup related to the full prover PXEs
239
232
  for(let i = 0; i < this.provenComponents.length; i++){
240
233
  await this.provenComponents[i].teardown();
@@ -244,44 +237,39 @@ const { E2E_DATA_PATH: dataPath } = process.env;
244
237
  await Barretenberg.destroySingleton();
245
238
  await this.bbConfigCleanup?.();
246
239
  await this.acvmConfigCleanup?.();
240
+ await teardown(this.context);
247
241
  }
248
- async applyMintSnapshot() {
249
- await this.snapshotManager.snapshot('mint', async ()=>{
250
- const { fakeProofsAsset: asset, accounts } = this;
251
- const privateAmount = 10000n;
252
- const publicAmount = 10000n;
253
- this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
254
- await asset.methods.mint_to_public(accounts[0], privateAmount + publicAmount).send({
255
- from: accounts[0]
256
- }).wait();
257
- this.logger.verbose(`Transferring ${privateAmount} to private...`);
258
- await asset.methods.transfer_to_private(accounts[0], privateAmount).send({
259
- from: accounts[0]
260
- }).wait();
261
- this.logger.verbose(`Minting complete.`);
262
- return {
263
- amount: publicAmount
264
- };
265
- }, async ({ amount })=>{
266
- const { fakeProofsAsset: asset, accounts: [address], tokenSim } = this;
267
- tokenSim.mintPublic(address, amount);
268
- const publicBalance = await asset.methods.balance_of_public(address).simulate({
269
- from: address
270
- });
271
- this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
272
- expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
273
- tokenSim.mintPrivate(address, amount);
274
- const privateBalance = await asset.methods.balance_of_private(address).simulate({
275
- from: address
276
- });
277
- this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
278
- expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
279
- const totalSupply = await asset.methods.total_supply().simulate({
280
- from: address
281
- });
282
- this.logger.verbose(`Total supply: ${totalSupply}`);
283
- expect(totalSupply).toEqual(tokenSim.totalSupply);
284
- return Promise.resolve();
242
+ async applyMint() {
243
+ this.logger.info('Applying mint setup');
244
+ const { fakeProofsAsset: asset, accounts } = this;
245
+ const privateAmount = 10000n;
246
+ const publicAmount = 10000n;
247
+ this.logger.verbose(`Minting ${privateAmount + publicAmount} publicly...`);
248
+ await asset.methods.mint_to_public(accounts[0], privateAmount + publicAmount).send({
249
+ from: accounts[0]
250
+ }).wait();
251
+ this.logger.verbose(`Transferring ${privateAmount} to private...`);
252
+ await asset.methods.transfer_to_private(accounts[0], privateAmount).send({
253
+ from: accounts[0]
254
+ }).wait();
255
+ this.logger.info(`Minting complete`);
256
+ const { fakeProofsAsset, accounts: [address], tokenSim } = this;
257
+ tokenSim.mintPublic(address, publicAmount);
258
+ const publicBalance = await fakeProofsAsset.methods.balance_of_public(address).simulate({
259
+ from: address
260
+ });
261
+ this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
262
+ expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
263
+ tokenSim.mintPrivate(address, publicAmount);
264
+ const privateBalance = await fakeProofsAsset.methods.balance_of_private(address).simulate({
265
+ from: address
266
+ });
267
+ this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
268
+ expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
269
+ const totalSupply = await fakeProofsAsset.methods.total_supply().simulate({
270
+ from: address
285
271
  });
272
+ this.logger.verbose(`Total supply: ${totalSupply}`);
273
+ expect(totalSupply).toEqual(tokenSim.totalSupply);
286
274
  }
287
275
  }