@aztec/end-to-end 0.0.0-test.1 → 0.0.1-commit.017a351

package/dest/bench/utils.js

@@ -1,23 +1,26 @@
-import { BatchCall, INITIAL_L2_BLOCK_NUM } from '@aztec/aztec.js';
-import { mean, stdDev, timesParallel } from '@aztec/foundation/collection';
-import { randomInt } from '@aztec/foundation/crypto';
-import { BenchmarkingContract } from '@aztec/noir-contracts.js/Benchmarking';
-import { createPXEService } from '@aztec/pxe/server';
-import { writeFileSync } from 'fs';
-import { mkdirpSync } from 'fs-extra';
-import { globSync } from 'glob';
-import { join } from 'path';
+import { AztecAddress } from '@aztec/aztec.js/addresses';
+import { BatchCall, NO_WAIT } from '@aztec/aztec.js/contracts';
+import { waitForTx } from '@aztec/aztec.js/node';
+import { mean, stdDev, times } from '@aztec/foundation/collection';
+import { BenchmarkingContract } from '@aztec/noir-test-contracts.js/Benchmarking';
+import { mkdirSync, writeFileSync } from 'fs';
+import path from 'path';
+import { PIPELINING_SETUP_OPTS } from '../fixtures/fixtures.js';
 import { setup } from '../fixtures/utils.js';
 /**
  * Setup for benchmarks. Initializes a remote node with a single account and deploys a benchmark contract.
  */ export async function benchmarkSetup(opts) {
     const context = await setup(1, {
+        ...PIPELINING_SETUP_OPTS,
         ...opts,
         telemetryConfig: {
             benchmark: true
         }
     });
-    const contract = await BenchmarkingContract.deploy(context.wallet).send().deployed();
+    const defaultAccountAddress = context.accounts[0];
+    const { contract } = await BenchmarkingContract.deploy(context.wallet).send({
+        from: defaultAccountAddress
+    });
     context.logger.info(`Deployed benchmarking contract at ${contract.address}`);
     const sequencer = context.aztecNode.getSequencer();
     const telemetry = context.telemetryClient;
@@ -28,7 +31,13 @@ import { setup } from '../fixtures/utils.js';
         await telemetry.flush();
         const data = telemetry.getMeters();
         const formatted = formatMetricsForGithubBenchmarkAction(data, opts.metrics);
+        if (formatted.length === 0) {
+            throw new Error(`No benchmark data generated. Please review your test setup.`);
+        }
         const benchOutput = opts.benchOutput ?? process.env.BENCH_OUTPUT ?? 'bench.json';
+        mkdirSync(path.dirname(benchOutput), {
+            recursive: true
+        });
         writeFileSync(benchOutput, JSON.stringify(formatted));
         context.logger.info(`Wrote ${data.length} metrics to ${benchOutput}`);
         await origTeardown();
@@ -40,17 +49,20 @@ import { setup } from '../fixtures/utils.js';
         sequencer
     };
 }
