test

Author: kumaryash90

packages/thirdweb/src/extensions/prebuilts/process-ref-deployments.test.ts

@@ -17,6 +17,9 @@ describe.runIf(process.env.TW_SECRET_KEY)(
     let mintfeeManagerModuleAddress: string;
     let mintfeeManagerCoreAddress: string;
     let claimableModuleAddress: string;
+    let wethAddress: string;
+    let forwarderAddress: string;
+    let multiwrapAddress: string;
 
     beforeAll(async () => {
       multisigAddress = await deployPublishedContract({
@@ -55,9 +58,59 @@ describe.runIf(process.env.TW_SECRET_KEY)(
         salt: "tw",
         publisher: "0x6453a486d52e0EB6E79Ec4491038E2522a926936",
       });
+
+      wethAddress = await deployPublishedContract({
+        client: TEST_CLIENT,
+        chain: ANVIL_CHAIN,
+        account: TEST_ACCOUNT_A,
+        contractId: "WETH9",
+        version: "0.0.1",
+        salt: "thirdweb",
+        publisher: "0x6453a486d52e0EB6E79Ec4491038E2522a926936",
+      });
+      forwarderAddress = await deployPublishedContract({
+        client: TEST_CLIENT,
+        chain: ANVIL_CHAIN,
+        account: TEST_ACCOUNT_A,
+        contractId: "Forwarder",
+        version: "0.0.1",
+        salt: "thirdweb",
+        publisher: "0x6453a486d52e0EB6E79Ec4491038E2522a926936",
+      });
+      multiwrapAddress = await deployPublishedContract({
+        client: TEST_CLIENT,
+        chain: ANVIL_CHAIN,
+        account: TEST_ACCOUNT_A,
+        contractId: "Multiwrap",
+        version: "0.0.4",
+        publisher: "0x6453a486d52e0EB6E79Ec4491038E2522a926936",
+        contractParams: {
+          _defaultAdmin: TEST_ACCOUNT_A.address,
+          _name: "test",
+          _symbol: "test",
+          _contractURI: "",
+          _trustedForwarders: {
+            defaultValue: "",
+            dynamicValue: {
+              type: "address[]",
+              refContracts: [
+                {
+                  contractId: "Forwarder",
+                  version: "0.0.1",
+                  publisherAddress:
+                    "0x6453a486d52e0EB6E79Ec4491038E2522a926936",
+                  salt: "",
+                },
+              ],
+            },
+          },
+          _royaltyRecipient: TEST_ACCOUNT_A.address,
+          _royaltyBps: 500n,
+        },
+      });
     }, 120000);
 
-    it("should set ref contract dependencies", async () => {
+    it("should set ref contract dependencies for direct deploy contracts", async () => {
       const claimableModule = getContract({
         client: TEST_CLIENT,
         chain: ANVIL_CHAIN,
@@ -99,5 +152,24 @@ describe.runIf(process.env.TW_SECRET_KEY)(
       expect(feeRecipient.toLowerCase()).to.eq(multisigAddress);
       expect(fee).to.eq(5n);
     });
+
+    it("should set ref contract dependencies for auto factory contracts", async () => {
+      const multiwrap = getContract({
+        client: TEST_CLIENT,
+        chain: ANVIL_CHAIN,
+        address: multiwrapAddress,
+      });
+      const fetchedWethAddress = await readContract({
+        contract: multiwrap,
+        method: "function nativeTokenWrapper() returns (address)",
+      });
+      const isTrustedForwarder = await readContract({
+        contract: multiwrap,
+        method: "function isTrustedForwarder(address forwarder) returns (bool)",
+        params: [forwarderAddress],
+      });
+      expect(fetchedWethAddress.toLowerCase()).to.eq(wethAddress);
+      expect(isTrustedForwarder).to.be.true;
+    });
   },
 );