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

Touchups to test_staker

pull/1029/head
David Núñez 2019-06-24 13:38:49 +02:00
parent 9d0089102e
commit ad2d63fd5e
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/blockchain/eth/entities/actors/test_staker.py
View File

@ -57,10 +57,10 @@ def test_staker_locking_tokens(testerchain, agency, staker, token_economics):
assert 0 == allowance assert 0 == allowance
# Staking starts after one period # Staking starts after one period
locked_tokens = staking_agent.contract.functions.getLockedTokens(staker.checksum_address).call() locked_tokens = staker.locked_tokens()
assert 0 == locked_tokens assert 0 == locked_tokens
locked_tokens = staking_agent.contract.functions.getLockedTokens(staker.checksum_address, 1).call() locked_tokens = staker.locked_tokens(periods=1)
assert token_economics.minimum_allowed_locked == locked_tokens assert token_economics.minimum_allowed_locked == locked_tokens
@ -106,7 +106,7 @@ def test_staker_collects_staking_reward(testerchain, staker, blockchain_ursulas,
initial_balance = staker.token_balance initial_balance = staker.token_balance
assert token_agent.get_balance(staker.checksum_address) == initial_balance assert token_agent.get_balance(staker.checksum_address) == initial_balance
# Mock Powerup consumption (Ursula-Worker) # Mock Powerup consumption (Staker)
testerchain.transacting_power = BlockchainPower(blockchain=testerchain, account=staker.checksum_address) testerchain.transacting_power = BlockchainPower(blockchain=testerchain, account=staker.checksum_address)
staker.initialize_stake(amount=NU(token_economics.minimum_allowed_locked, 'NuNit'), # Lock the minimum amount of tokens staker.initialize_stake(amount=NU(token_economics.minimum_allowed_locked, 'NuNit'), # Lock the minimum amount of tokens
@ -123,8 +123,6 @@ def test_staker_collects_staking_reward(testerchain, staker, blockchain_ursulas,
confirm_activity=False, confirm_activity=False,
blockchain=testerchain).pop() blockchain=testerchain).pop()
# TODO: Use the above code as a starting point for a non-staking worker fixture
# ...wait out the lock period... # ...wait out the lock period...
for _ in range(token_economics.minimum_locked_periods): for _ in range(token_economics.minimum_locked_periods):
testerchain.time_travel(periods=1) testerchain.time_travel(periods=1)
@ -133,7 +131,7 @@ def test_staker_collects_staking_reward(testerchain, staker, blockchain_ursulas,
# ...wait more... # ...wait more...
testerchain.time_travel(periods=2) testerchain.time_travel(periods=2)
# Mock Powerup consumption (Ursula-Worker) # Mock Powerup consumption (Staker)
testerchain.transacting_power = BlockchainPower(blockchain=testerchain, account=staker.checksum_address) testerchain.transacting_power = BlockchainPower(blockchain=testerchain, account=staker.checksum_address)
# Profit! # Profit!