unit-e/test/functional/feature_finalizer.py at master · kostyantyn/unit-e · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/usr/bin/env python3
# Copyright (c) 2019 The Unit-e developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""
FeatureFinalizerTest tests the following:
1. Finalizer can vote after the restart
"""

from test_framework.test_framework import UnitETestFramework
from test_framework.util import (
    assert_equal,
    assert_finalizationstate,
    connect_nodes,
    generate_block,
    sync_blocks,
)

class FeatureFinalizerTest(UnitETestFramework):
    def set_test_params(self):
        self.num_nodes = 2
        self.extra_args = [
            ['-esperanzaconfig={"epochLength": 5, "minDepositSize": 1500}'],
            ['-esperanzaconfig={"epochLength": 5, "minDepositSize": 1500}', '-validating=1'],
        ]
        self.setup_clean_chain = True

    def setup_network(self):
        self.setup_nodes()
        p, v = self.nodes
        connect_nodes(p, v.index)

    def run_test(self):
        # Check finalizer can vote after restart
        p, v = self.nodes
        self.setup_stake_coins(p, v)
        self.generate_sync(p)

        self.log.info("Setup deposit")
        v.new_address = v.getnewaddress("", "legacy")
        tx = v.deposit(v.new_address, 1500)
        self.wait_for_transaction(tx)
        generate_block(p)
        sync_blocks([p, v])

        self.log.info("Restart validator")
        self.restart_node(v.index)

        self.log.info("Leave insta justification")
        for _ in range(14):
            generate_block(p)
        assert_equal(p.getblockcount(), 16)
        assert_finalizationstate(p, {"currentEpoch": 4,
                                     "lastJustifiedEpoch": 2,
                                     "lastFinalizedEpoch": 2,
                                     "validators": 1})

        self.log.info("Check finalizer votes after restart")
        self.wait_for_vote_and_disconnect(finalizer=v, node=p)
        generate_block(p)

        assert_equal(p.getblockcount(), 17)
        assert_finalizationstate(p, {"currentEpoch": 4,
                                     "lastJustifiedEpoch": 3,
                                     "lastFinalizedEpoch": 3,
                                     "validators": 1})


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