From 4e411f2b373cac801e29f933a4a0e8170a2d63db Mon Sep 17 00:00:00 2001
From: r4bbit <445106+0x-r4bbit@users.noreply.github.com>
Date: Fri, 12 Jan 2024 13:07:31 +0100
Subject: [PATCH] test(StakeManager): add test to check that inital MPs are
 minted

This adds a test that ensures multiplier points are minted with a 1:1
ratio to the stake token amount.

This scenario covers the case where no lock up time is set during
staking.
---
 contracts/StakeManager.sol |  4 +--
 test/StakeManager.t.sol    | 73 ++++++++++++++++++++++++++++++++++++--
 2 files changed, 73 insertions(+), 4 deletions(-)

diff --git a/contracts/StakeManager.sol b/contracts/StakeManager.sol
index a8bb95f..3639b23 100644
--- a/contracts/StakeManager.sol
+++ b/contracts/StakeManager.sol
@@ -38,9 +38,9 @@ contract StakeManager is Ownable {
     uint256 public constant MP_APY = 1;
     uint256 public constant MAX_BOOST = 4;
 
-    mapping(address => Account) accounts;
+    mapping(address => Account) public accounts;
     mapping(uint256 => Epoch) epochs;
-    mapping(bytes32 => bool) isVault;
+    mapping(bytes32 => bool) public isVault;
 
     uint256 public currentEpoch;
     uint256 public pendingReward;
diff --git a/test/StakeManager.t.sol b/test/StakeManager.t.sol
index 97322b4..919178a 100644
--- a/test/StakeManager.t.sol
+++ b/test/StakeManager.t.sol
@@ -18,6 +18,7 @@ contract StakeManagerTest is Test {
     address internal stakeToken;
     address internal deployer;
     address internal testUser = makeAddr("testUser");
+    address internal testUser2 = makeAddr("testUser2");
 
     function setUp() public virtual {
         Deploy deployment = new Deploy();
@@ -40,8 +41,10 @@ contract StakeManagerTest is Test {
         vm.prank(owner);
         vault = vaultFactory.createVault();
 
-        vm.prank(deployer);
-        stakeManager.setVault(address(vault).codehash);
+        if (!stakeManager.isVault(address(vault).codehash)) {
+            vm.prank(deployer);
+            stakeManager.setVault(address(vault).codehash);
+        }
     }
 }
 
@@ -71,6 +74,32 @@ contract StakeTest is StakeManagerTest {
         vm.expectRevert(StakeManager.StakeManager__InvalidLockupPeriod.selector);
         userVault.stake(100, lockTime);
     }
+
+    function test_StakeWithoutLockUpTimeMintsMultiplierPoints() public {
+        // ensure user has funds
+        deal(stakeToken, testUser, 1000);
+        StakeVault userVault = _createTestVault(testUser);
+
+        vm.startPrank(testUser);
+        ERC20(stakeToken).approve(address(userVault), 100);
+
+        // stake without lockup time
+        userVault.stake(100, 0);
+
+        (,, uint256 multiplierPoints,,,) = stakeManager.accounts(address(userVault));
+
+        // total multiplier poitn supply
+        assertEq(stakeManager.multiplierSupply(), 100);
+        // user multiplier points
+        assertEq(multiplierPoints, 100);
+
+        userVault.unstake(100);
+
+        // multiplierpoints are burned after unstaking
+        (,, multiplierPoints,,,) = stakeManager.accounts(address(userVault));
+        assertEq(stakeManager.multiplierSupply(), 0);
+        assertEq(multiplierPoints, 0);
+    }
 }
 
 contract UnstakeTest is StakeManagerTest {
@@ -215,3 +244,43 @@ contract ExecuteAccountTest is StakeManagerTest {
         stakeManager.executeAccount(address(userVault), currentEpoch + 1);
     }
 }
+
+contract UserFlowsTest is StakeManagerTest {
+    function setUp() public override {
+        StakeManagerTest.setUp();
+    }
+
+    function test_StakedSupplyShouldIncreaseAndDecreaseAgain() public {
+        // ensure users have funds
+        deal(stakeToken, testUser, 1000);
+        deal(stakeToken, testUser2, 1000);
+
+        StakeVault userVault = _createTestVault(testUser);
+        StakeVault user2Vault = _createTestVault(testUser2);
+
+        vm.startPrank(testUser);
+        // approve user vault to spend user tokens
+        ERC20(stakeToken).approve(address(userVault), 100);
+        userVault.stake(100, 0);
+        vm.stopPrank();
+
+        vm.startPrank(testUser2);
+        ERC20(stakeToken).approve(address(user2Vault), 100);
+        user2Vault.stake(100, 0);
+        vm.stopPrank();
+
+        assertEq(ERC20(stakeToken).balanceOf(address(userVault)), 100);
+        assertEq(ERC20(stakeToken).balanceOf(address(user2Vault)), 100);
+        assertEq(stakeManager.stakeSupply(), 200);
+
+        vm.startPrank(testUser);
+        userVault.unstake(100);
+        assertEq(ERC20(stakeToken).balanceOf(address(userVault)), 0);
+        assertEq(stakeManager.stakeSupply(), 100);
+
+        vm.startPrank(testUser2);
+        user2Vault.unstake(100);
+        assertEq(ERC20(stakeToken).balanceOf(address(user2Vault)), 0);
+        assertEq(stakeManager.stakeSupply(), 0);
+    }
+}