feature_part_vote_2.py

#!/usr/bin/env python3
# Copyright (c) 2017-2021 The Particl Core developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.

from test_framework.test_particl import GhostTestFramework


class VoteTest2(GhostTestFramework):
    def set_test_params(self):
        self.setup_clean_chain = True
        self.num_nodes = 4
        self.extra_args = [ ['-debug','-noacceptnonstdtxn','-reservebalance=10000000'] for i in range(self.num_nodes)]

    def skip_test_if_missing_module(self):
        self.skip_if_no_wallet()

    def setup_network(self, split=False):
        self.add_nodes(self.num_nodes, extra_args=self.extra_args)
        self.start_nodes()

        self.connect_nodes_bi(0, 1)
        self.connect_nodes_bi(0, 2)
        self.connect_nodes_bi(0, 3)

        self.sync_all()

    def run_test(self):
        nodes = self.nodes

        self.import_genesis_coins_a(nodes[0])

        ro = nodes[0].setvote(1, 2, 0, 10)
        assert(ro['result'] == 'Voting for option 2 on proposal 1')
        assert(ro['from_height'] == 0)
        assert(ro['to_height'] == 10)

        ro = nodes[0].votehistory()
        assert(len(ro) == 1)
        assert(ro[0]['proposal'] == 1)
        assert(ro[0]['option'] == 2)

        self.stakeBlocks(1)

        ro = nodes[0].tallyvotes(1, 0, 10)
        assert(ro['blocks_counted'] == 1)
        assert(ro['Option 2'] == '1, 100.00%')


        ro = nodes[0].setvote(1, 3, 0, 10)
        assert(ro['result'] == 'Voting for option 3 on proposal 1')
        assert(ro['from_height'] == 0)
        assert(ro['to_height'] == 10)

        ro = nodes[0].votehistory()
        assert(len(ro) == 2)

        self.stakeBlocks(1)

        ro = nodes[0].tallyvotes(1, 0, 10)
        assert(ro['blocks_counted'] == 2)
        assert(ro['Option 3'] == '1, 50.00%')

        nodes[0].setvote(0, 0, 0, 0)
        assert(len(nodes[0].votehistory()) == 0)

        ro = nodes[0].setvote(1, 1, 0, 2)
        assert(len(nodes[0].votehistory()) == 1)
        assert(len(nodes[0].votehistory(True, True)) == 0)
        nodes[0].setvote(2, 1, 0, 100)
        nodes[0].setvote(2, 2, 0, 100)

        assert(len(nodes[0].votehistory()) == 3)
        ro = nodes[0].votehistory(True, True)
        assert(len(ro) == 1)
        assert(ro[0]['option'] == 2)
        nodes[0].setvote(2, 3, 100, 200)
        ro = nodes[0].votehistory(True, True)
        assert(len(ro) == 2)
        assert(ro[0]['option'] == 2)
        assert(ro[1]['option'] == 3)
        nodes[0].setvote(2, 4, 100, 150)
        ro = nodes[0].votehistory(True, True)
        assert(ro[0]['option'] == 2)
        assert(ro[1]['option'] == 4)
        assert(ro[2]['option'] == 3)

        nodes[0].setvote(2, 5, 160, 200)
        ro = nodes[0].votehistory(True, True)
        assert(len(ro) == 4)
        assert(ro[0]['option'] == 2)
        assert(ro[1]['option'] == 4)
        assert(ro[2]['option'] == 3)
        assert(ro[3]['option'] == 5)

        nodes[0].setvote(2, 6, 140, 160)
        ro = nodes[0].votehistory(True, True)
        assert(len(ro) == 4)
        assert(ro[0]['option'] == 2)
        assert(ro[1]['option'] == 4)
        assert(ro[2]['option'] == 6)
        assert(ro[3]['option'] == 5)

        nodes[0].setvote(3, 6, 140, 160)
        ro = nodes[0].votehistory(True, True)
        assert(len(ro) == 4)
        assert(ro[0]['option'] == 2)
        assert(ro[1]['option'] == 4)
        assert(ro[2]['option'] == 6)
        assert(ro[3]['option'] == 5)


if __name__ == '__main__':
    VoteTest2().main()