openzeppelin-contracts/certora/harnesses/GovernorHarness.sol

import "../../contracts/governance/Governor.sol";

contract GovernorHarness is Governor {

    function isExecuted(uint256 proposalId) public view returns (bool) {
        return _proposals[proposalId].executed;
    }
    
    function isCanceled(uint256 proposalId) public view returns (bool) {
        return _proposals[proposalId].canceled;
    }


    function initialized(uint256 proposalId) public view returns (bool){
        if (_proposals[proposalId].voteStart._deadline != 0 && _proposals[proposalId].voteEnd._deadline != 0) {
            return true;
        }
        return false;
    }


    mapping(uint256 => uint256) _quorum;

    function quorum(uint256 blockNumber) public view override virtual returns (uint256) {
        return _quorum[blockNumber];
    }


    mapping (address => mapping (uint256 => uint256)) _getVotes;

    function getVotes(address account, uint256 blockNumber) public view override virtual returns (uint256) {
        return _getVotes[account][blockNumber];
    }


    mapping (uint256 => bool) __quoromReached;

    function _quorumReached(uint256 proposalId) public view override virtual returns (bool) {
        return __quoromReached[proposalId];
    }


    mapping (uint256 => bool) __voteSucceeded;

    function _voteSucceeded(uint256 proposalId) public view override virtual returns (bool) {
        return __voteSucceeded[proposalId];
    }


    //string _COUNTING_MODE;
    function COUNTING_MODE() public pure override virtual returns (string memory) {
        return "dummy";
    }


    mapping(uint256 => mapping(address => bool)) _hasVoted;

    function hasVoted(uint256 proposalId, address account) public view override virtual returns (bool) {
        return _hasVoted[proposalId][account];
    }


    uint256 _votingDelay;

    function votingDelay() public view override virtual returns (uint256) {
        return _votingDelay;
    }


    uint256 _votingPeriod;
    
    function votingPeriod() public view override virtual returns (uint256) {
        return _votingPeriod;
    }


    constructor(string memory name) Governor(name) {}

    // _countVots == Sum of castVote
    // 
    // RHS:
    // 1) use counter_vote_power as a counter
    // 2) use counter_vote_power as a temp var for a ghost
    // 
    // LHS:
    // mapping of count
    // countMap

    mapping(uint256 => uint256) counted_weight;

    // uint decision;
    // uint numberOfOptions;
    function _countVote(
        uint256 proposalId,
        address account,
        uint8 support,
        uint256 weight
    ) internal override virtual {
        counted_weight[proposalId] += weight;
    }

    mapping(uint256 => uint256) public counter_vote_power_by_id;
    mapping(uint256 => uint256) public ghost_vote_power_by_id;
    
    function castVote(uint256 proposalId, uint8 support) public virtual override returns (uint256) {
        address voter = _msgSender();
        // 2)
        ghost_vote_power_by_id[proposalId] = _castVote(proposalId, voter, support, "");
        
        // 1)
        counter_vote_power_by_id[proposalId] += ghost_vote_power_by_id[proposalId];

        // return _castVote(proposalId, voter, support, "");
        return ghost_vote_power_by_id[proposalId];
    }

    function castVoteWithReason(
        uint256 proposalId,
        uint8 support,
        string calldata reason
    ) public virtual override returns (uint256) {
        address voter = _msgSender();
        // 2)
        ghost_vote_power_by_id[proposalId] = _castVote(proposalId, voter, support, reason);
        
        // 1)
        counter_vote_power_by_id[proposalId] += ghost_vote_power_by_id[proposalId];

        return ghost_vote_power_by_id[proposalId];
    }

    function castVoteBySig(
        uint256 proposalId,
        uint8 support,
        uint8 v,
        bytes32 r,
        bytes32 s
    ) public virtual override returns (uint256) {
        address voter = ECDSA.recover(
            _hashTypedDataV4(keccak256(abi.encode(BALLOT_TYPEHASH, proposalId, support))),
            v,
            r,
            s
        );
        // 2)
        ghost_vote_power_by_id[proposalId] = _castVote(proposalId, voter, support, "");

        // 1)
        counter_vote_power_by_id[proposalId] += ghost_vote_power_by_id[proposalId];

        return ghost_vote_power_by_id[proposalId];
    }
}