GovernorBasicHarness.sol 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  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. uint256 deltaWeight = super._castVote(proposalId, account, support, reason); //HARNESS
  42. ghost_sum_vote_power_by_id[proposalId] += deltaWeight;
  43. return deltaWeight;
  44. }
  45. /*
  46. mapping (address => mapping (uint256 => uint256)) _getVotes;
  47. function getVotesHarnness(address account, uint256 blockNumber) public {
  48. _getVotes[account][blockNumber] = getVotes(account, blockNumber);
  49. }
  50. */
  51. // The following functions are overrides required by Solidity.
  52. function quorum(uint256 blockNumber)
  53. public
  54. view
  55. override(IGovernor, GovernorVotesQuorumFraction)
  56. returns (uint256)
  57. {
  58. return super.quorum(blockNumber);
  59. }
  60. function getVotes(address account, uint256 blockNumber)
  61. public
  62. view
  63. override(IGovernor, GovernorVotes)
  64. returns (uint256)
  65. {
  66. return super.getVotes(account, blockNumber);
  67. }
  68. function state(uint256 proposalId)
  69. public
  70. view
  71. override(Governor, GovernorTimelockCompound)
  72. returns (ProposalState)
  73. {
  74. return super.state(proposalId);
  75. }
  76. function propose(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, string memory description)
  77. public
  78. override(Governor, IGovernor)
  79. returns (uint256)
  80. {
  81. return super.propose(targets, values, calldatas, description);
  82. }
  83. function _execute(uint256 proposalId, address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  84. internal
  85. override(Governor, GovernorTimelockCompound)
  86. {
  87. super._execute(proposalId, targets, values, calldatas, descriptionHash);
  88. }
  89. function _cancel(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
  90. internal
  91. override(Governor, GovernorTimelockCompound)
  92. returns (uint256)
  93. {
  94. return super._cancel(targets, values, calldatas, descriptionHash);
  95. }
  96. function _executor()
  97. internal
  98. view
  99. override(Governor, GovernorTimelockCompound)
  100. returns (address)
  101. {
  102. return super._executor();
  103. }
  104. function supportsInterface(bytes4 interfaceId)
  105. public
  106. view
  107. override(Governor, GovernorTimelockCompound)
  108. returns (bool)
  109. {
  110. return super.supportsInterface(interfaceId);
  111. }
  112. }