test(StakeManager): expand on deployment test

Added some additional assertions related to access control and exposed properties.
logos-co · Oct 9, 2023 · 2122001 · 2122001
1 parent fbcd7b5
commit 2122001
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/.gas-snapshot b/.gas-snapshot
@@ -0,0 +1 @@
+StakeManagerTest:testDeployment() (gas: 26172)
diff --git a/script/Deploy.s.sol b/script/Deploy.s.sol
@@ -8,7 +8,7 @@ import { StakeManager } from "../contracts/StakeManager.sol";
 contract Deploy is BaseScript {
     function run() public returns (StakeManager, DeploymentConfig) {
         DeploymentConfig deploymentConfig = new DeploymentConfig(broadcaster);
-        (address token,) = deploymentConfig.activeNetworkConfig();
+        (, address token) = deploymentConfig.activeNetworkConfig();
 
         vm.startBroadcast(broadcaster);
         StakeManager stakeManager = new StakeManager(token, address(0));

diff --git a/script/DeploymentConfig.s.sol b/script/DeploymentConfig.s.sol
@@ -29,7 +29,7 @@ contract DeploymentConfig is Script {
     }
 
     function getOrCreateAnvilEthConfig(address _deployer) public returns (NetworkConfig memory) {
-        vm.startBroadcast();
+        vm.startBroadcast(_deployer);
         MockERC20 token = new MockERC20();
         vm.stopBroadcast();
         return NetworkConfig({ token: address(token), deployer: _deployer });

diff --git a/test/StakeManager.t.sol b/test/StakeManager.t.sol
@@ -7,19 +7,26 @@ import { DeploymentConfig } from "../script/DeploymentConfig.s.sol";
 import { StakeManager } from "../contracts/StakeManager.sol";
 
 contract StakeManagerTest is Test {
-    address internal deployer;
-
     DeploymentConfig internal deploymentConfig;
     StakeManager internal stakeManager;
 
+    address internal stakeToken;
+    address internal deployer;
+
     function setUp() public virtual {
         Deploy deployment = new Deploy();
         (stakeManager, deploymentConfig) = deployment.run();
-        (deployer,) = deploymentConfig.activeNetworkConfig();
+        (deployer, stakeToken) = deploymentConfig.activeNetworkConfig();
     }
 
     function testDeployment() public {
         assertEq(stakeManager.owner(), deployer);
+        assertEq(stakeManager.currentEpoch(), 0);
+        assertEq(stakeManager.pendingReward(), 0);
+        assertEq(stakeManager.multiplierSupply(), 0);
+        assertEq(stakeManager.stakeSupply(), 0);
+        assertEq(address(stakeManager.stakedToken()), stakeToken);
+        assertEq(address(stakeManager.oldManager()), address(0));
         assertEq(stakeManager.totalSupply(), 0);
     }
 }