@aztec/end-to-end 0.0.0-test.0 → 0.0.1-commit.001888fc

package/dest/shared/submit-transactions.js

@@ -1,23 +1,22 @@
-import { getSchnorrAccount } from '@aztec/accounts/schnorr';
-import { Fr, GrumpkinScalar, TxStatus } from '@aztec/aztec.js';
+import { NO_WAIT } from '@aztec/aztec.js/contracts';
+import { Fr, GrumpkinScalar } from '@aztec/aztec.js/fields';
+import { TxStatus } from '@aztec/aztec.js/tx';
 import { times } from '@aztec/foundation/collection';
-// submits a set of transactions to the provided Private eXecution Environment (PXE)
-export const submitTxsTo = async (pxe, numTxs, wallet, logger)=>{
-    const txs = [];
+// submits a set of transactions to the provided Wallet
+export const submitTxsTo = async (wallet, submitter, numTxs, logger)=>{
+    const txHashes = [];
     await Promise.all(times(numTxs, async ()=>{
-        const accountManager = await getSchnorrAccount(pxe, Fr.random(), GrumpkinScalar.random(), Fr.random());
-        const tx = accountManager.deploy({
-            deployWallet: wallet
+        const accountManager = await wallet.createSchnorrAccount(Fr.random(), Fr.random(), GrumpkinScalar.random());
+        const deployMethod = await accountManager.getDeployMethod();
+        const { txHash } = await deployMethod.send({
+            from: submitter,
+            wait: NO_WAIT
         });
-        const txHash = await tx.getTxHash();
         logger.info(`Tx sent with hash ${txHash}`);
-        const receipt = await tx.getReceipt();
-        expect(receipt).toEqual(expect.objectContaining({
-            status: TxStatus.PENDING,
-            error: ''
-        }));
+        const receipt = await wallet.getTxReceipt(txHash);
+        expect(receipt.status).toBe(TxStatus.PENDING);
         logger.info(`Receipt received for ${txHash}`);
-        txs.push(tx);
+        txHashes.push(txHash);
     }));
-    return txs;
+    return txHashes;
 };

package/dest/shared/uniswap_l1_l2.d.ts

@@ -1,25 +1,3 @@
-import { type AccountWallet, type AztecNode, type CheatCodes, type Logger, type PXE } from '@aztec/aztec.js';
-import { type DeployL1ContractsReturnType, type ViemPublicClient, type ViemWalletClient } from '@aztec/ethereum';
-/** Objects to be returned by the uniswap setup function */
-export type UniswapSetupContext = {
-    /** Aztec Node instance */
-    aztecNode: AztecNode;
-    /** The Private eXecution Environment (PXE). */
-    pxe: PXE;
-    /** Logger instance named as the current test. */
-    logger: Logger;
-    /** Viem Public client instance. */
-    publicClient: ViemPublicClient;
-    /** Viem Wallet Client instance. */
-    walletClient: ViemWalletClient;
-    /** The owner wallet. */
-    ownerWallet: AccountWallet;
-    /** The sponsor wallet. */
-    sponsorWallet: AccountWallet;
-    /**  */
-    deployL1ContractsValues: DeployL1ContractsReturnType;
-    /** Cheat codes instance. */
-    cheatCodes: CheatCodes;
-};
-export declare const uniswapL1L2TestSuite: (setup: () => Promise<UniswapSetupContext>, cleanup: () => Promise<void>, expectedForkBlockNumber?: number) => void;
-//# sourceMappingURL=uniswap_l1_l2.d.ts.map
+import { type EndToEndContext } from '../fixtures/utils.js';
+export declare const uniswapL1L2TestSuite: (setup: () => Promise<EndToEndContext>, cleanup: () => Promise<void>, expectedForkBlockNumber?: number) => void;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidW5pc3dhcF9sMV9sMi5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3NoYXJlZC91bmlzd2FwX2wxX2wyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQXVCQSxPQUFPLEVBQUUsS0FBSyxlQUFlLEVBQW1DLE1BQU0sc0JBQXNCLENBQUM7QUFhN0YsZUFBTyxNQUFNLG9CQUFvQixpSEF1K0JoQyxDQUFDIn0=

package/dest/shared/uniswap_l1_l2.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"uniswap_l1_l2.d.ts","sourceRoot":"","sources":["../../src/shared/uniswap_l1_l2.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,aAAa,EAElB,KAAK,SAAS,EACd,KAAK,UAAU,EAGf,KAAK,MAAM,EACX,KAAK,GAAG,EAGT,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,KAAK,2BAA2B,EAChC,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EAGtB,MAAM,iBAAiB,CAAC;AAqBzB,2DAA2D;AAC3D,MAAM,MAAM,mBAAmB,GAAG;IAChC,0BAA0B;IAC1B,SAAS,EAAE,SAAS,CAAC;IACrB,+CAA+C;IAC/C,GAAG,EAAE,GAAG,CAAC;IACT,iDAAiD;IACjD,MAAM,EAAE,MAAM,CAAC;IACf,mCAAmC;IACnC,YAAY,EAAE,gBAAgB,CAAC;IAC/B,mCAAmC;IACnC,YAAY,EAAE,gBAAgB,CAAC;IAC/B,wBAAwB;IACxB,WAAW,EAAE,aAAa,CAAC;IAC3B,0BAA0B;IAC1B,aAAa,EAAE,aAAa,CAAC;IAC7B,OAAO;IACP,uBAAuB,EAAE,2BAA2B,CAAC;IACrD,4BAA4B;IAC5B,UAAU,EAAE,UAAU,CAAC;CACxB,CAAC;AAGF,eAAO,MAAM,oBAAoB,UACxB,MAAM,QAAQ,mBAAmB,CAAC,WAChC,MAAM,QAAQ,IAAI,CAAC,2CA89B7B,CAAC"}
+{"version":3,"file":"uniswap_l1_l2.d.ts","sourceRoot":"","sources":["../../src/shared/uniswap_l1_l2.ts"],"names":[],"mappings":"AAuBA,OAAO,EAAE,KAAK,eAAe,EAAmC,MAAM,sBAAsB,CAAC;AAa7F,eAAO,MAAM,oBAAoB,iHAu+BhC,CAAC"}

package/dest/shared/uniswap_l1_l2.js

@@ -1,11 +1,20 @@
-import { AztecAddress, EthAddress, Fr, computeAuthWitMessageHash, generateClaimSecret } from '@aztec/aztec.js';
-import { deployL1Contract, extractEvent } from '@aztec/ethereum';
-import { sha256ToField } from '@aztec/foundation/crypto';
-import { InboxAbi, RollupAbi, UniswapPortalAbi, UniswapPortalBytecode } from '@aztec/l1-artifacts';
+import { AztecAddress, EthAddress } from '@aztec/aztec.js/addresses';
+import { computeAuthWitMessageHash } from '@aztec/aztec.js/authorization';
+import { waitForProven } from '@aztec/aztec.js/contracts';
+import { generateClaimSecret } from '@aztec/aztec.js/ethereum';
+import { Fr } from '@aztec/aztec.js/fields';
+import { RollupContract } from '@aztec/ethereum/contracts';
+import { deployL1Contract } from '@aztec/ethereum/deploy-l1-contract';
+import { extractEvent } from '@aztec/ethereum/utils';
+import { EpochNumber } from '@aztec/foundation/branded-types';
+import { sha256ToField } from '@aztec/foundation/crypto/sha256';
+import { InboxAbi, UniswapPortalAbi, UniswapPortalBytecode } from '@aztec/l1-artifacts';
 import { UniswapContract } from '@aztec/noir-contracts.js/Uniswap';
+import { computeL2ToL1MessageHash } from '@aztec/stdlib/hash';
+import { computeL2ToL1MembershipWitness } from '@aztec/stdlib/messaging';
 import { jest } from '@jest/globals';
 import { getContract, parseEther, toFunctionSelector } from 'viem';
-import { ensureAccountsPubliclyDeployed } from '../fixtures/utils.js';
+import { ensureAccountContractsPublished } from '../fixtures/utils.js';
 import { CrossChainTestHarness } from './cross_chain_test_harness.js';
 // PSA: This tests works on forked mainnet. There is a dump of the data in `dumpedState` such that we
 // don't need to burn through RPC requests.
@@ -13,26 +22,20 @@ import { CrossChainTestHarness } from './cross_chain_test_harness.js';
 // To start an actual fork, use the command:
 // anvil --fork-url https://mainnet.infura.io/v3/9928b52099854248b3a096be07a6b23c --fork-block-number 17514288 --chain-id 31337
 // For CI, this is configured in `run_tests.sh` and `docker-compose-images.yml`
-// docs:start:uniswap_l1_l2_test_setup_const
 const TIMEOUT = 360_000;
-// docs:end:uniswap_l1_l2_test_setup_const
 export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 17514288)=>{
-    // docs:start:uniswap_l1_l2_test_beforeAll
     describe('uniswap_trade_on_l1_from_l2', ()=>{
         jest.setTimeout(TIMEOUT);
         const WETH9_ADDRESS = EthAddress.fromString('0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2');
         const DAI_ADDRESS = EthAddress.fromString('0x6B175474E89094C44Da98b954EedeAC495271d0F');
         let aztecNode;
-        let pxe;
         let logger;
-        let walletClient;
-        let publicClient;
-        let ownerWallet;
+        let l1Client;
+        let wallet;
         let ownerAddress;
         let ownerEthAddress;
         // does transactions on behalf of owner on Aztec:
-        let sponsorWallet;
-        // let sponsorAddress: AztecAddress;
+        let sponsorAddress;
         let daiCrossChainHarness;
         let wethCrossChainHarness;
         let deployL1ContractsValues;
@@ -44,54 +47,54 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
         const uniswapFeeTier = 3000n;
         const minimumOutputAmount = 0n;
         let cheatCodes;
+        let version;
         beforeAll(async ()=>{
-            ({ aztecNode, pxe, logger, publicClient, walletClient, ownerWallet, sponsorWallet, deployL1ContractsValues, cheatCodes } = await setup());
-            if (Number(await publicClient.getBlockNumber()) < expectedForkBlockNumber) {
+            const t = await setup();
+            ({ aztecNode, logger, deployL1ContractsValues, cheatCodes, wallet, accounts: [ownerAddress, sponsorAddress] } = t);
+            l1Client = deployL1ContractsValues.l1Client;
+            t.watcher.setIsMarkingAsProven(false);
+            if (Number(await l1Client.getBlockNumber()) < expectedForkBlockNumber) {
                 throw new Error('This test must be run on a fork of mainnet with the expected fork block');
             }
-            rollup = getContract({
-                address: deployL1ContractsValues.l1ContractAddresses.rollupAddress.toString(),
-                abi: RollupAbi,
-                client: deployL1ContractsValues.walletClient
-            });
-            ownerAddress = ownerWallet.getAddress();
-            // sponsorAddress = sponsorWallet.getAddress();
-            ownerEthAddress = EthAddress.fromString((await walletClient.getAddresses())[0]);
-            await ensureAccountsPubliclyDeployed(ownerWallet, [
-                ownerWallet,
-                sponsorWallet
+            rollup = new RollupContract(deployL1ContractsValues.l1Client, deployL1ContractsValues.l1ContractAddresses.rollupAddress);
+            version = Number(await rollup.getVersion());
+            ownerEthAddress = EthAddress.fromString((await l1Client.getAddresses())[0]);
+            await ensureAccountContractsPublished(wallet, [
+                ownerAddress,
+                sponsorAddress
             ]);
             logger.info('Deploying DAI Portal, initializing and deploying l2 contract...');
-            daiCrossChainHarness = await CrossChainTestHarness.new(aztecNode, pxe, publicClient, walletClient, ownerWallet, logger, DAI_ADDRESS);
+            daiCrossChainHarness = await CrossChainTestHarness.new(aztecNode, deployL1ContractsValues.l1Client, wallet, ownerAddress, logger, DAI_ADDRESS);
             logger.info('Deploying WETH Portal, initializing and deploying l2 contract...');
-            wethCrossChainHarness = await CrossChainTestHarness.new(aztecNode, pxe, publicClient, walletClient, ownerWallet, logger, WETH9_ADDRESS);
+            wethCrossChainHarness = await CrossChainTestHarness.new(aztecNode, l1Client, wallet, ownerAddress, logger, WETH9_ADDRESS);
             logger.info('Deploy Uniswap portal on L1 and L2...');
-            uniswapPortalAddress = await deployL1Contract(walletClient, publicClient, UniswapPortalAbi, UniswapPortalBytecode).then(({ address })=>address);
+            uniswapPortalAddress = await deployL1Contract(l1Client, UniswapPortalAbi, UniswapPortalBytecode).then(({ address })=>address);
             uniswapPortal = getContract({
                 address: uniswapPortalAddress.toString(),
                 abi: UniswapPortalAbi,
-                client: walletClient
+                client: l1Client
             });
             // deploy l2 uniswap contract and attach to portal
-            uniswapL2Contract = await UniswapContract.deploy(ownerWallet, uniswapPortalAddress).send().deployed();
-            const registryAddress = (await pxe.getNodeInfo()).l1ContractAddresses.registryAddress;
+            ({ contract: uniswapL2Contract } = await UniswapContract.deploy(wallet, uniswapPortalAddress).send({
+                from: ownerAddress
+            }));
+            const registryAddress = (await aztecNode.getNodeInfo()).l1ContractAddresses.registryAddress;
             await uniswapPortal.write.initialize([
                 registryAddress.toString(),
                 uniswapL2Contract.address.toString()
             ], {});
             // Give me some WETH so I can deposit to L2 and do the swap...
             logger.info('Getting some weth');
-            const hash = await walletClient.sendTransaction({
+            const hash = await l1Client.sendTransaction({
                 to: WETH9_ADDRESS.toString(),
                 value: parseEther('1000')
             });
-            await publicClient.waitForTransactionReceipt({
+            await l1Client.waitForTransactionReceipt({
                 hash
             });
             const wethBalance = await wethCrossChainHarness.getL1BalanceOf(ownerEthAddress);
             expect(wethBalance).toBe(parseEther('1000'));
         });
-        // docs:end:uniswap_l1_l2_test_beforeAll
         afterAll(async ()=>{
             await cleanup();
         });
@@ -114,14 +117,19 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // 3. Owner gives uniswap approval to transfer the funds to public to self on its behalf
             logger.info('Approving uniswap to transfer funds to public to self on my behalf');
             const nonceForWETHTransferToPublicApproval = new Fr(1n);
-            await ownerWallet.createAuthWit({
+            const transferToPublicAuhtwit = await wallet.createAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval)
             });
             // 4. Swap on L1 - sends L2 to L1 message to withdraw WETH to L1 and another message to swap assets.
             logger.info('Withdrawing weth to L1 and sending message to swap to dai');
             const [secretForDepositingSwappedDai, secretHashForDepositingSwappedDai] = await generateClaimSecret();
-            const l2UniswapInteractionReceipt = await uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, secretHashForDepositingSwappedDai, ownerEthAddress).send().wait();
+            const { receipt: l2UniswapInteractionReceipt } = await uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, secretHashForDepositingSwappedDai, ownerEthAddress).send({
+                from: ownerAddress,
+                authWitnesses: [
+                    transferToPublicAuhtwit
+                ]
+            });
             const swapPrivateFunction = 'swap_private(address,uint256,uint24,address,uint256,bytes32,address)';
             const swapPrivateContent = sha256ToField([
                 Buffer.from(toFunctionSelector(swapPrivateFunction).substring(2), 'hex'),
@@ -133,44 +141,52 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 secretHashForDepositingSwappedDai,
                 ownerEthAddress.toBuffer32()
             ]);
-            const swapPrivateLeaf = sha256ToField([
-                uniswapL2Contract.address,
-                new Fr(1),
-                EthAddress.fromString(uniswapPortal.address).toBuffer32(),
-                new Fr(publicClient.chain.id),
-                swapPrivateContent
-            ]);
+            const swapPrivateLeaf = computeL2ToL1MessageHash({
+                l2Sender: uniswapL2Contract.address,
+                l1Recipient: EthAddress.fromString(uniswapPortal.address),
+                content: swapPrivateContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
             const withdrawContent = sha256ToField([
                 Buffer.from(toFunctionSelector('withdraw(address,uint256,address)').substring(2), 'hex'),
                 uniswapPortalAddress.toBuffer32(),
                 new Fr(wethAmountToBridge),
                 uniswapPortalAddress.toBuffer32()
             ]);
-            const withdrawLeaf = sha256ToField([
-                wethCrossChainHarness.l2Bridge.address,
-                new Fr(1),
-                wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
-                new Fr(publicClient.chain.id),
-                withdrawContent
-            ]);
+            const withdrawLeaf = computeL2ToL1MessageHash({
+                l2Sender: wethCrossChainHarness.l2Bridge.address,
+                l1Recipient: wethCrossChainHarness.tokenPortalAddress,
+                content: withdrawContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
             // ensure that user's funds were burnt
             await wethCrossChainHarness.expectPrivateBalanceOnL2(ownerAddress, wethL2BalanceBeforeSwap - wethAmountToBridge);
             // ensure that uniswap contract didn't eat the funds.
             await wethCrossChainHarness.expectPublicBalanceOnL2(uniswapL2Contract.address, 0n);
-            // Since the outbox is only consumable when the block is proven, we need to set the block to be proven
-            await cheatCodes.rollup.markAsProven(await rollup.read.getPendingBlockNumber());
+            // Since the outbox is only consumable when the epoch is proven, we need to advance to the next epoch.
+            const swapResult = await computeL2ToL1MembershipWitness(aztecNode, swapPrivateLeaf, l2UniswapInteractionReceipt.txHash);
+            const { epochNumber: epoch } = swapResult;
+            await cheatCodes.rollup.advanceToEpoch(EpochNumber(epoch + 1));
+            await waitForProven(aztecNode, l2UniswapInteractionReceipt, {
+                provenTimeout: 300
+            });
             // 5. Consume L2 to L1 message by calling uniswapPortal.swap_private()
             logger.info('Execute withdraw and swap on the uniswapPortal!');
             const daiL1BalanceOfPortalBeforeSwap = await daiCrossChainHarness.getL1BalanceOf(daiCrossChainHarness.tokenPortalAddress);
-            const [swapPrivateL2MessageIndex, swapPrivateSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(l2UniswapInteractionReceipt.blockNumber, swapPrivateLeaf);
-            const [withdrawL2MessageIndex, withdrawSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(l2UniswapInteractionReceipt.blockNumber, withdrawLeaf);
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, withdrawLeaf, l2UniswapInteractionReceipt.txHash);
+            const swapPrivateL2MessageIndex = swapResult.leafIndex;
+            const swapPrivateSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
-                _l2BlockNumber: BigInt(l2UniswapInteractionReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
                 _path: withdrawSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
             const swapPrivateMessageMetadata = {
-                _l2BlockNumber: BigInt(l2UniswapInteractionReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(swapPrivateL2MessageIndex),
                 _path: swapPrivateSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
@@ -188,7 +204,7 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 ]
             ];
             // this should also insert a message into the inbox.
-            const txReceipt = await daiCrossChainHarness.publicClient.waitForTransactionReceipt({
+            const txReceipt = await daiCrossChainHarness.l1Client.waitForTransactionReceipt({
                 hash: await uniswapPortal.write.swapPrivate(swapArgs)
             });
             // We get the msg leaf from event so that we can later wait for it to be available for consumption
@@ -297,9 +313,9 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
         //       ownerEthAddress,
         //       nonceForSwap,
         //     );
-        //   await ownerWallet.setPublicAuthWit({ caller: sponsorAddress, action }, true).send().wait();
+        //   await ownerWallet.setPublicAuthWit({ caller: sponsorAddress, action }, true).send();
         //   // 4.2 Call swap_public from user2 on behalf of owner
-        //   const uniswapL2Interaction = await action.send().wait();
+        //   const uniswapL2Interaction = await action.send();
         //   const swapPublicContent = sha256ToField([
         //     Buffer.from(
         //       toFunctionSelector('swap_public(address,uint256,uint24,address,uint256,bytes32,bytes32,address)').substring(
@@ -352,14 +368,14 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
         //     withdrawLeaf,
         //   );
         //   const withdrawMessageMetadata = {
-        //     _l2BlockNumber: BigInt(uniswapL2Interaction.blockNumber!),
+        //     _epoch: epoch,
         //     _leafIndex: BigInt(withdrawL2MessageIndex),
         //     _path: withdrawSiblingPath
         //       .toBufferArray()
         //       .map((buf: Buffer) => `0x${buf.toString('hex')}`) as readonly `0x${string}`[],
         //   };
         //   const swapPrivateMessageMetadata = {
-        //     _l2BlockNumber: BigInt(uniswapL2Interaction.blockNumber!),
+        //     _epoch: epoch,
         //     _leafIndex: BigInt(swapPrivateL2MessageIndex),
         //     _path: swapPrivateSiblingPath
         //       .toBufferArray()
@@ -432,12 +448,11 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             const nonceForWETHTransferToPublicApproval = new Fr(2n);
             const expectedMessageHash = await computeAuthWitMessageHash({
                 caller: uniswapL2Contract.address,
-                action: wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval)
-            }, {
-                chainId: ownerWallet.getChainId(),
-                version: ownerWallet.getVersion()
-            });
-            await expect(uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, Fr.random(), ownerEthAddress).prove()).rejects.toThrow(`Unknown auth witness for message hash ${expectedMessageHash.toString()}`);
+                call: await wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval).getFunctionCall()
+            }, await wallet.getChainInfo());
+            await expect(uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, Fr.random(), ownerEthAddress).simulate({
+                from: ownerAddress
+            })).rejects.toThrow(`Unknown auth witness for message hash ${expectedMessageHash.toString()}`);
         });
         it("can't swap if user passes a token different to what the bridge tracks", async ()=>{
             // 1. give user private funds on L2:
@@ -446,13 +461,18 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // 2. owner gives uniswap approval to transfer the funds to public:
             logger.info('Approving uniswap to transfer funds to public to self on my behalf');
             const nonceForWETHTransferToPublicApproval = new Fr(3n);
-            await ownerWallet.createAuthWit({
+            const transferToPublicAuthwith = await wallet.createAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval)
             });
             // 3. Swap but send the wrong token address
             logger.info('Swap but send the wrong token address');
-            await expect(uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, daiCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, Fr.random(), ownerEthAddress).prove()).rejects.toThrow('Assertion failed: input_asset address is not the same as seen in the bridge contract');
+            await expect(uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, daiCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, Fr.random(), ownerEthAddress).send({
+                from: ownerAddress,
+                authWitnesses: [
+                    transferToPublicAuthwith
+                ]
+            })).rejects.toThrow('Assertion failed: input_asset address is not the same as seen in the bridge contract');
         });
         // edge cases for public flow:
         it("I don't need approval to call swap_public if I'm swapping on my own behalf", async ()=>{
@@ -460,14 +480,16 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             await wethCrossChainHarness.mintTokensPublicOnL2(wethAmountToBridge);
             // 2. Give approval to uniswap to transfer funds to itself
             const nonceForWETHTransferApproval = new Fr(2n);
-            const validateActionInteraction = await ownerWallet.setPublicAuthWit({
+            const validateActionInteraction = await wallet.setPublicAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_in_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferApproval)
             }, true);
-            await validateActionInteraction.send().wait();
+            await validateActionInteraction.send();
             // No approval to call `swap` but should work even without it:
             const [_, secretHashForDepositingSwappedDai] = await generateClaimSecret();
-            await uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, Fr.ZERO).send().wait();
+            await uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, Fr.ZERO).send({
+                from: ownerAddress
+            });
             // check weth balance of owner on L2 (we first bridged `wethAmountToBridge` into L2 and now withdrew it!)
             await wethCrossChainHarness.expectPublicBalanceOnL2(ownerAddress, 0n);
         });
@@ -477,23 +499,27 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             const nonceForWETHTransferApproval = new Fr(3n);
             const nonceForSwap = new Fr(3n);
             const secretHashForDepositingSwappedDai = new Fr(4n);
-            const action = uniswapL2Contract.withWallet(sponsorWallet).methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, nonceForSwap);
-            const validateActionInteraction = await ownerWallet.setPublicAuthWit({
+            const action = uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, nonceForSwap);
+            const validateActionInteraction = await wallet.setPublicAuthWit(ownerAddress, {
                 caller: approvedUser,
                 action
             }, true);
-            await validateActionInteraction.send().wait();
-            await expect(action.simulate()).rejects.toThrow(/unauthorized/);
+            await validateActionInteraction.send();
+            await expect(action.simulate({
+                from: sponsorAddress
+            })).rejects.toThrow(/unauthorized/);
         });
         it("uniswap can't pull funds without transfer approval", async ()=>{
             // swap should fail since no transfer approval to uniswap:
             const nonceForWETHTransferApproval = new Fr(4n);
-            const validateActionInteraction = await ownerWallet.setPublicAuthWit({
+            const validateActionInteraction = await wallet.setPublicAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_in_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferApproval)
             }, true);
-            await validateActionInteraction.send().wait();
-            await expect(uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, new Fr(420), uniswapFeeTier, minimumOutputAmount, ownerAddress, Fr.random(), ownerEthAddress, Fr.ZERO).simulate()).rejects.toThrow(/unauthorized/);
+            await validateActionInteraction.send();
+            await expect(uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, new Fr(420), uniswapFeeTier, minimumOutputAmount, ownerAddress, Fr.random(), ownerEthAddress, Fr.ZERO).simulate({
+                from: ownerAddress
+            })).rejects.toThrow(/unauthorized/);
         });
         // tests when trying to mix private and public flows:
         it("can't call swap_public on L1 if called swap_private on L2", async ()=>{
@@ -503,7 +529,7 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // Owner gives uniswap approval to transfer the funds to public to self on its behalf
             logger.info('Approving uniswap to transfer the funds to public to self on my behalf');
             const nonceForWETHTransferToPublicApproval = new Fr(4n);
-            await ownerWallet.createAuthWit({
+            const transferToPublicAuhtwit = await wallet.createAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval)
             });
@@ -511,7 +537,12 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // Swap
             logger.info('Withdrawing weth to L1 and sending message to swap to dai');
             const [, secretHashForDepositingSwappedDai] = await generateClaimSecret();
-            const withdrawReceipt = await uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, secretHashForDepositingSwappedDai, ownerEthAddress).send().wait();
+            const { receipt: withdrawReceipt } = await uniswapL2Contract.methods.swap_private(wethCrossChainHarness.l2Token.address, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferToPublicApproval, uniswapFeeTier, minimumOutputAmount, secretHashForDepositingSwappedDai, ownerEthAddress).send({
+                from: ownerAddress,
+                authWitnesses: [
+                    transferToPublicAuhtwit
+                ]
+            });
             const swapPrivateContent = sha256ToField([
                 Buffer.from(toFunctionSelector('swap_private(address,uint256,uint24,address,uint256,bytes32,address)').substring(2), 'hex'),
                 wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
@@ -522,42 +553,50 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 secretHashForDepositingSwappedDai,
                 ownerEthAddress.toBuffer32()
             ]);
