Unit test NFT transfers

TimeleapLabs · Mar 17, 2024 · 92cdad7 · 92cdad7
1 parent 367caa9
commit 92cdad7
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/Staking.test.js b/test/Staking.test.js
@@ -657,4 +657,16 @@ describe("Staking", function () {
     const user1Stake = await staking["stakeOf(bytes20)"](user1bls);
     expect(user1Stake.amount).to.equal(ethers.parseUnits("500"));
   });
+
+  it("reverts if the NFT is not the expected one", async function () {
+    await expect(
+      staking.onERC721Received(user1.address, user2.address, 123, "0x")
+    ).to.be.revertedWithCustomError(staking, "WrongNFT()");
+  });
+
+  it("reverts if accepting NFT is forbidden", async function () {
+    await expect(
+      nft.connect(user1).safeTransferFrom(user1.address, stakingAddr, 1)
+    ).to.be.revertedWithCustomError(staking, "Forbidden()");
+  });
 });