Merge pull request #14 from mmsaki/main

missing tests wip
ExtropyIO · Feb 9, 2024 · 2b377ac · 2b377ac
2 parents 09092c3 + 949b470
commit 2b377ac
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 4 deletions.
diff --git a/src/Gas.sol b/src/Gas.sol
@@ -316,8 +316,7 @@ contract GasContract is Ownable, Constants {
         emit WhiteListTransfer(_recipient);
     }
 
-
-    function getPaymentStatus(address sender) public returns (bool, uint256) {        
+    function getPaymentStatus(address sender) public view returns (bool, uint256) {
         return (whiteListStruct[sender].paymentStatus, whiteListStruct[sender].amount);
     }
 

diff --git a/test/Gas.UnitTests.t.sol b/test/Gas.UnitTests.t.sol
@@ -2,7 +2,7 @@
 pragma solidity ^0.8.0; 
 
 import "forge-std/Test.sol";
-import "../../src/Gas.sol";
+import "../src/Gas.sol";
 
 contract GasTest is Test {
     GasContract public gas;
@@ -156,6 +156,45 @@ contract GasTest is Test {
         assertEq(gas.balances(_recipient),(_preRecipientAmount + _amount) - gas.whitelist(_sender));
     }
 
+    function testBalanceOf() public {
+        uint256 bal = gas.balanceOf(owner);
+        assertEq(bal, totalSupply);
+    }
 
-}
+    function testCheckForAdmin() public {
+        bool isAdmin = gas.checkForAdmin(owner);
+        assertEq(isAdmin, true);
+    }
+
+    // TODO: No Specification
+    function testGetPaymentHistory() public {}
+
+    // TODO: No Specification
+    function testGetTradingMode() public {}
+
+    // TODO: No Specification
+    function testAddHistory() public {}
+
+    function testTransfer(uint256 _amount, address _recipient) public {
+        vm.assume(_amount <= totalSupply);
+        vm.startPrank(owner);
 
+        uint256 ownerBal = gas.balanceOf(owner);
+        uint256 balBefore = gas.balanceOf(_recipient);
+        gas.transfer(_recipient, _amount, "name");
+        uint256 balAfter = gas.balanceOf(_recipient);
+
+        assertEq(balAfter, balBefore + _amount);
+        assertEq(gas.balanceOf(owner), ownerBal - _amount);
+    }
+
+    function testAddToWhitelist(address user, uint256 tier) public {
+        vm.expectRevert();
+        vm.startPrank(user);
+        vm.assume(user != owner);
+        gas.addToWhitelist(user, tier);
+        vm.stopPrank();
+    }
+
+    function testGetPaymentStatus(address sender) public {
+}