mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import pytest
|
|
|
|
from bigchaindb.core import Bigchain
|
|
from bigchaindb.voting import (count_votes, partition_eligible_votes,
|
|
decide_votes, INVALID, VALID, UNDECIDED)
|
|
|
|
|
|
################################################################################
|
|
# Tests for checking vote eligibility
|
|
|
|
|
|
def test_partition_eligible_votes():
|
|
nodes = list(map(Bigchain, 'abc'))
|
|
votes = [n.vote('block', 'a', True) for n in nodes]
|
|
|
|
el, inel = partition_eligible_votes(votes, 'abc', lambda _: True)
|
|
|
|
assert el == votes
|
|
assert inel == []
|
|
|
|
|
|
def test_count_votes():
|
|
nodes = list(map(Bigchain, 'abc'))
|
|
votes = [n.vote('block', 'a', True) for n in nodes]
|
|
|
|
assert count_votes(votes) == ({
|
|
'n_valid': 3,
|
|
'n_invalid': 0,
|
|
'n_agree_prev_block': 3
|
|
}, {})
|
|
|
|
|
|
################################################################################
|
|
# Tests for vote decision making
|
|
|
|
|
|
DECISION_TESTS = [dict(
|
|
zip(['n_voters', 'n_valid', 'n_invalid', 'n_agree_prev_block'], t))
|
|
for t in [
|
|
(1, 1, 1, 1),
|
|
(2, 2, 1, 2),
|
|
(3, 2, 2, 2),
|
|
(4, 3, 2, 3),
|
|
(5, 3, 3, 3),
|
|
(6, 4, 3, 4),
|
|
(7, 4, 4, 4),
|
|
(8, 5, 4, 5),
|
|
]
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize('kwargs', DECISION_TESTS)
|
|
def test_decide_votes_valid(kwargs):
|
|
kwargs = kwargs.copy()
|
|
kwargs['n_invalid'] = 0
|
|
assert decide_votes(**kwargs) == VALID
|
|
kwargs['n_agree_prev_block'] -= 1
|
|
assert decide_votes(**kwargs) == INVALID
|
|
kwargs['n_valid'] -= 1
|
|
assert decide_votes(**kwargs) == UNDECIDED
|
|
|
|
|
|
@pytest.mark.parametrize('kwargs', DECISION_TESTS)
|
|
def test_decide_votes_invalid(kwargs):
|
|
kwargs = kwargs.copy()
|
|
kwargs['n_valid'] = 0
|
|
assert decide_votes(**kwargs) == INVALID
|
|
kwargs['n_invalid'] -= 1
|
|
assert decide_votes(**kwargs) == UNDECIDED
|
|
|
|
|
|
def test_decide_votes_checks_arguments():
|
|
with pytest.raises(ValueError):
|
|
decide_votes(n_voters=1, n_valid=2, n_invalid=0, n_agree_prev_block=0)
|
|
with pytest.raises(ValueError):
|
|
decide_votes(n_voters=1, n_valid=0, n_invalid=2, n_agree_prev_block=0)
|
|
with pytest.raises(ValueError):
|
|
decide_votes(n_voters=1, n_valid=0, n_invalid=0, n_agree_prev_block=2)
|