test

Author: kumaryash90

packages/thirdweb/src/extensions/erc1155/customDrop1155.test.ts

@@ -7,13 +7,16 @@ import { TEST_CLIENT } from "../../../test/src/test-clients.js";
 import {
   TEST_ACCOUNT_B,
   TEST_ACCOUNT_C,
+  TEST_ACCOUNT_D,
 } from "../../../test/src/test-wallets.js";
 import { type ThirdwebContract, getContract } from "../../contract/contract.js";
 import { sendAndConfirmTransaction } from "../../transaction/actions/send-and-confirm-transaction.js";
 import { deployContractfromDeployMetadata } from "../prebuilts/deploy-published.js";
 import { balanceOf } from "./__generated__/IERC1155/read/balanceOf.js";
 import { nextTokenIdToMint } from "./__generated__/IERC1155Enumerable/read/nextTokenIdToMint.js";
+import { getClaimConditions } from "./drops/read/getClaimConditions.js";
 import { claimTo } from "./drops/write/claimTo.js";
+import { resetClaimEligibility } from "./drops/write/resetClaimEligibility.js";
 import { setClaimConditions } from "./drops/write/setClaimConditions.js";
 import { getNFT } from "./read/getNFT.js";
 import { lazyMint } from "./write/lazyMint.js";
@@ -92,6 +95,14 @@ describe.runIf(process.env.TW_SECRET_KEY)(
         }),
         account: TEST_ACCOUNT_C,
       });
+
+      const condition = await getClaimConditions({
+        contract,
+        tokenId: 0n,
+      });
+      expect(condition.length).to.eq(1);
+      expect(condition[0]?.maxClaimableSupply).to.eq(10n);
+
       const claimTx = claimTo({
         contract,
         to: TEST_ACCOUNT_C.address,
@@ -166,5 +177,80 @@ describe.runIf(process.env.TW_SECRET_KEY)(
           chainId: ${contract.chain.id}]
         `);
     });
+
+    it("should reset claim eligibility", async () => {
+      await sendAndConfirmTransaction({
+        transaction: setClaimConditions({
+          contract,
+          phases: [
+            {
+              startTime: new Date(0),
+              maxClaimableSupply: 10n,
+              maxClaimablePerWallet: 1n,
+            },
+          ],
+          tokenId: 0n,
+          singlePhaseDrop: true,
+        }),
+        account: TEST_ACCOUNT_C,
+      });
+      // claim one token
+      const claimTx = claimTo({
+        contract,
+        to: TEST_ACCOUNT_D.address,
+        tokenId: 0n,
+        quantity: 1n,
+        singlePhaseDrop: true,
+      });
+      await sendAndConfirmTransaction({
+        transaction: claimTx,
+        account: TEST_ACCOUNT_D,
+      });
+      await expect(
+        balanceOf({ contract, owner: TEST_ACCOUNT_D.address, tokenId: 0n }),
+      ).resolves.toBe(1n);
+
+      // attempt to claim another token (this should fail)
+      await expect(
+        sendAndConfirmTransaction({
+          transaction: claimTo({
+            tokenId: 0n,
+            contract,
+            to: TEST_ACCOUNT_D.address,
+            quantity: 1n,
+          }),
+          account: TEST_ACCOUNT_D,
+        }),
+      ).rejects.toThrowErrorMatchingInlineSnapshot(`
+        [TransactionError: Error - !Qty
+
+        contract: ${contract.address}
+        chainId: ${contract.chain.id}]
+      `);
+
+      // reset claim eligibility
+      await sendAndConfirmTransaction({
+        transaction: resetClaimEligibility({
+          tokenId: 0n,
+          contract,
+          singlePhaseDrop: true,
+        }),
+        account: TEST_ACCOUNT_C,
+      });
+      // attempt to claim another token (this should succeed)
+      await sendAndConfirmTransaction({
+        transaction: claimTo({
+          tokenId: 0n,
+          contract,
+          to: TEST_ACCOUNT_D.address,
+          quantity: 1n,
+        }),
+        account: TEST_ACCOUNT_D,
+      });
+      // check that the account has claimed two tokens
+      await expect(
+        balanceOf({ tokenId: 0n, contract, owner: TEST_ACCOUNT_D.address }),
+      ).resolves.toBe(2n);
+    });
   },
 );