coverage

blockful-io · Jun 30, 2024 · 852da7d · 852da7d
1 parent 9412d82
commit 852da7d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/Staking.sol b/src/Staking.sol
@@ -242,8 +242,6 @@ contract Staking is ERC20VotesUpgradeable, Ownable2StepUpgradeable {
         stakingToken.safeTransferFrom(keyper, address(this), amount);
 
         emit Staked(keyper, amount, sharesToMint, lockPeriod);
-
-        return stakeId;
     }
 
     /// @notice Unstake SHU

diff --git a/test/RewardsDistributor.t.sol b/test/RewardsDistributor.t.sol
@@ -37,7 +37,7 @@ contract RewardsDistributorTest is Test {
 
 contract Constructor is RewardsDistributorTest {
     function test_SetUp() public view {
-        assertEq(address(rewardsDistributor.rewardToken()), address(govToken));
+        assertEq(rewardsDistributor.rewardToken(), address(govToken));
         assertEq(Ownable(address(rewardsDistributor)).owner(), address(this));
     }
 }

diff --git a/test/Staking.t.sol b/test/Staking.t.sol
@@ -150,6 +150,8 @@ contract Initializer is StakingTest {
         );
         assertEq(staking.lockPeriod(), LOCK_PERIOD, "Wrong lock period");
         assertEq(staking.minStake(), MIN_STAKE, "Wrong min stake");
+
+        assertEq(staking.exposed_nextStakeId(), 1);
     }
 
     function test_RevertIf_InitializeImplementation() public {