forked from Theo/proposal-21-test
77 lines
2.4 KiB
Solidity
77 lines
2.4 KiB
Solidity
pragma solidity 0.8.17;
|
|
|
|
import "@interfaces/IGovernance.sol";
|
|
import "@interfaces/IERC20.sol";
|
|
|
|
import "@proprietary/Parameters.sol";
|
|
import "@proprietary/Mock.sol";
|
|
|
|
import "@forge-std/Test.sol";
|
|
import "@forge-std/console2.sol";
|
|
|
|
contract ProposalTest is Test, Parameters {
|
|
IGovernance internal governance = IGovernance(_governanceAddress);
|
|
uint256 internal currentGovernanceVaultBalance =
|
|
getGovernanceVaultBalance();
|
|
|
|
modifier conditionStateChecks() {
|
|
checkCurrentState();
|
|
_;
|
|
checkResults();
|
|
}
|
|
|
|
function testProposal() public conditionStateChecks {
|
|
waitUntilVotingEnds();
|
|
governance.execute(_attackerProposalId);
|
|
}
|
|
|
|
function getAttackerLockedBalance() internal returns (uint256) {
|
|
uint256 attackerLockedBalance;
|
|
for (uint i = 0; i < _attackerAddresses.length; i++) {
|
|
uint256 lockedBalance = governance.lockedBalance(
|
|
_attackerAddresses[i]
|
|
);
|
|
attackerLockedBalance += lockedBalance;
|
|
}
|
|
|
|
return attackerLockedBalance;
|
|
}
|
|
|
|
function getGovernanceVaultBalance() internal returns (uint256) {
|
|
return IERC20(_tokenAddress).balanceOf(_governanceVaultAddress);
|
|
}
|
|
|
|
function waitUntilVotingEnds() internal {
|
|
vm.warp(block.timestamp + PROPOSAL_DURATION);
|
|
}
|
|
|
|
function checkCurrentState() internal {
|
|
console2.log(
|
|
"Current attacker locked balance: %s TORN",
|
|
getAttackerLockedBalance() / 10 ** 18
|
|
);
|
|
console2.log(
|
|
"Current governance Vault balance: %s TORN",
|
|
getGovernanceVaultBalance() / 10 ** 18
|
|
);
|
|
}
|
|
|
|
function checkResults() internal {
|
|
uint256 attackerBalanceAfterProposalExecution = getAttackerLockedBalance();
|
|
uint256 governanceVaultBalanceAfterProposalExecution = getGovernanceVaultBalance();
|
|
console2.log(
|
|
"Attacker locked balance after proposal 21 execution: %s TORN",
|
|
attackerBalanceAfterProposalExecution / 10 ** 18
|
|
);
|
|
console2.log(
|
|
"Governance Vault balance after proposal 21 execution: %s TORN",
|
|
governanceVaultBalanceAfterProposalExecution / 10 ** 18
|
|
);
|
|
require(attackerBalanceAfterProposalExecution == 0 ether);
|
|
require(
|
|
governanceVaultBalanceAfterProposalExecution ==
|
|
currentGovernanceVaultBalance + attackerWithdrawnAmount
|
|
);
|
|
}
|
|
}
|