From cd3f238a08d899e997b9728181d95515c612299d Mon Sep 17 00:00:00 2001
From: Ricardo Guilherme Schmidt <3esmit@gmail.com>
Date: Mon, 4 Mar 2024 12:11:23 -0300
Subject: [PATCH] chore(StakeManager.t): add test for unstake more than balance

---
 test/StakeManager.t.sol | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/test/StakeManager.t.sol b/test/StakeManager.t.sol
index 03f9d84..e26e7f8 100644
--- a/test/StakeManager.t.sol
+++ b/test/StakeManager.t.sol
@@ -317,6 +317,14 @@ contract UnstakeTest is StakeManagerTest {
         assertEq(totalSupplyMPAfter, totalSupplyMPBefore - (currentMPBefore * percentToBurn / 100));
         assertEq(ERC20(stakeToken).balanceOf(testUser), unstakeAmount);
     }
+
+    function test_RevertWhen_AmountMoreThanBalance() public {
+        uint256 stakeAmount = 100;
+        StakeVault userVault = _createStakingAccount(testUser, stakeAmount);
+        vm.startPrank(testUser);
+        vm.expectRevert(StakeManager.StakeManager__InsufficientFunds.selector);
+        userVault.unstake(stakeAmount + 1);
+    }
 }
 
 contract LockTest is StakeManagerTest {