-            const swapPrivateLeaf = sha256ToField([
-                uniswapL2Contract.address,
-                new Fr(1),
-                EthAddress.fromString(uniswapPortal.address).toBuffer32(),
-                new Fr(publicClient.chain.id),
-                swapPrivateContent
-            ]);
+            const swapPrivateLeaf = computeL2ToL1MessageHash({
+                l2Sender: uniswapL2Contract.address,
+                l1Recipient: EthAddress.fromString(uniswapPortal.address),
+                content: swapPrivateContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
             const withdrawContent = sha256ToField([
                 Buffer.from(toFunctionSelector('withdraw(address,uint256,address)').substring(2), 'hex'),
                 uniswapPortalAddress.toBuffer32(),
                 new Fr(wethAmountToBridge),
                 uniswapPortalAddress.toBuffer32()
             ]);
-            const withdrawLeaf = sha256ToField([
-                wethCrossChainHarness.l2Bridge.address,
-                new Fr(1),
-                wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
-                new Fr(publicClient.chain.id),
-                withdrawContent
-            ]);
-            const [swapPrivateL2MessageIndex, swapPrivateSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(withdrawReceipt.blockNumber, swapPrivateLeaf);
-            const [withdrawL2MessageIndex, withdrawSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(withdrawReceipt.blockNumber, withdrawLeaf);
+            const withdrawLeaf = computeL2ToL1MessageHash({
+                l2Sender: wethCrossChainHarness.l2Bridge.address,
+                l1Recipient: wethCrossChainHarness.tokenPortalAddress,
+                content: withdrawContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
+            const swapResult = await computeL2ToL1MembershipWitness(aztecNode, swapPrivateLeaf, withdrawReceipt.txHash);
+            const { epochNumber: epoch } = swapResult;
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, withdrawLeaf, withdrawReceipt.txHash);
+            const swapPrivateL2MessageIndex = swapResult.leafIndex;
+            const swapPrivateSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
-                _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
                 _path: withdrawSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
             const swapPrivateMessageMetadata = {
-                _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(swapPrivateL2MessageIndex),
                 _path: swapPrivateSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
             // ensure that user's funds were burnt
             await wethCrossChainHarness.expectPrivateBalanceOnL2(ownerAddress, wethL2BalanceBeforeSwap - wethAmountToBridge);
-            // Since the outbox is only consumable when the block is proven, we need to set the block to be proven
-            await cheatCodes.rollup.markAsProven(await rollup.read.getPendingBlockNumber());
+            // Since the outbox is only consumable when the epoch is proven, we need to advance to the next epoch.
+            await cheatCodes.rollup.advanceToEpoch(EpochNumber(epoch + 1));
+            await waitForProven(aztecNode, withdrawReceipt, {
+                provenTimeout: 300
+            });
             // On L1 call swap_public!
             logger.info('call swap_public on L1');
             const swapArgs = [
@@ -583,14 +622,16 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             await wethCrossChainHarness.mintTokensPublicOnL2(wethAmountToBridge);
             // Owner gives uniswap approval to transfer funds on its behalf
             const nonceForWETHTransferApproval = new Fr(5n);
-            const validateActionInteraction = await ownerWallet.setPublicAuthWit({
+            const validateActionInteraction = await wallet.setPublicAuthWit(ownerAddress, {
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_in_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferApproval)
             }, true);
-            await validateActionInteraction.send().wait();
+            await validateActionInteraction.send();
             // Call swap_public on L2
             const secretHashForDepositingSwappedDai = Fr.random();
-            const withdrawReceipt = await uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, Fr.ZERO).send().wait();
+            const { receipt: withdrawReceipt } = await uniswapL2Contract.methods.swap_public(ownerAddress, wethCrossChainHarness.l2Bridge.address, wethAmountToBridge, daiCrossChainHarness.l2Bridge.address, nonceForWETHTransferApproval, uniswapFeeTier, minimumOutputAmount, ownerAddress, secretHashForDepositingSwappedDai, ownerEthAddress, Fr.ZERO).send({
+                from: ownerAddress
+            });
             const swapPublicContent = sha256ToField([
                 Buffer.from(toFunctionSelector('swap_public(address,uint256,uint24,address,uint256,bytes32,bytes32,address)').substring(2), 'hex'),
                 wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
@@ -602,42 +643,50 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 secretHashForDepositingSwappedDai,
                 ownerEthAddress.toBuffer32()
             ]);
-            const swapPublicLeaf = sha256ToField([
-                uniswapL2Contract.address,
-                new Fr(1),
-                EthAddress.fromString(uniswapPortal.address).toBuffer32(),
-                new Fr(publicClient.chain.id),
-                swapPublicContent
-            ]);
+            const swapPublicLeaf = computeL2ToL1MessageHash({
+                l2Sender: uniswapL2Contract.address,
+                l1Recipient: EthAddress.fromString(uniswapPortal.address),
+                content: swapPublicContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
             const withdrawContent = sha256ToField([
                 Buffer.from(toFunctionSelector('withdraw(address,uint256,address)').substring(2), 'hex'),
                 uniswapPortalAddress.toBuffer32(),
                 new Fr(wethAmountToBridge),
                 uniswapPortalAddress.toBuffer32()
             ]);
-            const withdrawLeaf = sha256ToField([
-                wethCrossChainHarness.l2Bridge.address,
-                new Fr(1),
-                wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
-                new Fr(publicClient.chain.id),
-                withdrawContent
-            ]);
-            const [swapPublicL2MessageIndex, swapPublicSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(withdrawReceipt.blockNumber, swapPublicLeaf);
-            const [withdrawL2MessageIndex, withdrawSiblingPath] = await aztecNode.getL2ToL1MessageMembershipWitness(withdrawReceipt.blockNumber, withdrawLeaf);
+            const withdrawLeaf = computeL2ToL1MessageHash({
+                l2Sender: wethCrossChainHarness.l2Bridge.address,
+                l1Recipient: wethCrossChainHarness.tokenPortalAddress,
+                content: withdrawContent,
+                rollupVersion: new Fr(version),
+                chainId: new Fr(l1Client.chain.id)
+            });
+            const swapResult = await computeL2ToL1MembershipWitness(aztecNode, swapPublicLeaf, withdrawReceipt.txHash);
+            const { epochNumber: epoch } = swapResult;
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, withdrawLeaf, withdrawReceipt.txHash);
+            const swapPublicL2MessageIndex = swapResult.leafIndex;
+            const swapPublicSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
-                _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
                 _path: withdrawSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
             const swapPublicMessageMetadata = {
-                _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
+                _epoch: BigInt(epoch),
                 _leafIndex: BigInt(swapPublicL2MessageIndex),
                 _path: swapPublicSiblingPath.toBufferArray().map((buf)=>`0x${buf.toString('hex')}`)
             };
             // check weth balance of owner on L2 (we first bridged `wethAmountToBridge` into L2 and now withdrew it!)
             await wethCrossChainHarness.expectPublicBalanceOnL2(ownerAddress, 0n);
-            // Since the outbox is only consumable when the block is proven, we need to set the block to be proven
-            await cheatCodes.rollup.markAsProven(await rollup.read.getPendingBlockNumber());
+            // Since the outbox is only consumable when the epoch is proven, we need to advance to the next epoch.
+            await cheatCodes.rollup.advanceToEpoch(EpochNumber(epoch + 1));
+            await waitForProven(aztecNode, withdrawReceipt, {
+                provenTimeout: 300
+            });
             // Call swap_private on L1
             logger.info('Execute withdraw and swap on the uniswapPortal!');
             const swapArgs = [