test(StakeManager): expand on deployment test (#27)

Added some additional assertions related to access control and exposed
properties.
This commit is contained in:
r4bbit 2023-10-10 13:53:03 +02:00 committed by GitHub
parent abe11d6806
commit 725d380547
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 13 additions and 5 deletions

1
.gas-snapshot Normal file
View File

@ -0,0 +1 @@
StakeManagerTest:testDeployment() (gas: 26172)

View File

@ -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));

View File

@ -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 });

View File

@ -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);
}
}