Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Cleaned up fixtures and reduced their scope to "module."

pull/145/head
jMyles 2017-12-14 21:21:09 -08:00
parent 9cf06c58e5
commit e486d5a2b1
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/fixtures.py
View File

@ -5,12 +5,12 @@ import pytest
from nkms.characters import congregate, Alice, Bob from nkms.characters import congregate, Alice, Bob
from nkms.network import blockchain_client from nkms.network import blockchain_client
from nkms.policy.constants import NON_PAYMENT from nkms.policy.constants import NON_PAYMENT
from nkms.policy.models import PolicyManagerForAlice, Contract from nkms.policy.models import PolicyManagerForAlice
from tests.utilities import NUMBER_OF_URSULAS_IN_NETWORK, MockNetworkyStuff, make_ursulas, \ from tests.utilities import NUMBER_OF_URSULAS_IN_NETWORK, MockNetworkyStuff, make_ursulas, \
URSULA_PORT, EVENT_LOOP URSULA_PORT, EVENT_LOOP
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def idle_policy(alice, bob): def idle_policy(alice, bob):
""" """
Creates a PolicyGroup, in a manner typical of how Alice might do it, with a unique uri. Creates a PolicyGroup, in a manner typical of how Alice might do it, with a unique uri.
@ -29,22 +29,22 @@ def idle_policy(alice, bob):
return policy_group return policy_group
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def enacted_policy(idle_policy, ursulas): def enacted_policy(idle_policy, ursulas):
# Alice has a policy in mind and knows of enough qualifies Ursulas; she crafts an offer for them. # Alice has a policy in mind and knows of enough qualifies Ursulas; she crafts an offer for them.
deposit = NON_PAYMENT deposit = NON_PAYMENT
contract_end_datetime = datetime.datetime.now() + datetime.timedelta(days=5) contract_end_datetime = datetime.datetime.now() + datetime.timedelta(days=5)
contract = Contract(idle_policy.n, deposit, contract_end_datetime) # contract = Contract(idle_policy.n, deposit, contract_end_datetime)
networky_stuff = MockNetworkyStuff(ursulas) networky_stuff = MockNetworkyStuff(ursulas)
idle_policy.find_ursulas(networky_stuff, deposit, found_ursulas = idle_policy.find_ursulas(networky_stuff, deposit, expiration=contract_end_datetime)
expiration=datetime.datetime.now() + datetime.timedelta(days=5)) idle_policy.match_kfrags_to_found_ursulas(found_ursulas)
idle_policy.enact(networky_stuff) # REST call happens here, as does population of TreasureMap. idle_policy.enact(networky_stuff) # REST call happens here, as does population of TreasureMap.
return idle_policy return idle_policy
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def alice(ursulas): def alice(ursulas):
ALICE = Alice() ALICE = Alice()
ALICE.attach_server() ALICE.attach_server()
@ -54,7 +54,7 @@ def alice(ursulas):
return ALICE return ALICE
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def bob(alice, ursulas): def bob(alice, ursulas):
BOB = Bob(alice=alice) BOB = Bob(alice=alice)
BOB.attach_server() BOB.attach_server()
@ -64,13 +64,13 @@ def bob(alice, ursulas):
return BOB return BOB
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def ursulas(): def ursulas():
URSULAS = make_ursulas(NUMBER_OF_URSULAS_IN_NETWORK, URSULA_PORT) URSULAS = make_ursulas(NUMBER_OF_URSULAS_IN_NETWORK, URSULA_PORT)
yield URSULAS yield URSULAS
blockchain_client._ursulas_on_blockchain.clear() blockchain_client._ursulas_on_blockchain.clear()
@pytest.fixture(scope="session") @pytest.fixture(scope="module")
def treasure_map_is_set_on_dht(alice, enacted_policy_group): def treasure_map_is_set_on_dht(alice, enacted_policy):
_, _, _, _ = enacted_policy_group.publish_treasure_map() enacted_policy.publish_treasure_map()