chore: update tests

ourzora · Aug 2, 2023 · c043596 · c043596
1 parent cba58ca
commit c043596
Show file tree

Hide file tree

Showing 5 changed files with 136 additions and 6 deletions.
diff --git a/test/ProtocolRewardsTest.sol b/test/ProtocolRewardsTest.sol
@@ -1,4 +1,4 @@
-// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: MIT
 pragma solidity 0.8.17;
 
 import "forge-std/Test.sol";
@@ -8,12 +8,15 @@ import "../src/ProtocolRewards.sol";
 import "./utils/MockNFTs.sol";
 
 contract ProtocolRewardsTest is Test {
+    uint256 internal constant ETH_SUPPLY = 120_200_000 ether;
+
     ProtocolRewards internal protocolRewards;
 
     address internal collector;
     address internal creator;
-    address internal mintReferral;
     address internal createReferral;
+    address internal mintReferral;
+    address internal firstMinter;
     address internal zora;
 
     function setUp() public virtual {
@@ -23,8 +26,9 @@ contract ProtocolRewardsTest is Test {
 
         collector = makeAddr("collector");
         creator = makeAddr("creator");
-        mintReferral = makeAddr("mintReferral");
         createReferral = makeAddr("createReferral");
+        mintReferral = makeAddr("mintReferral");
+        firstMinter = makeAddr("firstMinter");
         zora = makeAddr("zora");
     }
 }
diff --git a/test/unit/Deposit.t.sol b/test/unit/Deposit.t.sol
@@ -0,0 +1,126 @@
+// SPDX-License-Identifier: MIT
+pragma solidity 0.8.17;
+
+import "../ProtocolRewardsTest.sol";
+import "../../src/abstract/RewardSplits.sol";
+
+contract DepositTest is ProtocolRewardsTest {
+    function setUp() public override {
+        super.setUp();
+    }
+
+    function testDeposit(uint256 amount, address to) public {
+        vm.assume(amount < ETH_SUPPLY);
+        vm.assume(to != address(0));
+
+        vm.deal(collector, amount);
+
+        vm.prank(collector);
+        protocolRewards.deposit{value: amount}(to, "test");
+
+        assertEq(protocolRewards.balanceOf(to), amount);
+    }
+
+    function testRevert_CannotDepositToAddressZero(uint256 amount) public {
+        vm.assume(amount < ETH_SUPPLY);
+
+        vm.deal(collector, amount);
+
+        vm.prank(collector);
+        vm.expectRevert(abi.encodeWithSignature("ADDRESS_ZERO()"));
+        protocolRewards.deposit{value: amount}(address(0), "test");
+    }
+
+    function testDepositBatch(uint8 numRecipients) public {
+        address[] memory recipients = new address[](numRecipients);
+        uint256[] memory amounts = new uint256[](numRecipients);
+
+        uint256 totalValue;
+
+        for (uint256 i; i < numRecipients; ++i) {
+            recipients[i] = makeAddr(vm.toString(i + 1));
+            amounts[i] = i + 1 ether;
+
+            totalValue += amounts[i];
+        }
+
+        vm.deal(collector, totalValue);
+        vm.prank(collector);
+        protocolRewards.depositBatch{value: totalValue}(recipients, amounts, "test");
+
+        for (uint256 i; i < numRecipients; ++i) {
+            assertEq(protocolRewards.balanceOf(recipients[i]), amounts[i]);
+        }
+    }
+
+    function testRevert_RecipientsAndAmountsLengthMismatch(uint8 numRecipients, uint8 numAmounts) public {
+        vm.assume(numRecipients != numAmounts);
+
+        address[] memory recipients = new address[](numRecipients);
+        uint256[] memory amounts = new uint256[](numAmounts);
+
+        uint256 totalValue;
+
+        for (uint256 i; i < numAmounts; ++i) {
+            amounts[i] = i + 1 ether;
+
+            totalValue += amounts[i];
+        }
+
+        for (uint256 i; i < numRecipients; ++i) {
+            recipients[i] = makeAddr(vm.toString(i + 1));
+        }
+
+        vm.deal(collector, totalValue);
+
+        vm.prank(collector);
+        vm.expectRevert(abi.encodeWithSignature("ARRAY_LENGTH_MISMATCH()"));
+        protocolRewards.depositBatch{value: totalValue}(recipients, amounts, "test");
+    }
+
+    function testRevert_InvalidDepositMsgValue(uint8 numRecipients) public {
+        vm.assume(numRecipients > 0);
+
+        address[] memory recipients = new address[](numRecipients);
+        uint256[] memory amounts = new uint256[](numRecipients);
+
+        uint256 totalValue;
+
+        for (uint256 i; i < numRecipients; ++i) {
+            recipients[i] = makeAddr(vm.toString(i + 1));
+            amounts[i] = i + 1 ether;
+
+            totalValue += amounts[i];
+        }
+
+        vm.deal(collector, totalValue);
+
+        vm.prank(collector);
+        vm.expectRevert(abi.encodeWithSignature("INVALID_DEPOSIT()"));
+        protocolRewards.depositBatch{value: 0}(recipients, amounts, "test");
+    }
+
+    function testRevert_RecipientCannotBeAddressZero(uint8 numRecipients) public {
+        vm.assume(numRecipients > 0);
+
+        address[] memory recipients = new address[](numRecipients);
+        uint256[] memory amounts = new uint256[](numRecipients);
+
+        uint256 totalValue;
+
+        for (uint256 i; i < numRecipients; ++i) {
+            recipients[i] = makeAddr(vm.toString(i + 1));
+            amounts[i] = i + 1 ether;
+
+            totalValue += amounts[i];
+        }
+
+        recipients[0] = address(0);
+
+        vm.deal(collector, totalValue);
+
+        vm.prank(collector);
+        vm.expectRevert(abi.encodeWithSignature("ADDRESS_ZERO()"));
+        protocolRewards.depositBatch{value: totalValue}(recipients, amounts, "test");
+    }
+}
diff --git a/test/unit/ERC1155Rewards.t.sol b/test/unit/ERC1155Rewards.t.sol
@@ -1,4 +1,4 @@
-// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: MIT
 pragma solidity 0.8.17;
 
 import "../ProtocolRewardsTest.sol";

diff --git a/test/unit/ERC721Rewards.t.sol b/test/unit/ERC721Rewards.t.sol
@@ -1,4 +1,4 @@
-// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: MIT
 pragma solidity 0.8.17;
 
 import "../ProtocolRewardsTest.sol";

diff --git a/test/utils/MockNFTs.sol b/test/utils/MockNFTs.sol
@@ -1,4 +1,4 @@
-// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: MIT
 pragma solidity 0.8.17;
 
 import {ERC721} from "./ERC721.sol";