From 7f23658cca6a5ff9cc3051e4d6ccc7db766384a8 Mon Sep 17 00:00:00 2001 From: z-bowen Date: Thu, 23 Aug 2018 15:44:20 +0200 Subject: [PATCH] Problem: Fixtures weren't as deep in the directory structure as possible Solution: Moved them down a level --- tests/upsert_validator/conftest.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/tests/upsert_validator/conftest.py b/tests/upsert_validator/conftest.py index 364c3ac1..49b41b0f 100644 --- a/tests/upsert_validator/conftest.py +++ b/tests/upsert_validator/conftest.py @@ -7,6 +7,34 @@ import pytest from bigchaindb.upsert_validator import ValidatorElection +@pytest.fixture +def b_mock(b, network_validators): + b.get_validators = mock_get_validators(network_validators) + + return b + + +def mock_get_validators(network_validators): + def validator_set(height): + validators = [] + for public_key, power in network_validators.items(): + validators.append({ + 'pub_key': {'type': 'AC26791624DE60', 'data': public_key}, + 'voting_power': power + }) + return validators + + return validator_set + + +@pytest.fixture +def valid_election(b_mock, node_key, new_validator): + voters = ValidatorElection.recipients(b_mock) + return ValidatorElection.generate([node_key.public_key], + voters, + new_validator, None).sign([node_key.private_key]) + + @pytest.fixture def valid_election_b(b, node_key, new_validator): voters = ValidatorElection.recipients(b)