updated contracts

gab0071 · Aug 22, 2024 · 5ef971e · 5ef971e
1 parent ae0718b
commit 5ef971e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/src/Main.sol b/src/Main.sol
@@ -23,7 +23,7 @@ contract Main is Ownable {
     /////////////////////////////////////////////////////////////////////////////////////////////////////////*/
     StakingToken public immutable i_stakingToken;
     RewardToken public immutable i_rewardToken;
-    address[] private stakers;
+    address[] public stakers;
 
     struct StakeInfo {
         uint256 StakedBalance;
@@ -117,12 +117,4 @@ contract Main is Ownable {
             }
         }
     }
-
-    /*/////////////////////////////////////////////////////////////////////////////////////////////////////////
-                                            Getter Functions
-    /////////////////////////////////////////////////////////////////////////////////////////////////////////*/
-    function getStakerAt(uint256 index) public view returns (address) {
-        require(index < stakers.length, "Index out of bounds");
-        return stakers[index];
-    }
 }
diff --git a/test/MainTest.t.sol b/test/MainTest.t.sol
@@ -76,7 +76,7 @@ contract MainTest is Test {
         assertEq(stakingTokenBalanceOfMainContractNew, newStakedBalance);
 
         // verify that user2 was succefully added into the stakers array
-        address staker = mainContract.getStakerAt(1);
+        address staker = mainContract.stakers(1);
         assertEq(staker, user2, "Staker number 2 was not added to the array.");