From c457ad5d6c6b5bb12c9a72ceb81f546b27b091a6 Mon Sep 17 00:00:00 2001
From: Stefan Stefanov <stefan.stefanooov@gmail.com>
Date: Thu, 2 Nov 2023 15:40:34 +0200
Subject: [PATCH] Adding tests for ERC-2981

Signed-off-by: Stefan Stefanov <stefan.stefanooov@gmail.com>
---
 contracts/solidity/oz/ERC-2981/ERC2981.sol |  34 +++++++
 test/solidity/oz/erc2981/erc2981.js        | 108 +++++++++++++++++++++
 2 files changed, 142 insertions(+)
 create mode 100644 contracts/solidity/oz/ERC-2981/ERC2981.sol
 create mode 100644 test/solidity/oz/erc2981/erc2981.js

diff --git a/contracts/solidity/oz/ERC-2981/ERC2981.sol b/contracts/solidity/oz/ERC-2981/ERC2981.sol
new file mode 100644
index 000000000..ea8ebaf20
--- /dev/null
+++ b/contracts/solidity/oz/ERC-2981/ERC2981.sol
@@ -0,0 +1,34 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.20;
+
+import "@openzeppelin/contracts/token/common/ERC2981.sol";
+
+contract ERC2981Test is ERC2981 {
+    constructor() ERC2981() {}
+
+    function setTokenRoyalty(uint256 tokenId, address receiver, uint96 feeNumerator)
+        external
+    {
+        _setTokenRoyalty(tokenId, receiver, feeNumerator);
+    }
+
+    function feeDenominator()
+        pure
+        external
+        returns (uint256)
+    {
+       return _feeDenominator();
+    }
+
+    function setDefaultRoyalty(address receiver, uint96 feeNumerator)
+        external
+    {
+        _setDefaultRoyalty(receiver, feeNumerator);
+    }
+
+    function resetTokenRoyalty(uint256 tokenId)
+        external
+    {
+        _resetTokenRoyalty(tokenId);
+    }
+}
\ No newline at end of file
diff --git a/test/solidity/oz/erc2981/erc2981.js b/test/solidity/oz/erc2981/erc2981.js
new file mode 100644
index 000000000..f3995986e
--- /dev/null
+++ b/test/solidity/oz/erc2981/erc2981.js
@@ -0,0 +1,108 @@
+/*-
+ *
+ * Hedera Smart Contracts
+ *
+ * Copyright (C) 2023 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+const { expect } = require('chai')
+const { ethers } = require('hardhat')
+
+const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms))
+
+describe('@solidityequiv3 ERC-2981', function () {
+    let signers, wallet, wallet2
+    let contract
+    const TOKEN_ID = 666
+    const DEFAULT_FEE_NUMERATOR = 20
+    const DEFAULT_FEE_DENOMINATOR = 10000
+
+    before(async function () {
+        signers = await ethers.getSigners()
+        wallet = signers[0]
+        wallet2 = signers[1]
+
+        const factory = await ethers.getContractFactory('ERC2981Test')
+        contract = await factory.deploy()
+    })
+
+    it('should deploy the contract', async function () {
+        const deployed = await contract.deployed();
+        expect(deployed).to.exist
+    })
+
+    it('should return the default Fee Denominator', async function () {
+        const res = await contract.feeDenominator()
+
+        expect(res).to.equal(DEFAULT_FEE_DENOMINATOR)
+    })
+
+    it('should set the Default Royalty', async function () {
+        const trx = await contract.setDefaultRoyalty(wallet2.address, DEFAULT_FEE_NUMERATOR, {gasLimit : 10_000_000})
+        await trx.wait()
+        const royaltyInfoDefault = await contract.royaltyInfo(ethers.constants.AddressZero, 10000)
+
+        expect(royaltyInfoDefault[0]).to.equal(wallet2.address)
+        expect(royaltyInfoDefault[1]).to.equal(DEFAULT_FEE_NUMERATOR)
+    })
+
+    it('should return error for setting ZERO Address for receiver', async function () {
+        let hasError = false
+        try {
+            const trx = await contract.setDefaultRoyalty(ethers.constants.AddressZero, DEFAULT_FEE_NUMERATOR)
+            await trx.wait()
+        } catch (error) {
+            hasError = true
+        }
+
+        expect(hasError).to.equal(true)
+    })
+
+    it('should return error for setting too big of feeNumerator', async function () {
+        let hasError = false
+        try {
+            const trx = await contract.setDefaultRoyalty(wallet.address, DEFAULT_FEE_DENOMINATOR + 1)
+            await trx.wait()
+        } catch (error) {
+            hasError = true
+        }
+
+        expect(hasError).to.equal(true)
+    })
+
+    it('should return Royalty info for token', async function () {
+        const salePrice = 200
+        const royaltyFraction = 400
+        const feeDenominator = await contract.feeDenominator()
+        const calculatedRoyalty = (salePrice * royaltyFraction) / feeDenominator
+        const trx = await contract.setTokenRoyalty(TOKEN_ID, wallet.address, royaltyFraction)
+        trx.wait()
+        const royaltyInfoDefault = await contract.royaltyInfo(TOKEN_ID, salePrice)
+
+        expect(royaltyInfoDefault[0]).to.equal(wallet.address)
+        expect(royaltyInfoDefault[1]).to.equal(calculatedRoyalty)
+    })
+
+    it('should reset Royalty Info', async function () {
+        const trx = await contract.resetTokenRoyalty(TOKEN_ID)
+        await trx.wait()
+        const royaltyInfoDefault = await contract.royaltyInfo(TOKEN_ID, DEFAULT_FEE_NUMERATOR)
+
+        expect(royaltyInfoDefault[0]).to.equal(wallet2.address)
+        expect(royaltyInfoDefault[1]).to.equal(0)
+    })
+  
+})