Skip to content

Instantly share code, notes, and snippets.

@TopLiu
Created August 3, 2020 08:32
Show Gist options
  • Save TopLiu/49f2349b5bb4667ee25e6e7e0dd4f315 to your computer and use it in GitHub Desktop.
Save TopLiu/49f2349b5bb4667ee25e6e7e0dd4f315 to your computer and use it in GitHub Desktop.

Revisions

  1. TopLiu created this gist Aug 3, 2020.
    65 changes: 65 additions & 0 deletions ballot.sol
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,65 @@
    pragma solidity >=0.4.22 <0.6.0;
    contract Ballot {

    struct Voter {
    uint weight;
    bool voted;
    uint8 vote;
    address delegate;
    }
    struct Proposal {
    uint voteCount;
    }

    address chairperson;
    mapping(address => Voter) voters;
    Proposal[] proposals;

    /// Create a new ballot with $(_numProposals) different proposals.
    constructor(uint8 _numProposals) public {
    chairperson = msg.sender;
    voters[chairperson].weight = 1;
    proposals.length = _numProposals;
    }

    /// Give $(toVoter) the right to vote on this ballot.
    /// May only be called by $(chairperson).
    function giveRightToVote(address toVoter) public {
    if (msg.sender != chairperson || voters[toVoter].voted) return;
    voters[toVoter].weight = 1;
    }

    /// Delegate your vote to the voter $(to).
    function delegate(address to) public {
    Voter storage sender = voters[msg.sender]; // assigns reference
    if (sender.voted) return;
    while (voters[to].delegate != address(0) && voters[to].delegate != msg.sender)
    to = voters[to].delegate;
    if (to == msg.sender) return;
    sender.voted = true;
    sender.delegate = to;
    Voter storage delegateTo = voters[to];
    if (delegateTo.voted)
    proposals[delegateTo.vote].voteCount += sender.weight;
    else
    delegateTo.weight += sender.weight;
    }

    /// Give a single vote to proposal $(toProposal).
    function vote(uint8 toProposal) public {
    Voter storage sender = voters[msg.sender];
    if (sender.voted || toProposal >= proposals.length) return;
    sender.voted = true;
    sender.vote = toProposal;
    proposals[toProposal].voteCount += sender.weight;
    }

    function winningProposal() public view returns (uint8 _winningProposal) {
    uint256 winningVoteCount = 0;
    for (uint8 prop = 0; prop < proposals.length; prop++)
    if (proposals[prop].voteCount > winningVoteCount) {
    winningVoteCount = proposals[prop].voteCount;
    _winningProposal = prop;
    }
    }
    }
    21 changes: 21 additions & 0 deletions ballot_test.sol
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,21 @@

    pragma solidity >=0.4.22 <0.6.0;
    import "remix_tests.sol"; // this import is automatically injected by Remix.
    import "./ballot.sol";

    contract test3 {

    Ballot ballotToTest;
    function beforeAll () public {
    ballotToTest = new Ballot(2);
    }

    function checkWinningProposal () public {
    ballotToTest.vote(1);
    Assert.equal(ballotToTest.winningProposal(), uint(1), "1 should be the winning proposal");
    }

    function checkWinninProposalWithReturnValue () public view returns (bool) {
    return ballotToTest.winningProposal() == 1;
    }
    }
    8 changes: 8 additions & 0 deletions lclTest.sol
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,8 @@

    pragma solidity >=0.4.22 <0.6.0;

    contract HelloWorld {
    function helloWorld() external pure returns (string memory) {
    return "Hello, World!";
    }
    }