mock compound govenror

Author: anajuliabit

src/proposals/GovernorBravoProposal.sol

@@ -3,11 +3,7 @@ pragma solidity ^0.8.0;
 
 import "@forge-std/console.sol";
 
-import {
-    IGovernorBravo,
-    ITimelockBravo,
-    IERC20VotesComp
-} from "@interface/IGovernorBravo.sol";
+import {IERC20VotesComp, IGovernorBravo, ITimelockBravo} from "@interface/IGovernorBravo.sol";
 
 import {Address} from "@utils/Address.sol";
 
@@ -26,18 +22,8 @@ abstract contract GovernorBravoProposal is Proposal {
     }
 
     /// @notice Getter function for `GovernorBravoDelegate.propose()` calldata
-    function getCalldata()
-        public
-        view
-        virtual
-        override
-        returns (bytes memory data)
-    {
-        (
-            address[] memory targets,
-            uint256[] memory values,
-            bytes[] memory calldatas
-        ) = getProposalActions();
+    function getCalldata() public view virtual override returns (bytes memory data) {
+        (address[] memory targets, uint256[] memory values, bytes[] memory calldatas) = getProposalActions();
         string[] memory signatures = new string[](targets.length);
 
         data = abi.encodeWithSignature(
@@ -52,21 +38,12 @@ abstract contract GovernorBravoProposal is Proposal {
 
     /// @notice Check if there are any on-chain proposals that match the
     /// proposal calldata
-    function getProposalId()
-        public
-        view
-        override
-        returns (uint256 proposalId)
-    {
+    function getProposalId() public view override returns (uint256 proposalId) {
         uint256 proposalCount = governor.proposalCount();
 
         while (proposalCount > 0) {
-            (
-                address[] memory targets,
-                uint256[] memory values,
-                string[] memory signatures,
-                bytes[] memory calldatas
-            ) = governor.getActions(proposalCount);
+            (address[] memory targets, uint256[] memory values, string[] memory signatures, bytes[] memory calldatas) =
+                governor.getActions(proposalCount);
 
             bytes memory onchainCalldata = abi.encodeWithSignature(
                 "propose(address[],uint256[],string[],bytes[],string)",
@@ -81,10 +58,7 @@ abstract contract GovernorBravoProposal is Proposal {
 
             if (keccak256(proposalCalldata) == keccak256(onchainCalldata)) {
                 if (DEBUG) {
-                    console.log(
-                        "Proposal calldata matches on-chain calldata with proposalId: ",
-                        proposalCount
-                    );
+                    console.log("Proposal calldata matches on-chain calldata with proposalId: ", proposalCount);
                 }
                 return proposalCount;
             }
@@ -98,13 +72,19 @@ abstract contract GovernorBravoProposal is Proposal {
     function simulate() public override {
         address proposerAddress = address(1);
         IERC20VotesComp governanceToken = governor.comp();
+        ITimelockBravo timelock = ITimelockBravo(governor.timelock());
+
+        // Compound has migrated the Governor to 0x309a862bbC1A00e45506cB8A802D1ff10004c8C0,
+        // so the timelock admin doesn't correspond to the governor address on mainnet anymore
+        // This is a workaround since this proposal is not compatible with the new Compound governor
+        // In Compound's timelock, admin is stored at slot 0
+        vm.store(address(timelock), bytes32(uint256(0)), bytes32(uint256(uint160(address(governor)))));
+
         {
             // Ensure proposer has meets minimum proposal threshold and quorum votes to pass the proposal
             uint256 quorumVotes = governor.quorumVotes();
             uint256 proposalThreshold = governor.proposalThreshold();
-            uint256 votingPower = quorumVotes > proposalThreshold
-                ? quorumVotes
-                : proposalThreshold;
+            uint256 votingPower = quorumVotes > proposalThreshold ? quorumVotes : proposalThreshold;
             deal(address(governanceToken), proposerAddress, votingPower);
             // Delegate proposer's votes to itself
             vm.prank(proposerAddress);
@@ -120,40 +100,31 @@ abstract contract GovernorBravoProposal is Proposal {
         uint256 proposalId = abi.decode(data, (uint256));
 
         // Check proposal is in Pending state
-        require(
-            governor.state(proposalId) == IGovernorBravo.ProposalState.Pending
-        );
+        require(governor.state(proposalId) == IGovernorBravo.ProposalState.Pending);
 
         // Roll to Active state (voting period)
         vm.roll(block.number + governor.votingDelay() + 1);
-        require(
-            governor.state(proposalId) == IGovernorBravo.ProposalState.Active
-        );
+        require(governor.state(proposalId) == IGovernorBravo.ProposalState.Active);
 
         // Vote YES
         vm.prank(proposerAddress);
         governor.castVote(proposalId, 1);
 
         // Roll to allow proposal state transitions
         vm.roll(block.number + governor.votingPeriod());
-        require(
-            governor.state(proposalId) == IGovernorBravo.ProposalState.Succeeded
-        );
+        require(governor.state(proposalId) == IGovernorBravo.ProposalState.Succeeded);
 
         // Queue the proposal
         governor.queue(proposalId);
-        require(
-            governor.state(proposalId) == IGovernorBravo.ProposalState.Queued
-        );
+        require(governor.state(proposalId) == IGovernorBravo.ProposalState.Queued);
 
         // Warp to allow proposal execution on timelock
         ITimelockBravo timelock = ITimelockBravo(governor.timelock());
+
         vm.warp(block.timestamp + timelock.delay());
 
         // Execute the proposal
         governor.execute(proposalId);
-        require(
-            governor.state(proposalId) == IGovernorBravo.ProposalState.Executed
-        );
+        require(governor.state(proposalId) == IGovernorBravo.ProposalState.Executed);
     }
 }