def test_bucket_rate():
timer = Timer()
alice = Account()
token = ERC20Token()
bkt = Bucket(token=token)
token.mint(bkt, 1000000)
assert token.balance_of(bkt) == 1000000
alices_tap = Tap(withdrawer=alice, upstream=bkt, rate=100, timer=timer)
assert alices_tap.available == 0
timer.increment(1000)
alices_tap.update()
assert alices_tap.available == 100000
assert token.balance_of(alice) == 0
assert token.balance_of(bkt) == 1000000
res = alices_tap.withdraw(100000)
assert res is True
assert token.balance_of(alice) == 100000
assert token.balance_of(bkt) == 1000000 - 100000
res = alices_tap.withdraw(900000)
assert res is False
assert token.balance_of(alice) == 100000
assert token.balance_of(bkt) == 1000000 - 100000
timer.increment(2000)
res = alices_tap.withdraw(200000)
assert res is True
assert token.balance_of(alice) == 300000
assert token.balance_of(bkt) == 1000000 - 300000
res = alices_tap.withdraw(1)
assert res is False
assert token.balance_of(alice) == 300000
assert token.balance_of(bkt) == 1000000 - 300000
def test_mint_to_single_account():
alice = Account()
token = ERC20Token()
assert token.total_supply == 0
token.mint(alice, 100)
assert token.balance_of(alice) == 100
assert token.total_supply == 100
def test_mint_and_transfer_all():
alice = Account()
bob = Account()
token = ERC20Token()
token.mint(alice, 100)
token.transfer(alice, bob, 100)
assert token.balance_of(alice) == 0
assert token.balance_of(bob) == 100
assert token.total_supply == 100
def test_chain_built_by_set_overflow_bucket():
token = ERC20Token()
bkt_two = Bucket(token=token, max_volume=sys.maxsize)
bkt_one = Bucket(token=token, max_volume=1289)
bkt_one.set_overflow_bucket(bkt_two)
token.mint(bkt_one, 1000000000)
assert token.balance_of(bkt_one) == 1000000000
bkt_one.flush()
assert token.balance_of(bkt_one) == 1289
assert token.balance_of(bkt_two) == 1000000000 - 1289
def test_generate_random_payments_and_process():
rand_p = generate_random_payments()
token = ERC20Token()
bkt_two = Bucket(token=token, name="02")
bkt_one = Bucket(token=token, max_volume=1000, overflow_bkt=bkt_two, name="01")
res = process_ingress_payments(rand_p, token, bkt_one, bkt_two)
assert isinstance(res, (tuple))
res = convert_to_df(res)
assert res['income'].sum() == 1000000
assert res['01'].iloc[-1] == 1000
assert res['02'].iloc[-1] == 1000000 - 1000
def test_bucket_overflow():
token = ERC20Token()
of_bkt = Bucket(token=token)
bkt = Bucket(token=token, overflow_bkt=of_bkt, max_volume=1000)
token.mint(bkt, 1000000)
bkt.flush()
assert token.balance_of(bkt) == 1000
assert token.balance_of(of_bkt) == 1000000 - 1000
assert token.total_supply == 1000000
token.mint(bkt, 1000)
bkt.flush()
assert token.balance_of(of_bkt) == 1000000
def test_tap_states():
token = ERC20Token()
bkt = Bucket(token=token)
alice = Account()
timer = Timer()
alices_tap = Tap(withdrawer=alice, upstream=bkt, rate=100, timer=timer)
alices_tap.close()
assert alices_tap.rate == 0
alices_tap.open()
assert alices_tap.rate == sys.maxsize
alices_tap.set_rate(100)
assert alices_tap.rate == 100
def test_flush_chain():
token = ERC20Token()
bkt_four = Bucket(token=token, max_volume=sys.maxsize)
bkt_three = Bucket(token=token, max_volume=1289, overflow_bkt=bkt_four)
bkt_two = Bucket(token=token, max_volume=1312, overflow_bkt=bkt_three)
bkt_one = Bucket(token=token, max_volume=1100, overflow_bkt=bkt_two)
token.mint(bkt_one, 1000000000)
assert token.balance_of(bkt_one) == 1000000000
bkt_one.flush()
assert token.balance_of(bkt_one) == 1100
assert token.balance_of(bkt_two) == 1312
assert token.balance_of(bkt_three) == 1289
assert token.balance_of(bkt_four) == 1000000000 - 1100 - 1312 - 1289
def test_mint_and_transfer_partial():
alice = Account()
bob = Account()
token = ERC20Token()
token.mint(alice, 100)
token.transfer(alice, bob, 99)
assert token.balance_of(alice) == 1
assert token.balance_of(bob) == 99
token.transfer(alice, bob, 1)
assert token.balance_of(alice) == 0
assert token.balance_of(bob) == 100
token.transfer(bob, alice, 2)
assert token.balance_of(alice) == 2
assert token.balance_of(bob) == 98
assert token.total_supply == 100
def test_withdraw_all():
token = ERC20Token()
bkt = Bucket(token=token)
token.mint(bkt, 1000000000)
alice = Account()
timer = Timer()
alices_tap = Tap(withdrawer=alice, upstream=bkt, rate=100, timer=timer)
timer.increment(seconds=10)
assert token.balance_of(alice) == 0
assert alices_tap.get_available() == 1000
res = alices_tap.withdraw_all()
assert res is True
assert token.balance_of(alice) == 1000
assert alices_tap.available == 0
assert alices_tap.get_available() == 0
assert token.balance_of(alices_tap) == 0
res = alices_tap.withdraw_all()
assert res is False
def test_bucket_and_tap():
timer = Timer()
alice = Account()
bob = Account()
token = ERC20Token()
bkt = Bucket(token=token)
token.mint(bkt, 1000000)
assert token.balance_of(bkt) == 1000000
alices_tap = Tap(withdrawer=alice, upstream=bkt, timer=timer)
bobs_tap = Tap(withdrawer=bob, upstream=bkt, timer=timer, rate=0)
timer.increment(1)
assert token.balance_of(alice) == 0
res = alices_tap.withdraw(1)
assert res is True
assert token.balance_of(alice) == 1
assert token.balance_of(bob) == 0
res = bobs_tap.withdraw(1)
assert res is False
assert token.balance_of(bob) == 0
bobs_tap.set_rate(100)
timer.increment(1)
res = bobs_tap.withdraw(1)
assert res is True
assert token.balance_of(bob) == 1