GovernorBasicHarness.sol 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. // SPDX-License-Identifier: MIT
  2. pragma solidity ^0.8.2;
  3. import "../../contracts/governance/Governor.sol";
  4. import "../../contracts/governance/extensions/GovernorCountingSimple.sol";
  5. import "../../contracts/governance/extensions/GovernorVotes.sol";
  6. import "../../contracts/governance/extensions/GovernorVotesQuorumFraction.sol";
  7. import "../../contracts/governance/extensions/GovernorTimelockCompound.sol";
  8. /*
  9. Wizard options:
  10. ERC20Votes
  11. TimelockCompound
  12. */
  13. contract GovernorBasicHarness is Governor, GovernorCountingSimple, GovernorVotes, GovernorVotesQuorumFraction, GovernorTimelockCompound {
  14. constructor(ERC20Votes _token, ICompoundTimelock _timelock, string memory name, uint256 quorumFraction)
  15. Governor(name)
  16. GovernorVotes(_token)
  17. GovernorVotesQuorumFraction(quorumFraction)
  18. GovernorTimelockCompound(_timelock)
  19. {}
  20. function isExecuted(uint256 proposalId) public view returns (bool) {
  21. return _proposals[proposalId].executed;
  22. }
  23. function isCanceled(uint256 proposalId) public view returns (bool) {
  24. return _proposals[proposalId].canceled;
  25. }
  26. uint256 _votingDelay;
  27. function votingDelay() public view override virtual returns (uint256) {
  28. return _votingDelay;
  29. }
  30. uint256 _votingPeriod;
  31. function votingPeriod() public view override virtual returns (uint256) {
  32. return _votingPeriod;
  33. }
  34. mapping(uint256 => uint256) public ghost_sum_vote_power_by_id;
  35. function _castVote(
  36. uint256 proposalId,
  37. address account,
  38. uint8 support,
  39. string memory reason
  40. ) internal override virtual returns (uint256) {
  41. uint deltaWeight = super._castVote(proposalId, account, support, reason); //HARNESS
  42. ghost_sum_vote_power_by_id[proposalId] += deltaWeight;
  43. return deltaWeight;
  44. }
  45. // The following functions are overrides required by Solidity.
  46. function quorum(uint256 blockNumber)
  47. public
  48. view
  49. override(IGovernor, GovernorVotesQuorumFraction)
  50. returns (uint256)
  51. {
  52. return super.quorum(blockNumber);
  53. }
  54. function getVotes(address account, uint256 blockNumber)
  55. public
  56. view
  57. override(IGovernor, GovernorVotes)
  58. returns (uint256)
  59. {
  60. return super.getVotes(account, blockNumber);
  61. }
  62. function state(uint256 proposalId)
  63. public
  64. view
  65. override(Governor, GovernorTimelockCompound)
  66. returns (ProposalState)
  67. {
  68. return super.state(proposalId);
  69. }
  70. function propose(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, string memory description)
  71. public
  72. override(Governor, IGovernor)
  73. returns (uint256)
  74. {
  75. return super.propose(targets, values, calldatas, description);
  76. }
  77. function _execute(uint256 proposalId, address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  78. internal
  79. override(Governor, GovernorTimelockCompound)
  80. {
  81. super._execute(proposalId, targets, values, calldatas, descriptionHash);
  82. }
  83. function _cancel(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  84. internal
  85. override(Governor, GovernorTimelockCompound)
  86. returns (uint256)
  87. {
  88. return super._cancel(targets, values, calldatas, descriptionHash);
  89. }
  90. function _executor()
  91. internal
  92. view
  93. override(Governor, GovernorTimelockCompound)
  94. returns (address)
  95. {
  96. return super._executor();
  97. }
  98. function supportsInterface(bytes4 interfaceId)
  99. public
  100. view
  101. override(Governor, GovernorTimelockCompound)
  102. returns (bool)
  103. {
  104. return super.supportsInterface(interfaceId);
  105. }
  106. }