@aztec/end-to-end 0.0.0-test.1 → 0.0.1-fake-ceab37513c

package/dest/shared/gas_portal_test_harness.js

@@ -7,8 +7,8 @@ export class FeeJuicePortalTestingHarnessFactory {
         this.config = config;
     }
     async createReal() {
-        const { aztecNode, pxeService, publicClient, walletClient, wallet, logger } = this.config;
-        const ethAccount = EthAddress.fromString((await walletClient.getAddresses())[0]);
+        const { aztecNode, aztecNodeAdmin, pxeService, l1Client, wallet, logger } = this.config;
+        const ethAccount = EthAddress.fromString((await l1Client.getAddresses())[0]);
         const l1ContractAddresses = (await pxeService.getNodeInfo()).l1ContractAddresses;
         const feeJuiceAddress = l1ContractAddresses.feeJuiceAddress;
         const feeJuicePortalAddress = l1ContractAddresses.feeJuicePortalAddress;
@@ -16,7 +16,7 @@ export class FeeJuicePortalTestingHarnessFactory {
             throw new Error('Fee Juice portal not deployed on L1');
         }
         const gasL2 = await FeeJuiceContract.at(ProtocolContractAddress.FeeJuice, wallet);
-        return new GasBridgingTestHarness(aztecNode, pxeService, logger, gasL2, ethAccount, feeJuicePortalAddress, feeJuiceAddress, publicClient, walletClient);
+        return new GasBridgingTestHarness(aztecNode, aztecNodeAdmin, pxeService, logger, gasL2, ethAccount, feeJuicePortalAddress, feeJuiceAddress, l1ContractAddresses.feeAssetHandlerAddress, l1Client);
     }
     static create(config) {
         const factory = new FeeJuicePortalTestingHarnessFactory(config);
@@ -28,54 +28,62 @@ export class FeeJuicePortalTestingHarnessFactory {
  * shared between cross chain tests.
  */ export class GasBridgingTestHarness {
     aztecNode;
+    aztecNodeAdmin;
     pxeService;
     logger;
     feeJuice;
     ethAccount;
     feeJuicePortalAddress;
     l1FeeJuiceAddress;
-    publicClient;
-    walletClient;
+    feeAssetHandlerAddress;
+    l1Client;
     l1TokenManager;
     feeJuicePortalManager;
-    constructor(/** Aztec node */ aztecNode, /** Private eXecution Environment (PXE). */ pxeService, /** Logger. */ logger, /** L2 Token/Bridge contract. */ feeJuice, /** Eth account to interact with. */ ethAccount, /** Portal address. */ feeJuicePortalAddress, /** Underlying token for portal tests. */ l1FeeJuiceAddress, /** Viem Public client instance. */ publicClient, /** Viem Wallet Client instance. */ walletClient){
+    constructor(/** Aztec node */ aztecNode, /** Aztec node admin interface */ aztecNodeAdmin, /** Private eXecution Environment (PXE). */ pxeService, /** Logger. */ logger, /** L2 Token/Bridge contract. */ feeJuice, /** Eth account to interact with. */ ethAccount, /** Portal address. */ feeJuicePortalAddress, /** Underlying token for portal tests. */ l1FeeJuiceAddress, /** Fee asset handler address. */ feeAssetHandlerAddress, /** Viem Extended client instance. */ l1Client){
         this.aztecNode = aztecNode;
+        this.aztecNodeAdmin = aztecNodeAdmin;
         this.pxeService = pxeService;
         this.logger = logger;
         this.feeJuice = feeJuice;
         this.ethAccount = ethAccount;
         this.feeJuicePortalAddress = feeJuicePortalAddress;
         this.l1FeeJuiceAddress = l1FeeJuiceAddress;
-        this.publicClient = publicClient;
-        this.walletClient = walletClient;
-        this.feeJuicePortalManager = new L1FeeJuicePortalManager(this.feeJuicePortalAddress, this.l1FeeJuiceAddress, this.publicClient, this.walletClient, this.logger);
+        this.feeAssetHandlerAddress = feeAssetHandlerAddress;
+        this.l1Client = l1Client;
+        this.feeJuicePortalManager = new L1FeeJuicePortalManager(this.feeJuicePortalAddress, this.l1FeeJuiceAddress, this.feeAssetHandlerAddress, this.l1Client, this.logger);
         this.l1TokenManager = this.feeJuicePortalManager.getTokenManager();
     }
-    async mintTokensOnL1(amount, to = this.ethAccount) {
-        const balanceBefore = await this.l1TokenManager.getL1TokenBalance(to.toString());
-        await this.l1TokenManager.mint(amount, to.toString());
-        expect(await this.l1TokenManager.getL1TokenBalance(to.toString())).toEqual(balanceBefore + amount);
+    async mintTokensOnL1(to = this.ethAccount) {
+        // const balanceBefore = await this.l1TokenManager.getL1TokenBalance(to.toString());
+        await this.l1TokenManager.mint(to.toString());
+    // expect(await this.l1TokenManager.getL1TokenBalance(to.toString())).toEqual(balanceBefore + amount);
     }
     async getL1FeeJuiceBalance(address) {
         return await this.l1TokenManager.getL1TokenBalance(address.toString());
     }
-    sendTokensToPortalPublic(bridgeAmount, l2Address, mint = false) {
-        return this.feeJuicePortalManager.bridgeTokensPublic(l2Address, bridgeAmount, mint);
+    sendTokensToPortalPublic(bridgeAmount, l2Address) {
+        return this.feeJuicePortalManager.bridgeTokensPublic(l2Address, bridgeAmount, false);
     }
-    async consumeMessageOnAztecAndClaimPrivately(owner, claim) {
+    async consumeMessageOnAztecAndClaimPrivately(owner, claimer, claim) {
         this.logger.info('Consuming messages on L2 Privately');
         const { claimAmount, claimSecret, messageLeafIndex } = claim;
-        await this.feeJuice.methods.claim(owner, claimAmount, claimSecret, messageLeafIndex).send().wait();
+        await this.feeJuice.methods.claim(owner, claimAmount, claimSecret, messageLeafIndex).send({
+            from: claimer
+        }).wait();
     }
     async getL2PublicBalanceOf(owner) {
-        return await this.feeJuice.methods.balance_of_public(owner).simulate();
+        return await this.feeJuice.methods.balance_of_public(owner).simulate({
+            from: owner
+        });
     }
     async expectPublicBalanceOnL2(owner, expectedBalance) {
         const balance = await this.getL2PublicBalanceOf(owner);
         expect(balance).toBe(expectedBalance);
     }
-    async prepareTokensOnL1(bridgeAmount, owner) {
-        const claim = await this.sendTokensToPortalPublic(bridgeAmount, owner, true);
+    async prepareTokensOnL1(owner) {
+        const bridgeAmount = await this.l1TokenManager.getMintAmount();
+        await this.mintTokensOnL1();
+        const claim = await this.sendTokensToPortalPublic(bridgeAmount, owner);
         const isSynced = async ()=>await this.aztecNode.isL1ToL2MessageSynced(Fr.fromHexString(claim.messageHash));
         await retryUntil(isSynced, `message ${claim.messageHash} sync`, 24, 1);
         // Progress by 2 L2 blocks so that the l1ToL2Message added above will be available to use on L2.
@@ -83,16 +91,26 @@ export class FeeJuicePortalTestingHarnessFactory {
         await this.advanceL2Block();
         return claim;
     }
-    async bridgeFromL1ToL2(bridgeAmount, owner) {
+    async bridgeFromL1ToL2(owner, claimer) {
         // Prepare the tokens on the L1 side
-        const claim = await this.prepareTokensOnL1(bridgeAmount, owner);
+        const claim = await this.prepareTokensOnL1(owner);
         // Consume L1 -> L2 message and claim tokens privately on L2
-        await this.consumeMessageOnAztecAndClaimPrivately(owner, claim);
-        await this.expectPublicBalanceOnL2(owner, bridgeAmount);
+        await this.consumeMessageOnAztecAndClaimPrivately(owner, claimer, claim);
     }
     async advanceL2Block() {
         const initialBlockNumber = await this.aztecNode.getBlockNumber();
-        await this.aztecNode.flushTxs();
+        let minTxsPerBlock = undefined;
+        if (this.aztecNodeAdmin) {
+            ({ minTxsPerBlock } = await this.aztecNodeAdmin.getConfig());
+            await this.aztecNodeAdmin.setConfig({
+                minTxsPerBlock: 0
+            }); // Set to 0 to ensure we can advance the block
+        }
         await retryUntil(async ()=>await this.aztecNode.getBlockNumber() >= initialBlockNumber + 1);
+        if (this.aztecNodeAdmin && minTxsPerBlock !== undefined) {
+            await this.aztecNodeAdmin.setConfig({
+                minTxsPerBlock
+            });
+        }
     }
 } // docs:end:cross_chain_test_harness

package/dest/shared/submit-transactions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"submit-transactions.d.ts","sourceRoot":"","sources":["../../src/shared/submit-transactions.ts"],"names":[],"mappings":"AACA,OAAO,EAAsB,KAAK,MAAM,EAAE,KAAK,MAAM,EAAY,KAAK,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEtG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAGpD,eAAO,MAAM,WAAW,QACjB,UAAU,UACP,MAAM,UACN,MAAM,UACN,MAAM,KACb,QAAQ,MAAM,EAAE,CAqBlB,CAAC"}
+{"version":3,"file":"submit-transactions.d.ts","sourceRoot":"","sources":["../../src/shared/submit-transactions.ts"],"names":[],"mappings":"AACA,OAAO,EAAsB,KAAK,MAAM,EAAE,KAAK,MAAM,EAAY,KAAK,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEtG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAGpD,eAAO,MAAM,WAAW,GACtB,KAAK,UAAU,EACf,QAAQ,MAAM,EACd,QAAQ,MAAM,EACd,QAAQ,MAAM,KACb,OAAO,CAAC,MAAM,EAAE,CAqBlB,CAAC"}

package/dest/shared/uniswap_l1_l2.d.ts

@@ -1,5 +1,6 @@
-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';
+import { type AccountWallet, type AztecNode, type Logger, type PXE } from '@aztec/aztec.js';
+import { CheatCodes } from '@aztec/aztec/testing';
+import { type DeployL1ContractsReturnType, type ExtendedViemWalletClient } from '@aztec/ethereum';
 /** Objects to be returned by the uniswap setup function */
 export type UniswapSetupContext = {
     /** Aztec Node instance */
@@ -8,10 +9,8 @@ export type UniswapSetupContext = {
     pxe: PXE;
     /** Logger instance named as the current test. */
     logger: Logger;
-    /** Viem Public client instance. */
-    publicClient: ViemPublicClient;
-    /** Viem Wallet Client instance. */
-    walletClient: ViemWalletClient;
+    /** The L1 wallet client, extended with public actions. */
+    l1Client: ExtendedViemWalletClient;
     /** The owner wallet. */
     ownerWallet: AccountWallet;
     /** The sponsor wallet. */

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":"AAAA,OAAO,EACL,KAAK,aAAa,EAElB,KAAK,SAAS,EAGd,KAAK,MAAM,EACX,KAAK,GAAG,EAGT,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EACL,KAAK,2BAA2B,EAChC,KAAK,wBAAwB,EAI9B,MAAM,iBAAiB,CAAC;AAuBzB,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,0DAA0D;IAC1D,QAAQ,EAAE,wBAAwB,CAAC;IACnC,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,GAC/B,OAAO,MAAM,OAAO,CAAC,mBAAmB,CAAC,EACzC,SAAS,MAAM,OAAO,CAAC,IAAI,CAAC,EAC5B,gCAAkC,SAg+BnC,CAAC"}

package/dest/shared/uniswap_l1_l2.js

@@ -1,11 +1,13 @@
 import { AztecAddress, EthAddress, Fr, computeAuthWitMessageHash, generateClaimSecret } from '@aztec/aztec.js';
-import { deployL1Contract, extractEvent } from '@aztec/ethereum';
+import { RollupContract, deployL1Contract, extractEvent } from '@aztec/ethereum';
 import { sha256ToField } from '@aztec/foundation/crypto';
-import { InboxAbi, RollupAbi, UniswapPortalAbi, UniswapPortalBytecode } from '@aztec/l1-artifacts';
+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.
@@ -25,14 +27,13 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
         let aztecNode;
         let pxe;
         let logger;
-        let walletClient;
-        let publicClient;
+        let l1Client;
         let ownerWallet;
         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,36 +45,36 @@ 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) {
+            ({ aztecNode, pxe, logger, l1Client, ownerWallet, sponsorWallet, deployL1ContractsValues, cheatCodes } = await setup());
+            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
-            });
+            rollup = new RollupContract(deployL1ContractsValues.l1Client, deployL1ContractsValues.l1ContractAddresses.rollupAddress);
+            version = Number(await rollup.getVersion());
             ownerAddress = ownerWallet.getAddress();
-            // sponsorAddress = sponsorWallet.getAddress();
-            ownerEthAddress = EthAddress.fromString((await walletClient.getAddresses())[0]);
-            await ensureAccountsPubliclyDeployed(ownerWallet, [
+            sponsorAddress = sponsorWallet.getAddress();
+            ownerEthAddress = EthAddress.fromString((await l1Client.getAddresses())[0]);
+            await ensureAccountContractsPublished(ownerWallet, [
                 ownerWallet,
                 sponsorWallet
             ]);
             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, pxe, deployL1ContractsValues.l1Client, ownerWallet, 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, pxe, l1Client, ownerWallet, 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();
+            uniswapL2Contract = await UniswapContract.deploy(ownerWallet, uniswapPortalAddress).send({
+                from: ownerAddress
+            }).deployed();
             const registryAddress = (await pxe.getNodeInfo()).l1ContractAddresses.registryAddress;
             await uniswapPortal.write.initialize([
                 registryAddress.toString(),
@@ -81,11 +82,11 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             ], {});
             // 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);
@@ -114,14 +115,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 ownerWallet.createAuthWit({
                 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 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
+                ]
+            }).wait();
             const swapPrivateFunction = 'swap_private(address,uint256,uint24,address,uint256,bytes32,address)';
             const swapPrivateContent = sha256ToField([
                 Buffer.from(toFunctionSelector(swapPrivateFunction).substring(2), 'hex'),
@@ -133,37 +139,41 @@ 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());
+            await cheatCodes.rollup.markAsProven(await rollup.getBlockNumber());
             // 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 swapResult = await computeL2ToL1MembershipWitness(aztecNode, l2UniswapInteractionReceipt.blockNumber, swapPrivateLeaf);
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, l2UniswapInteractionReceipt.blockNumber, withdrawLeaf);
+            const swapPrivateL2MessageIndex = swapResult.leafIndex;
+            const swapPrivateSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
                 _l2BlockNumber: BigInt(l2UniswapInteractionReceipt.blockNumber),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
@@ -188,7 +198,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
@@ -437,7 +447,9 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 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()}`);
+            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 +458,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 ownerWallet.createAuthWit({
                 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).prove({
+                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 ()=>{
@@ -464,10 +481,14 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_in_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferApproval)
             }, true);
-            await validateActionInteraction.send().wait();
+            await validateActionInteraction.send({
+                from: ownerAddress
+            }).wait();
             // 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
+            }).wait();
             // check weth balance of owner on L2 (we first bridged `wethAmountToBridge` into L2 and now withdrew it!)
             await wethCrossChainHarness.expectPublicBalanceOnL2(ownerAddress, 0n);
         });
@@ -482,8 +503,12 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 caller: approvedUser,
                 action
             }, true);
-            await validateActionInteraction.send().wait();
-            await expect(action.simulate()).rejects.toThrow(/unauthorized/);
+            await validateActionInteraction.send({
+                from: ownerAddress
+            }).wait();
+            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:
@@ -492,8 +517,12 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 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({
+                from: ownerAddress
+            }).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({
+                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 +532,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 ownerWallet.createAuthWit({
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_to_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferToPublicApproval)
             });
@@ -511,7 +540,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 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
+                ]
+            }).wait();
             const swapPrivateContent = sha256ToField([
                 Buffer.from(toFunctionSelector('swap_private(address,uint256,uint24,address,uint256,bytes32,address)').substring(2), 'hex'),
                 wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
@@ -522,28 +556,32 @@ 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, withdrawReceipt.blockNumber, swapPrivateLeaf);
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, withdrawReceipt.blockNumber, withdrawLeaf);
+            const swapPrivateL2MessageIndex = swapResult.leafIndex;
+            const swapPrivateSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
                 _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
@@ -557,7 +595,7 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // 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());
+            await cheatCodes.rollup.markAsProven(await rollup.getBlockNumber());
             // On L1 call swap_public!
             logger.info('call swap_public on L1');
             const swapArgs = [
@@ -587,10 +625,14 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
                 caller: uniswapL2Contract.address,
                 action: wethCrossChainHarness.l2Token.methods.transfer_in_public(ownerAddress, uniswapL2Contract.address, wethAmountToBridge, nonceForWETHTransferApproval)
             }, true);
-            await validateActionInteraction.send().wait();
+            await validateActionInteraction.send({
+                from: ownerAddress
+            }).wait();
             // 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 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
+            }).wait();
             const swapPublicContent = sha256ToField([
                 Buffer.from(toFunctionSelector('swap_public(address,uint256,uint24,address,uint256,bytes32,bytes32,address)').substring(2), 'hex'),
                 wethCrossChainHarness.tokenPortalAddress.toBuffer32(),
@@ -602,28 +644,32 @@ 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, withdrawReceipt.blockNumber, swapPublicLeaf);
+            const withdrawResult = await computeL2ToL1MembershipWitness(aztecNode, withdrawReceipt.blockNumber, withdrawLeaf);
+            const swapPublicL2MessageIndex = swapResult.leafIndex;
+            const swapPublicSiblingPath = swapResult.siblingPath;
+            const withdrawL2MessageIndex = withdrawResult.leafIndex;
+            const withdrawSiblingPath = withdrawResult.siblingPath;
             const withdrawMessageMetadata = {
                 _l2BlockNumber: BigInt(withdrawReceipt.blockNumber),
                 _leafIndex: BigInt(withdrawL2MessageIndex),
@@ -637,7 +683,7 @@ export const uniswapL1L2TestSuite = (setup, cleanup, expectedForkBlockNumber = 1
             // 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());
+            await cheatCodes.rollup.markAsProven(await rollup.getBlockNumber());
             // Call swap_private on L1
             logger.info('Execute withdraw and swap on the uniswapPortal!');
             const swapArgs = [

package/dest/simulators/lending_simulator.d.ts

@@ -1,9 +1,8 @@
-import { AztecAddress, type CheatCodes, Fr } from '@aztec/aztec.js';
+import { AztecAddress, Fr } from '@aztec/aztec.js';
+import { CheatCodes } from '@aztec/aztec/testing';
+import type { RollupContract } from '@aztec/ethereum';
 import type { TestDateProvider } from '@aztec/foundation/timer';
-import type { RollupAbi } from '@aztec/l1-artifacts';
 import type { LendingContract } from '@aztec/noir-contracts.js/Lending';
-import type { Account, GetContractReturnType, HttpTransport, WalletClient } from 'viem';
-import type * as chains from 'viem/chains';
 import type { TokenSimulator } from './token_simulator.js';
 /**
  * Contains utilities to compute the "key" for private holdings in the public state.
@@ -29,7 +28,7 @@ export declare class LendingSimulator {
     private rate;
     private ethereumSlotDuration;
     /** the rollup contract */
-    rollup: GetContractReturnType<typeof RollupAbi, WalletClient<HttpTransport, chains.Chain, Account>>;
+    rollup: RollupContract;
     /** the lending contract */
     lendingContract: LendingContract;
     /** the collateral asset used in the lending contract */
@@ -46,7 +45,7 @@ export declare class LendingSimulator {
     private mintedOutside;
     constructor(cc: CheatCodes, account: LendingAccount, rate: bigint, ethereumSlotDuration: number, 
     /** the rollup contract */
-    rollup: GetContractReturnType<typeof RollupAbi, WalletClient<HttpTransport, chains.Chain, Account>>, 
+    rollup: RollupContract, 
     /** the lending contract */
     lendingContract: LendingContract, 
     /** the collateral asset used in the lending contract */