+function isMetricDefinition(f) {
+    return 'description' in f;
+}
 function formatMetricsForGithubBenchmarkAction(data, filter) {
-    const allFilters = filter.map((f)=>typeof f === 'string' ? {
-            name: f,
+    const allFilters = filter.map((f)=>isMetricDefinition(f) ? {
+            name: f.name,
             source: f,
             transform: (x)=>x,
-            unit: undefined
+            unit: f.unit
         } : f);
     return data.flatMap((meter)=>{
-        return meter.metrics.filter((metric)=>allFilters.map((f)=>f.source).includes(metric.name)).map((metric)=>[
+        return meter.metrics.filter((metric)=>allFilters.map((f)=>f.source.name).includes(metric.name)).map((metric)=>[
                 metric,
-                allFilters.find((f)=>f.source === metric.name)
+                allFilters.find((f)=>f.source.name === metric.name)
             ]).map(([metric, filter])=>({
                 name: `${meter.name}/${filter.name}`,
                 unit: filter.unit ?? metric.unit ?? 'unknown',
@@ -76,28 +88,6 @@ function getMetricValues(points) {
         };
     }
 }
-/**
- * Creates and returns a directory with the current job name and a random number.
- * @param index - Index to merge into the dir path.
- * @returns A path to a created dir.
- */ export function makeDataDirectory(index) {
-    const testName = expect.getState().currentTestName.split(' ')[0].replaceAll('/', '_');
-    const db = join('data', testName, index.toString(), `${randomInt(99)}`);
-    mkdirpSync(db);
-    return db;
-}
-/**
- * Returns the size in disk of a folder.
- * @param path - Path to the folder.
- * @returns Size in bytes.
- */ export function getFolderSize(path) {
-    return globSync('**', {
-        stat: true,
-        cwd: path,
-        nodir: true,
-        withFileTypes: true
-    }).reduce((accum, file)=>accum + file.size, 0);
-}
 /**
  * Returns a call to the benchmark contract. Each call has a private execution (account entrypoint),
  * a nested private call (create_note), a public call (increment_balance), and a nested public
@@ -108,17 +98,19 @@ function getMetricValues(points) {
  * @param contract - Benchmarking contract.
  * @param heavyPublicCompute - Whether the transactions include heavy public compute (like a big sha256).
  * @returns A BatchCall instance.
- */ export async function makeCall(index, context, contract, heavyPublicCompute) {
-    const owner = context.wallet.getAddress();
-    const sender = owner;
+ */ async function makeCall(index, context, contract, heavyPublicCompute) {
     if (heavyPublicCompute) {
         return new BatchCall(context.wallet, [
-            await contract.methods.sha256_hash_2048(randomBytesAsBigInts(2048)).request()
+            contract.methods.sha256_hash_1024(randomBytesAsBigInts(1024))
         ]);
     } else {
+        // We use random address for the new note owner because we can emit at most UNFINALIZED_TAGGING_INDEXES_WINDOW_LEN
+        // logs for a given sender-recipient-contract tuple.
+        const ownerOfNewNote = await AztecAddress.random();
+        const [ownerOfBalance] = context.accounts;
         return new BatchCall(context.wallet, [
-            await contract.methods.create_note(owner, sender, index + 1).request(),
-            await contract.methods.increment_balance(owner, index + 1).request()
+            contract.methods.create_note(ownerOfNewNote, index + 1),
+            contract.methods.increment_balance(ownerOfBalance, index + 1)
         ]);
     }
 }
@@ -131,40 +123,22 @@ function getMetricValues(points) {
  * @param heavyPublicCompute - Whether the transactions include heavy public compute (like a big sha256).
  * @returns Array of sent txs.
  */ export async function sendTxs(txCount, context, contract, heavyPublicCompute = false) {
-    const calls = await timesParallel(txCount, (index)=>makeCall(index, context, contract, heavyPublicCompute));
+    const calls = await Promise.all(times(txCount, (index)=>makeCall(index, context, contract, heavyPublicCompute)));
     context.logger.info(`Creating ${txCount} txs`);
-    const provenTxs = await Promise.all(calls.map((call)=>call.prove({
-            skipPublicSimulation: true
-        })));
+    const [from] = context.accounts;
     context.logger.info(`Sending ${txCount} txs`);
-    return provenTxs.map((tx)=>tx.send());
+    return Promise.all(calls.map(async (call)=>{
+        const { txHash } = await call.send({
+            from,
+            wait: NO_WAIT
+        });
+        return txHash;
+    }));
 }
 export async function waitTxs(txs, context, txWaitOpts) {
     context.logger.info(`Awaiting ${txs.length} txs to be mined`);
-    await Promise.all(txs.map((tx)=>tx.wait(txWaitOpts)));
-    context.logger.info(`All ${txs.length} txs have been mined`);
-}
-/**
- * Creates a new PXE
- * @param node - Node to connect the pxe to.
- * @param contract - Benchmark contract to add to the pxe.
- * @param startingBlock - First l2 block to process.
- * @returns The new PXE.
- */ export async function createNewPXE(node, contract, startingBlock = INITIAL_L2_BLOCK_NUM) {
-    const l1Contracts = await node.getL1ContractAddresses();
-    const { l1ChainId, protocolVersion } = await node.getNodeInfo();
-    const pxeConfig = {
-        l2StartingBlock: startingBlock,
-        l2BlockPollingIntervalMS: 100,
-        dataDirectory: undefined,
-        dataStoreMapSizeKB: 1024 * 1024,
-        l1Contracts,
-        l1ChainId,
-        version: protocolVersion
-    };
-    const pxe = await createPXEService(node, pxeConfig);
-    await pxe.registerContract(contract);
-    return pxe;
+    await Promise.all(txs.map((txHash)=>waitForTx(context.aztecNode, txHash, txWaitOpts)));
+    context.logger.info(`${txs.length} txs have been mined`);
 }
 function randomBytesAsBigInts(length) {
     return [

package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts

@@ -1,8 +1,17 @@
-import { type AccountWallet, type CompleteAddress, Fr, type Logger, type TxHash } from '@aztec/aztec.js';
-import { DocsExampleContract } from '@aztec/noir-contracts.js/DocsExample';
+import { AztecAddress } from '@aztec/aztec.js/addresses';
+import { Fr } from '@aztec/aztec.js/fields';
+import { type Logger } from '@aztec/aztec.js/log';
+import type { AztecNode } from '@aztec/aztec.js/node';
+import type { TxHash } from '@aztec/aztec.js/tx';
+import type { CheatCodes } from '@aztec/aztec/testing';
 import { TokenBlacklistContract } from '@aztec/noir-contracts.js/TokenBlacklist';
-import { type SubsystemsContext } from '../fixtures/snapshot_manager.js';
+import { GenericProxyContract } from '@aztec/noir-test-contracts.js/GenericProxy';
+import { InvalidAccountContract } from '@aztec/noir-test-contracts.js/InvalidAccount';
+import type { SequencerClient } from '@aztec/sequencer-client';
+import type { AztecNodeDebug } from '@aztec/stdlib/interfaces/client';
+import { type EndToEndContext, type SetupOptions } from '../fixtures/setup.js';
 import { TokenSimulator } from '../simulators/token_simulator.js';
+import type { TestWallet } from '../test-wallet/test_wallet.js';
 export declare class Role {
     private isAdmin;
     private isMinter;
@@ -17,30 +26,31 @@ export declare class Role {
     };
 }
 export declare class BlacklistTokenContractTest {
-    #private;
-    static DELAY: number;
-    private snapshotManager;
+    static CHANGE_ROLES_DELAY: number;
+    context: EndToEndContext;
     logger: Logger;
-    wallets: AccountWallet[];
-    accounts: CompleteAddress[];
+    wallet: TestWallet;
     asset: TokenBlacklistContract;
     tokenSim: TokenSimulator;
-    badAccount: DocsExampleContract;
-    admin: AccountWallet;
-    other: AccountWallet;
-    blacklisted: AccountWallet;
+    badAccount: InvalidAccountContract;
+    authwitProxy: GenericProxyContract;
+    cheatCodes: CheatCodes;
+    sequencer: SequencerClient;
+    aztecNode: AztecNode & AztecNodeDebug;
+    adminAddress: AztecAddress;
+    otherAddress: AztecAddress;
+    blacklistedAddress: AztecAddress;
     constructor(testName: string);
-    mineBlocks(amount?: number): Promise<void>;
+    crossTimestampOfChange(): Promise<void>;
     /**
-     * Adds two state shifts to snapshot manager.
+     * Applies base setup:
      * 1. Add 3 accounts.
      * 2. Publicly deploy accounts, deploy token contract and a "bad account".
      */
-    applyBaseSnapshots(): Promise<void>;
-    setup(): Promise<void>;
-    snapshot: <T>(name: string, apply: (context: SubsystemsContext) => Promise<T>, restore?: (snapshotData: T, context: SubsystemsContext) => Promise<void>) => Promise<void>;
+    applyBaseSetup(): Promise<void>;
+    setup(opts?: Partial<SetupOptions>): Promise<void>;
     teardown(): Promise<void>;
-    addPendingShieldNoteToPXE(contract: TokenBlacklistContract, wallet: AccountWallet, amount: bigint, secretHash: Fr, txHash: TxHash): Promise<void>;
-    applyMintSnapshot(): Promise<void>;
+    addPendingShieldNoteToPXE(contract: TokenBlacklistContract, recipient: AztecAddress, amount: bigint, secretHash: Fr, txHash: TxHash): Promise<void>;
+    applyMint(): Promise<void>;
 }
-//# sourceMappingURL=blacklist_token_contract_test.d.ts.map
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYmxhY2tsaXN0X3Rva2VuX2NvbnRyYWN0X3Rlc3QuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9lMmVfYmxhY2tsaXN0X3Rva2VuX2NvbnRyYWN0L2JsYWNrbGlzdF90b2tlbl9jb250cmFjdF90ZXN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUV6RCxPQUFPLEVBQUUsRUFBRSxFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFDNUMsT0FBTyxFQUFFLEtBQUssTUFBTSxFQUFnQixNQUFNLHFCQUFxQixDQUFDO0FBQ2hFLE9BQU8sS0FBSyxFQUFFLFNBQVMsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQ3RELE9BQU8sS0FBSyxFQUFFLE1BQU0sRUFBRSxNQUFNLG9CQUFvQixDQUFDO0FBQ2pELE9BQU8sS0FBSyxFQUFFLFVBQVUsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBRXZELE9BQU8sRUFBRSxzQkFBc0IsRUFBRSxNQUFNLHlDQUF5QyxDQUFDO0FBQ2pGLE9BQU8sRUFBRSxvQkFBb0IsRUFBRSxNQUFNLDRDQUE0QyxDQUFDO0FBQ2xGLE9BQU8sRUFBRSxzQkFBc0IsRUFBRSxNQUFNLDhDQUE4QyxDQUFDO0FBQ3RGLE9BQU8sS0FBSyxFQUFFLGVBQWUsRUFBRSxNQUFNLHlCQUF5QixDQUFDO0FBQy9ELE9BQU8sS0FBSyxFQUFFLGNBQWMsRUFBRSxNQUFNLGlDQUFpQyxDQUFDO0FBSXRFLE9BQU8sRUFDTCxLQUFLLGVBQWUsRUFDcEIsS0FBSyxZQUFZLEVBTWxCLE1BQU0sc0JBQXNCLENBQUM7QUFDOUIsT0FBTyxFQUFFLGNBQWMsRUFBRSxNQUFNLGtDQUFrQyxDQUFDO0FBQ2xFLE9BQU8sS0FBSyxFQUFFLFVBQVUsRUFBRSxNQUFNLCtCQUErQixDQUFDO0FBRWhFLHFCQUFhLElBQUk7SUFDZixPQUFPLENBQUMsT0FBTyxDQUFTO0lBQ3hCLE9BQU8sQ0FBQyxRQUFRLENBQVM7SUFDekIsT0FBTyxDQUFDLGFBQWEsQ0FBUztJQUU5QixTQUFTLFNBR1I7SUFFRCxVQUFVLFNBR1Q7SUFFRCxlQUFlLFNBR2Q7SUFFRCxZQUFZOzs7O01BSVg7Q0FDRjtBQUVELHFCQUFhLDBCQUEwQjtJQUVyQyxNQUFNLENBQUMsa0JBQWtCLFNBQVM7SUFFbEMsT0FBTyxFQUFHLGVBQWUsQ0FBQztJQUMxQixNQUFNLEVBQUUsTUFBTSxDQUFDO0lBQ2YsTUFBTSxFQUFHLFVBQVUsQ0FBQztJQUNwQixLQUFLLEVBQUcsc0JBQXNCLENBQUM7SUFDL0IsUUFBUSxFQUFHLGNBQWMsQ0FBQztJQUMxQixVQUFVLEVBQUcsc0JBQXNCLENBQUM7SUFDcEMsWUFBWSxFQUFHLG9CQUFvQixDQUFDO0lBQ3BDLFVBQVUsRUFBRyxVQUFVLENBQUM7SUFDeEIsU0FBUyxFQUFHLGVBQWUsQ0FBQztJQUM1QixTQUFTLEVBQUcsU0FBUyxHQUFHLGNBQWMsQ0FBQztJQUV2QyxZQUFZLEVBQUcsWUFBWSxDQUFDO0lBQzVCLFlBQVksRUFBRyxZQUFZLENBQUM7SUFDNUIsa0JBQWtCLEVBQUcsWUFBWSxDQUFDO0lBRWxDLFlBQVksUUFBUSxFQUFFLE1BQU0sRUFFM0I7SUFFSyxzQkFBc0Isa0JBUTNCO0lBRUQ7Ozs7T0FJRztJQUNHLGNBQWMsa0JBOERuQjtJQUVLLEtBQUssQ0FBQyxJQUFJLEdBQUUsT0FBTyxDQUFDLFlBQVksQ0FBTSxpQkFRM0M7SUFFSyxRQUFRLGtCQUViO0lBRUsseUJBQXlCLENBQzdCLFFBQVEsRUFBRSxzQkFBc0IsRUFDaEMsU0FBUyxFQUFFLFlBQVksRUFDdkIsTUFBTSxFQUFFLE1BQU0sRUFDZCxVQUFVLEVBQUUsRUFBRSxFQUNkLE1BQU0sRUFBRSxNQUFNLGlCQWNmO0lBRUssU0FBUyxrQkFtRGQ7Q0FDRiJ9

package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"blacklist_token_contract_test.d.ts","sourceRoot":"","sources":["../../src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts"],"names":[],"mappings":"AACA,OAAO,EACL,KAAK,aAAa,EAClB,KAAK,eAAe,EACpB,EAAE,EACF,KAAK,MAAM,EACX,KAAK,MAAM,EAGZ,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAE3E,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AAIjF,OAAO,EAEL,KAAK,iBAAiB,EAIvB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAIlE,qBAAa,IAAI;IACf,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,aAAa,CAAS;IAE9B,SAAS;IAKT,UAAU;IAKV,eAAe;IAKf,YAAY;;;;;CAKb;AAED,qBAAa,0BAA0B;;IAGrC,MAAM,CAAC,KAAK,SAAK;IAEjB,OAAO,CAAC,eAAe,CAAmB;IAC1C,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,aAAa,EAAE,CAAM;IAC9B,QAAQ,EAAE,eAAe,EAAE,CAAM;IACjC,KAAK,EAAG,sBAAsB,CAAC;IAC/B,QAAQ,EAAG,cAAc,CAAC;IAC1B,UAAU,EAAG,mBAAmB,CAAC;IAEjC,KAAK,EAAG,aAAa,CAAC;IACtB,KAAK,EAAG,aAAa,CAAC;IACtB,WAAW,EAAG,aAAa,CAAC;gBAEhB,QAAQ,EAAE,MAAM;IAKtB,UAAU,CAAC,MAAM,GAAE,MAAyC;IAMlE;;;;OAIG;IACG,kBAAkB;IA0DlB,KAAK;IAIX,QAAQ,YACA,MAAM,SACL,CAAC,OAAO,EAAE,iBAAiB,KAAK,QAAQ,CAAC,CAAC,YACxC,CAAC,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,iBAAiB,KAAK,QAAQ,IAAI,CAAC,KACtE,QAAQ,IAAI,CAAC,CAAwD;IAElE,QAAQ;IAQR,yBAAyB,CAC7B,QAAQ,EAAE,sBAAsB,EAChC,MAAM,EAAE,aAAa,EACrB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,EAAE,EACd,MAAM,EAAE,MAAM;IAgBV,iBAAiB;CAmExB"}
+{"version":3,"file":"blacklist_token_contract_test.d.ts","sourceRoot":"","sources":["../../src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAEzD,OAAO,EAAE,EAAE,EAAE,MAAM,wBAAwB,CAAC;AAC5C,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,qBAAqB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAEvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,MAAM,8CAA8C,CAAC;AACtF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAItE,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,YAAY,EAMlB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAEhE,qBAAa,IAAI;IACf,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,aAAa,CAAS;IAE9B,SAAS,SAGR;IAED,UAAU,SAGT;IAED,eAAe,SAGd;IAED,YAAY;;;;MAIX;CACF;AAED,qBAAa,0BAA0B;IAErC,MAAM,CAAC,kBAAkB,SAAS;IAElC,OAAO,EAAG,eAAe,CAAC;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAG,UAAU,CAAC;IACpB,KAAK,EAAG,sBAAsB,CAAC;IAC/B,QAAQ,EAAG,cAAc,CAAC;IAC1B,UAAU,EAAG,sBAAsB,CAAC;IACpC,YAAY,EAAG,oBAAoB,CAAC;IACpC,UAAU,EAAG,UAAU,CAAC;IACxB,SAAS,EAAG,eAAe,CAAC;IAC5B,SAAS,EAAG,SAAS,GAAG,cAAc,CAAC;IAEvC,YAAY,EAAG,YAAY,CAAC;IAC5B,YAAY,EAAG,YAAY,CAAC;IAC5B,kBAAkB,EAAG,YAAY,CAAC;IAElC,YAAY,QAAQ,EAAE,MAAM,EAE3B;IAEK,sBAAsB,kBAQ3B;IAED;;;;OAIG;IACG,cAAc,kBA8DnB;IAEK,KAAK,CAAC,IAAI,GAAE,OAAO,CAAC,YAAY,CAAM,iBAQ3C;IAEK,QAAQ,kBAEb;IAEK,yBAAyB,CAC7B,QAAQ,EAAE,sBAAsB,EAChC,SAAS,EAAE,YAAY,EACvB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,EAAE,EACd,MAAM,EAAE,MAAM,iBAcf;IAEK,SAAS,kBAmDd;CACF"}

package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js

@@ -1,12 +1,12 @@
-import { getSchnorrWallet } from '@aztec/accounts/schnorr';
-import { Fr, computeSecretHash, createLogger } from '@aztec/aztec.js';
-import { MAX_NOTE_HASHES_PER_TX } from '@aztec/constants';
-import { DocsExampleContract } from '@aztec/noir-contracts.js/DocsExample';
+import { computeSecretHash } from '@aztec/aztec.js/crypto';
+import { Fr } from '@aztec/aztec.js/fields';
+import { createLogger } from '@aztec/aztec.js/log';
 import { TokenBlacklistContract } from '@aztec/noir-contracts.js/TokenBlacklist';
+import { GenericProxyContract } from '@aztec/noir-test-contracts.js/GenericProxy';
+import { InvalidAccountContract } from '@aztec/noir-test-contracts.js/InvalidAccount';
 import { jest } from '@jest/globals';
-import { createSnapshotManager, deployAccounts, publicDeployAccounts } from '../fixtures/snapshot_manager.js';
+import { deployAccounts, ensureAuthRegistryPublished, publicDeployAccounts, setup, teardown } from '../fixtures/setup.js';
 import { TokenSimulator } from '../simulators/token_simulator.js';
-const { E2E_DATA_PATH: dataPath } = process.env;
 export class Role {
     isAdmin = false;
     isMinter = false;
@@ -34,124 +34,155 @@ export class Role {
     }
 }
 export class BlacklistTokenContractTest {
-    // A low delay is really poor ux, but we need to keep it low for the tests to run "quickly".
     // This value MUST match the same value that we have in the contract
-    static DELAY = 2;
-    snapshotManager;
+    static CHANGE_ROLES_DELAY = 86400;
+    context;
     logger;
-    wallets = [];
-    accounts = [];
+    wallet;
     asset;
     tokenSim;
     badAccount;
-    admin;
-    other;
-    blacklisted;
+    authwitProxy;
+    cheatCodes;
+    sequencer;
+    aztecNode;
+    adminAddress;
+    otherAddress;
+    blacklistedAddress;
     constructor(testName){
         this.logger = createLogger(`e2e:e2e_blacklist_token_contract:${testName}`);
-        this.snapshotManager = createSnapshotManager(`e2e_blacklist_token_contract/${testName}`, dataPath);
     }
-    async mineBlocks(amount = BlacklistTokenContractTest.DELAY) {
-        for(let i = 0; i < amount; ++i){
-            await this.asset.methods.get_roles(this.admin.getAddress()).send().wait();
-        }
+    async crossTimestampOfChange() {
+        // Under AUTOMINE_E2E_OPTS, the 86400s warp crosses many epochs without any proofs being
+        // submitted. Mark current pending checkpoints as proven first so the rollup contract's
+        // pruning window doesn't reset the chain tip to genesis (which would make the warp's
+        // own empty-checkpoint propose fail with Rollup__InvalidArchive). See the AutomineSequencer
+        // README "Epoch proving caveat" and the equivalent pattern in lending_simulator.progressSlots.
+        await this.cheatCodes.rollup.markAsProven();
+        await this.cheatCodes.warpL2TimeAtLeastBy(this.aztecNode, BlacklistTokenContractTest.CHANGE_ROLES_DELAY);
     }
     /**
-   * Adds two state shifts to snapshot manager.
+   * Applies base setup:
    * 1. Add 3 accounts.
    * 2. Publicly deploy accounts, deploy token contract and a "bad account".
-   */ async applyBaseSnapshots() {
-        // Adding a timeout of 2 minutes in here such that it is propagated to the underlying tests
-        jest.setTimeout(120_000);
-        await this.snapshotManager.snapshot('3_accounts', deployAccounts(3, this.logger), async ({ deployedAccounts }, { pxe })=>{
-            this.wallets = await Promise.all(deployedAccounts.map((a)=>getSchnorrWallet(pxe, a.address, a.signingKey)));
-            this.admin = this.wallets[0];
-            this.other = this.wallets[1];
-            this.blacklisted = this.wallets[2];
-            this.accounts = this.wallets.map((w)=>w.getCompleteAddress());
-        });
-        await this.snapshotManager.snapshot('e2e_blacklist_token_contract', async ()=>{
-            // Create the token contract state.
-            // Move this account thing to addAccounts above?
-            this.logger.verbose(`Public deploy accounts...`);
-            await publicDeployAccounts(this.wallets[0], this.accounts.slice(0, 3));
-            this.logger.verbose(`Deploying TokenContract...`);
-            this.asset = await TokenBlacklistContract.deploy(this.admin, this.admin.getAddress()).send().deployed();
-            this.logger.verbose(`Token deployed to ${this.asset.address}`);
-            this.logger.verbose(`Deploying bad account...`);
-            this.badAccount = await DocsExampleContract.deploy(this.wallets[0]).send().deployed();
-            this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
-            await this.mineBlocks();
-            return {
-                tokenContractAddress: this.asset.address,
-                badAccountAddress: this.badAccount.address
-            };
-        }, async ({ tokenContractAddress, badAccountAddress })=>{
-            // Restore the token contract state.
-            this.asset = await TokenBlacklistContract.at(tokenContractAddress, this.wallets[0]);
-            this.logger.verbose(`Token contract address: ${this.asset.address}`);
-            this.tokenSim = new TokenSimulator(this.asset, this.wallets[0], this.logger, this.accounts.map((a)=>a.address));
-            this.badAccount = await DocsExampleContract.at(badAccountAddress, this.wallets[0]);
-            this.logger.verbose(`Bad account address: ${this.badAccount.address}`);
-            expect(await this.asset.methods.get_roles(this.admin.getAddress()).simulate()).toEqual(new Role().withAdmin().toNoirStruct());
+   */ async applyBaseSetup() {
+        // Bumped from 2 min: pipelined cadence (~24s/dependent-tx) makes the 3-account deploy plus token/bad-account/
+        // proxy deploys exceed the original window.
+        jest.setTimeout(600_000);
+        this.logger.info('Deploying 3 accounts');
+        const { deployedAccounts } = await deployAccounts(3, this.logger)({
+            wallet: this.context.wallet,
+            initialFundedAccounts: this.context.initialFundedAccounts
         });
+        this.cheatCodes = this.context.cheatCodes;
+        this.aztecNode = this.context.aztecNodeService;
+        this.sequencer = this.context.sequencer;
+        this.wallet = this.context.wallet;
+        this.adminAddress = deployedAccounts[0].address;
+        this.otherAddress = deployedAccounts[1].address;
+        this.blacklistedAddress = deployedAccounts[2].address;
+        this.logger.info('Setting up blacklist token contract');
+        await ensureAuthRegistryPublished(this.wallet, this.adminAddress);
+        // Create the token contract state.
+        this.logger.verbose(`Public deploy accounts...`);
+        await publicDeployAccounts(this.wallet, [
+            this.adminAddress,
+            this.otherAddress,
+            this.blacklistedAddress
+        ]);
+        this.logger.verbose(`Deploying TokenContract...`);
+        ({ contract: this.asset } = await TokenBlacklistContract.deploy(this.wallet, this.adminAddress).send({
+            from: this.adminAddress
+        }));
+        this.logger.verbose(`Token deployed to ${this.asset.address}`);
+        this.logger.verbose(`Deploying bad account...`);
+        ({ contract: this.badAccount } = await InvalidAccountContract.deploy(this.wallet).send({
+            from: this.adminAddress
+        }));
+        this.logger.verbose(`Deployed to ${this.badAccount.address}.`);
+        // Deploy a proxy contract for "on behalf of other" tests. The note owner must be the tx sender
+        // (so their notes are in scope), but msg_sender in the target must differ from the note owner
+        // to trigger authwit validation. The proxy forwards calls so that msg_sender != tx sender.
+        this.logger.verbose(`Deploying generic proxy...`);
+        ({ contract: this.authwitProxy } = await GenericProxyContract.deploy(this.wallet).send({
+            from: this.adminAddress
+        }));
+        this.logger.verbose(`Deployed to ${this.authwitProxy.address}.`);
+        await this.crossTimestampOfChange();
+        this.tokenSim = new TokenSimulator(this.asset, this.wallet, this.adminAddress, this.logger, [
+            this.adminAddress,
+            this.otherAddress,
+            this.blacklistedAddress
+        ]);
+        expect((await this.asset.methods.get_roles(this.adminAddress).simulate({
+            from: this.adminAddress
+        })).result).toEqual(new Role().withAdmin().toNoirStruct());
     }
-    async setup() {
-        await this.snapshotManager.setup();
+    async setup(opts = {}) {
+        this.logger.info('Setting up fresh context');
+        this.context = await setup(0, {
+            ...opts,
+            fundSponsoredFPC: true,
+            skipAccountDeployment: true
+        });
+        await this.applyBaseSetup();
     }
-    snapshot = (name, apply, restore = ()=>Promise.resolve())=>this.snapshotManager.snapshot(name, apply, restore);
     async teardown() {
-        await this.snapshotManager.teardown();
+        await teardown(this.context);
     }
-    #toBoundedVec(arr, maxLen) {
-        return {
-            len: arr.length,
-            storage: arr.concat(new Array(maxLen - arr.length).fill(new Fr(0)))
-        };
-    }
-    async addPendingShieldNoteToPXE(contract, wallet, amount, secretHash, txHash) {
-        const txEffects = await wallet.getTxEffect(txHash);
-        await contract.methods.deliver_transparent_note(contract.address, amount, secretHash, txHash.hash, this.#toBoundedVec(txEffects.data.noteHashes, MAX_NOTE_HASHES_PER_TX), txEffects.data.nullifiers[0], wallet.getAddress()).simulate();
+    async addPendingShieldNoteToPXE(contract, recipient, amount, secretHash, txHash) {
+        const txEffects = await this.aztecNode.getTxEffect(txHash);
+        await contract.methods.process_transparent_note(contract.address, amount, secretHash, txHash.hash, txEffects.data.noteHashes, txEffects.data.nullifiers[0], recipient).simulate({
+            from: recipient
+        });
     }
-    async applyMintSnapshot() {
-        await this.snapshotManager.snapshot('mint', async ()=>{
-            const { asset, accounts, wallets } = this;
-            const amount = 10000n;
-            const adminMinterRole = new Role().withAdmin().withMinter();
-            await this.asset.withWallet(this.admin).methods.update_roles(this.admin.getAddress(), adminMinterRole.toNoirStruct()).send().wait();
-            const blacklistRole = new Role().withBlacklisted();
-            await this.asset.withWallet(this.admin).methods.update_roles(this.blacklisted.getAddress(), blacklistRole.toNoirStruct()).send().wait();
-            await this.mineBlocks(); // This gets us past the block of change
-            expect(await this.asset.methods.get_roles(this.admin.getAddress()).simulate()).toEqual(adminMinterRole.toNoirStruct());
-            this.logger.verbose(`Minting ${amount} publicly...`);
-            await asset.methods.mint_public(accounts[0].address, amount).send().wait();
-            this.logger.verbose(`Minting ${amount} privately...`);
-            const secret = Fr.random();
-            const secretHash = await computeSecretHash(secret);
-            const receipt = await asset.methods.mint_private(amount, secretHash).send().wait();
-            await this.addPendingShieldNoteToPXE(asset, wallets[0], amount, secretHash, receipt.txHash);
-            const txClaim = asset.methods.redeem_shield(accounts[0].address, amount, secret).send();
-            await txClaim.wait({
-                debug: true
-            });
-            this.logger.verbose(`Minting complete.`);
-            return {
-                amount
-            };
-        }, async ({ amount })=>{
-            const { asset, accounts: [{ address }], tokenSim } = this;
-            tokenSim.mintPublic(address, amount);
-            const publicBalance = await asset.methods.balance_of_public(address).simulate();
-            this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
-            expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(address));
-            tokenSim.mintPrivate(address, amount);
-            const privateBalance = await asset.methods.balance_of_private(address).simulate();
-            this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
-            expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(address));
-            const totalSupply = await asset.methods.total_supply().simulate();
-            this.logger.verbose(`Total supply: ${totalSupply}`);
-            expect(totalSupply).toEqual(tokenSim.totalSupply);
-            return Promise.resolve();
+    async applyMint() {
+        this.logger.info('Applying mint setup');
+        const { asset, tokenSim } = this;
+        const amount = 10000n;
+        const adminMinterRole = new Role().withAdmin().withMinter();
+        await this.asset.methods.update_roles(this.adminAddress, adminMinterRole.toNoirStruct()).send({
+            from: this.adminAddress
+        });
+        const blacklistRole = new Role().withBlacklisted();
+        await this.asset.methods.update_roles(this.blacklistedAddress, blacklistRole.toNoirStruct()).send({
+            from: this.adminAddress
+        });
+        await this.crossTimestampOfChange();
+        expect((await this.asset.methods.get_roles(this.adminAddress).simulate({
+            from: this.adminAddress
+        })).result).toEqual(adminMinterRole.toNoirStruct());
+        this.logger.verbose(`Minting ${amount} publicly...`);
+        await asset.methods.mint_public(this.adminAddress, amount).send({
+            from: this.adminAddress
+        });
+        this.logger.verbose(`Minting ${amount} privately...`);
+        const secret = Fr.random();
+        const secretHash = await computeSecretHash(secret);
+        const { receipt } = await asset.methods.mint_private(amount, secretHash).send({
+            from: this.adminAddress
+        });
+        await this.addPendingShieldNoteToPXE(asset, this.adminAddress, amount, secretHash, receipt.txHash);
+        await asset.methods.redeem_shield(this.adminAddress, amount, secret).send({
+            from: this.adminAddress
+        });
+        this.logger.verbose(`Minting complete.`);
+        tokenSim.mintPublic(this.adminAddress, amount);
+        const { result: publicBalance } = await asset.methods.balance_of_public(this.adminAddress).simulate({
+            from: this.adminAddress
+        });
+        this.logger.verbose(`Public balance of wallet 0: ${publicBalance}`);
+        expect(publicBalance).toEqual(this.tokenSim.balanceOfPublic(this.adminAddress));
+        tokenSim.mintPrivate(this.adminAddress, amount);
+        const { result: privateBalance } = await asset.methods.balance_of_private(this.adminAddress).simulate({
+            from: this.adminAddress
+        });
+        this.logger.verbose(`Private balance of wallet 0: ${privateBalance}`);
+        expect(privateBalance).toEqual(tokenSim.balanceOfPrivate(this.adminAddress));
+        const { result: totalSupply } = await asset.methods.total_supply().simulate({
+            from: this.adminAddress
         });
+        this.logger.verbose(`Total supply: ${totalSupply}`);
+        expect(totalSupply).toEqual(tokenSim.totalSupply);
     }
 }