chore: make functions external

mento-protocol · Jan 30, 2024 · 4ab9755 · 4ab9755
1 parent 9d1a579
commit 4ab9755
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/contracts/vesting/VestingLock.sol b/contracts/vesting/VestingLock.sol
@@ -94,15 +94,15 @@ contract VestingLock {
   /**
    * @notice returns the amount of tokens not yet vested.
    */
-  function getLockedHedgeyBalance() public view returns (uint256) {
+  function getLockedHedgeyBalance() external view returns (uint256) {
     require(planId != 0, "VestingLock: no plan id set");
     return ((totalAmountToLock * 2) - totalUnlockedTokens);
   }
 
   /**
    * @notice returns the amount of tokens redeemable from hedgey contract.
    */
-  function getRedeemableHedgeyBalance() public view returns (uint256) {
+  function getRedeemableHedgeyBalance() external view returns (uint256) {
     require(planId != 0, "VestingLock: no plan id set");
 
     (uint256 balance, , ) = ITokenVestingPlans(hedgeyVestingContract).planBalanceOf(
@@ -123,7 +123,7 @@ contract VestingLock {
   /**
    * @notice returns the amount of tokens redeemable from veMentoLocking contract.
    */
-  function getRedeemableVeMentoBalance() public returns (uint256) {
+  function getRedeemableVeMentoBalance() external returns (uint256) {
     return (uint256(ILockingExtended(veMentoLockingContract).getAvailableForWithdraw(address(this))));
   }
 

diff --git a/test/vesting/VestingLock.t.sol b/test/vesting/VestingLock.t.sol
@@ -78,7 +78,7 @@ contract VestingLockTest_constructor is VestingLockTest {
   }
 
   function test_constructor_whenLockingSlopeEndWeekIsSmallerThanLockingCliffEndWeek_shouldRevert() public {
-    vm.expectRevert("VestingLock: lockingSlopeEndWeek is smaller than lockingCliffEndWeek");
+    vm.expectRevert("VestingLock: lockingSlopeEndWeek is smaller lockingCliffEndWeek");
     vestingLock = new VestingLock(beneficiary, hedgeyVestingAddr, veMentoLockingAddr, 10, 5);
   }