diff --git a/test/TestContractsState.sol b/test/TestContractsState.sol new file mode 100644 index 0000000..b57f583 --- /dev/null +++ b/test/TestContractsState.sol @@ -0,0 +1,68 @@ +// SPDX-License-Identifier: MIT +pragma solidity ^0.6.12; +pragma experimental ABIEncoderV2; + +import { MockProposal } from "./MockProposal.sol"; +import { console2 } from "@forge-std/console2.sol"; +import { GovernancePatchUpgrade } from "@root/v4-patch/GovernancePatchUpgrade.sol"; +import { RelayerRegistry } from "@root/v4-patch/RelayerRegistry.sol"; +import { TornadoStakingRewards } from "@root/v4-patch/TornadoStakingRewards.sol"; + +contract TestContractsState is MockProposal { + function testLockedBalanceSaved() public { + uint256 lockedBalanceBeforeExecution = governance.lockedBalance(TEST_REAL_ADDRESS_WITH_BALANCE); + console2.log("User locked balance before execution: %s TORN", lockedBalanceBeforeExecution / 10 ** 18); + + createAndExecuteProposal(); + + uint256 lockedBalanceAfterExecution = governance.lockedBalance(TEST_REAL_ADDRESS_WITH_BALANCE); + console2.log("User locked balance before execution: %s TORN", lockedBalanceAfterExecution / 10 ** 18); + + require(lockedBalanceBeforeExecution == lockedBalanceAfterExecution, "Wrong locked balance after execution"); + } + + function testGovernanceStakingStateChanged() public { + TornadoStakingRewards oldStaking = TornadoStakingRewards(getStakingProxyAddress()); + uint256 accumulatedRewardsBeforeExecution = oldStaking.checkReward(TEST_REAL_ADDRESS_WITH_BALANCE); + console2.log("User rewards balance (bugged) before execution: %s TORN", accumulatedRewardsBeforeExecution / 10 ** 18); + console2.log("Bugged value of accumulated rewards per TORN: %s", oldStaking.accumulatedRewardPerTorn()); + + createAndExecuteProposal(); + + TornadoStakingRewards newStaking = TornadoStakingRewards(getStakingProxyAddress()); + uint256 accumulatedRewardsPerTORNAfterExecution = newStaking.accumulatedRewardPerTorn(); + uint256 accumulatedRewardsAfterExecution = newStaking.checkReward(TEST_REAL_ADDRESS_WITH_BALANCE); + + console2.log("User rewards balance before execution: %s TORN", accumulatedRewardsAfterExecution / 10 ** 18); + console2.log("Value of accumulated rewards per TORN after contract redeployment: %s", accumulatedRewardsPerTORNAfterExecution); + + require(accumulatedRewardsBeforeExecution >= accumulatedRewardsAfterExecution, "Wtf"); + require(accumulatedRewardsAfterExecution == 0, "Accumulated rewards isn't nullified"); + require(accumulatedRewardsPerTORNAfterExecution == 0, "Accumulated rewards per TORN isn't nullified"); + } + + function testRelayerRegistryStateSaved() public { + RelayerRegistry registry = RelayerRegistry(getRelayerRegistryProxyAddress()); + + bool isRelayerRegisteredBeforeExecution = registry.isRelayer(TEST_RELAYER_ADDRESS); + uint256 relayerStakedBalanceBeforeExecution = registry.getRelayerBalance(TEST_RELAYER_ADDRESS); + console2.log( + "Relayer balance in relayer registry contract before proposal execution: %s TORN", + relayerStakedBalanceBeforeExecution / 10 ** 18 + ); + + require(isRelayerRegisteredBeforeExecution, "Relayer not registered"); + + createAndExecuteProposal(); + + bool isRelayerRegisteredAfterExecution = registry.isRelayer(TEST_RELAYER_ADDRESS); + uint256 relayerStakedBalanceAfterExecution = registry.getRelayerBalance(TEST_RELAYER_ADDRESS); + + console2.log( + "Relayer balance in relayer registry contract after proposal execution: %s TORN", relayerStakedBalanceAfterExecution / 10 ** 18 + ); + + require(isRelayerRegisteredAfterExecution, "Relayer isn't registered after proposal execution"); + require(relayerStakedBalanceBeforeExecution == relayerStakedBalanceAfterExecution, "Relayer stake balance differs after execution"); + } +}