mirror of https://github.com/nucypher/nucypher.git
Remove test for sampling in contract
David Núñez
parent
49932b4c14
commit
9d88930e3c
|
|
@ -1,143 +0,0 @@
|
|||
"""
|
||||
This file is part of nucypher.
|
||||
|
||||
nucypher is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU Affero General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
nucypher is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU Affero General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Affero General Public License
|
||||
along with nucypher. If not, see <https://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
|
||||
import pytest
|
||||
from eth_tester.exceptions import TransactionFailed
|
||||
|
||||
from nucypher.blockchain.eth.interfaces import BlockchainInterface
|
||||
|
||||
@pytest.mark.slow
|
||||
def test_sampling(testerchain, token, escrow_contract):
|
||||
escrow = escrow_contract(5 * 10 ** 8)
|
||||
creator = testerchain.etherbase_account
|
||||
|
||||
# Give Escrow tokens for reward and initialize contract
|
||||
tx = token.functions.transfer(escrow.address, 10 ** 9).transact({'from': creator})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
tx = escrow.functions.initialize().transact({'from': creator})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
|
||||
stakers = testerchain.stakers_accounts
|
||||
amount = token.functions.balanceOf(creator).call() // 2
|
||||
largest_locked = amount
|
||||
|
||||
# Airdrop
|
||||
for staker in stakers:
|
||||
tx = token.functions.transfer(staker, amount).transact({'from': creator})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
amount = amount // 2
|
||||
|
||||
# Can't use sample without points, with zero periods value, or not sorted in ascending order.
|
||||
with pytest.raises(TransactionFailed):
|
||||
escrow.functions.sample([], 1).call()
|
||||
with pytest.raises(TransactionFailed):
|
||||
escrow.functions.sample([1], 0).call()
|
||||
with pytest.raises(TransactionFailed):
|
||||
escrow.functions.sample([3, 2, 1], 0).call()
|
||||
|
||||
# No stakers yet
|
||||
addresses = escrow.functions.sample([1], 1).call()
|
||||
assert 1 == len(addresses)
|
||||
assert BlockchainInterface.NULL_ADDRESS == addresses[0]
|
||||
|
||||
all_locked_tokens = 0
|
||||
# All stakers lock tokens for different lock periods
|
||||
for index, staker in enumerate(stakers):
|
||||
balance = token.functions.balanceOf(staker).call()
|
||||
tx = token.functions.approve(escrow.address, balance).transact({'from': staker})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
tx = escrow.functions.deposit(balance, index + 2).transact({'from': staker})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
tx = escrow.functions.setWorker(staker).transact({'from': staker})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
tx = escrow.functions.confirmActivity().transact({'from': staker})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
all_locked_tokens += balance
|
||||
|
||||
# Stakers are active starting from the next period
|
||||
assert 0 == escrow.functions.getAllLockedTokens(1).call()
|
||||
|
||||
# So sampling in current period is useless
|
||||
addresses = escrow.functions.sample([1], 1).call()
|
||||
assert 1 == len(addresses)
|
||||
assert BlockchainInterface.NULL_ADDRESS == addresses[0]
|
||||
|
||||
# Wait next period and check all locked tokens
|
||||
testerchain.time_travel(hours=1)
|
||||
assert all_locked_tokens == escrow.functions.getAllLockedTokens(1).call()
|
||||
assert all_locked_tokens > escrow.functions.getAllLockedTokens(2).call()
|
||||
assert 0 < escrow.functions.getAllLockedTokens(len(stakers)).call()
|
||||
assert 0 == escrow.functions.getAllLockedTokens(len(stakers) + 1).call()
|
||||
|
||||
# All stakers confirm activity
|
||||
for staker in stakers:
|
||||
tx = escrow.functions.confirmActivity().transact({'from': staker})
|
||||
testerchain.wait_for_receipt(tx)
|
||||
|
||||
# Sample one staker by value less than first staker's stake
|
||||
addresses = escrow.functions.sample([largest_locked - 1], 1).call()
|
||||
assert 1 == len(addresses)
|
||||
assert stakers[0] == addresses[0]
|
||||
|
||||
# Sample two stakers by values that are equal to first and second stakes
|
||||
# In the result must be second and third stakers because of strict condition in the sampling
|
||||
addresses = escrow.functions.sample([largest_locked, largest_locked + largest_locked // 2], 1).call()
|
||||
assert 2 == len(addresses)
|
||||
assert stakers[1] == addresses[0]
|
||||
assert stakers[2] == addresses[1]
|
||||
|
||||
# Sample two stakers by values that within first and second stakes
|
||||
# In the result must be second and third stakers because of strict condition in the sampling
|
||||
# sumOfLockedTokens > point
|
||||
addresses = escrow.functions.sample([largest_locked - 1, largest_locked + 1], 1).call()
|
||||
assert 2 == len(addresses)
|
||||
assert stakers[0] == addresses[0]
|
||||
assert stakers[1] == addresses[1]
|
||||
|
||||
# Sample staker by the max duration of the longest stake
|
||||
# The result is the staker who has the longest stake
|
||||
addresses = escrow.functions.sample([1], len(stakers)).call()
|
||||
assert 1 == len(addresses)
|
||||
assert stakers[-1] == addresses[0]
|
||||
|
||||
# Sample staker using the duration more than the longest stake
|
||||
# The result is empty
|
||||
addresses = escrow.functions.sample([1], len(stakers) + 1).call()
|
||||
assert 1 == len(addresses)
|
||||
assert BlockchainInterface.NULL_ADDRESS == addresses[0]
|
||||
|
||||
# Sample stakers by different durations and minimum value
|
||||
# Each result is the first appropriate stake by length
|
||||
for index, _ in enumerate(stakers[:-1], start=1):
|
||||
addresses = escrow.functions.sample([1], index + 1).call()
|
||||
assert 1 == len(addresses)
|
||||
assert stakers[index] == addresses[0]
|
||||
|
||||
# Sample all stakers by values as stake minus one
|
||||
# The result must contain all stakers
|
||||
points = [escrow.functions.getLockedTokens(staker).call() for staker in stakers]
|
||||
points[0] = points[0] - 1
|
||||
for i in range(1, len(points)):
|
||||
points[i] += points[i-1]
|
||||
addresses = escrow.functions.sample(points, 1).call()
|
||||
assert stakers == addresses
|
||||
|
||||
# Test stakers iteration
|
||||
assert len(stakers) == escrow.functions.getStakersLength().call()
|
||||
for index, staker in enumerate(stakers):
|
||||
assert stakers[index] == escrow.functions.stakers(index).call()
|
||||
Loading…
Reference in New Issue