Add tests

skalenetwork · Aug 28, 2024 · 2ac3295 · 2ac3295
1 parent cf56841
commit 2ac3295
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 0 deletions.
diff --git a/contracts/test/SkaleTokenInternalTester.sol b/contracts/test/SkaleTokenInternalTester.sol
@@ -21,6 +21,7 @@
 
 pragma solidity 0.8.17;
 
+import { IERC777Recipient } from "@openzeppelin/contracts/token/ERC777/IERC777Recipient.sol";
 import { SkaleToken } from "../SkaleToken.sol";
 import { ISkaleTokenInterfaceTester } from "./interfaces/ISkaleTokenInterfaceTester.sol";
 
@@ -32,6 +33,30 @@ contract SkaleTokenInternalTester is SkaleToken, ISkaleTokenInterfaceTester {
     // solhint-disable-next-line no-empty-blocks
     { }
 
+    function callTokensReceived(
+        address implementer,
+        address operator,
+        address from,
+        address to,
+        uint256 amount,
+        bytes memory userData,
+        bytes memory operatorData
+    )
+    external
+    override
+    {
+        if (implementer != address(0)) {
+            IERC777Recipient(implementer).tokensReceived({
+                operator: operator,
+                from: from,
+                to: to,
+                amount: amount,
+                userData: userData,
+                operatorData: operatorData
+            });
+        }
+    }
+
     function getMsgData() external view override returns (bytes memory msgData) {
         return _msgData();
     }

diff --git a/contracts/test/interfaces/ISkaleTokenInterfaceTester.sol b/contracts/test/interfaces/ISkaleTokenInterfaceTester.sol
@@ -23,5 +23,15 @@ pragma solidity 0.8.17;
 
 
 interface ISkaleTokenInterfaceTester {
+    function callTokensReceived(
+        address implementer,
+        address operator,
+        address from,
+        address to,
+        uint256 amount,
+        bytes memory userData,
+        bytes memory operatorData
+    )
+    external;
     function getMsgData() external view returns (bytes memory msgData);
 }
diff --git a/test/delegation/Delegation.ts b/test/delegation/Delegation.ts
@@ -234,6 +234,30 @@ describe("Delegation", () => {
             .should.be.eventually.rejectedWith("Limit of validators is reached");
     });
 
+    it("should revert on incorrect transfers", async () => {
+        const skaleTokenInternalTesterFactory = await ethers.getContractFactory("SkaleTokenInternalTester");
+        const skaleTokenInternalTester = await skaleTokenInternalTesterFactory.deploy(contractManager, []);
+        await contractManager.setContractsAddress("SkaleToken", skaleTokenInternalTester);
+        await skaleTokenInternalTester.callTokensReceived(
+            distributor,
+            ethers.ZeroAddress,
+            holder1,
+            holder2, // Receiver always needs to be the Distributor
+            5,
+            "0x",
+            "0x"
+        ).should.be.revertedWithCustomError(distributor, "ReceiverIsIncorrect");
+        await skaleTokenInternalTester.callTokensReceived(
+            distributor,
+            ethers.ZeroAddress,
+            holder1,
+            distributor,
+            5,
+            "0x", // Data length always must be 32 bytes
+            "0x"
+        ).should.be.revertedWithCustomError(distributor, "DataLengthIsIncorrect");
+    })
+
     describe("when holders have tokens and validator is registered", () => {
         let validatorId: bigint;
         fastBeforeEach(async () => {