Fixed balance tests

api-breakage
Morty Space 5 years ago
parent 2d7e7ed9e9
commit 7bfd14e1cc
  1. 1
      .gitignore
  2. 12
      tests/test_private.py

1
.gitignore vendored

@ -1,5 +1,6 @@
__pycache__/ __pycache__/
docs/build/ docs/build/
docs/source/_build/
build/ build/
dist/ dist/
venv/ venv/

@ -8,14 +8,16 @@ import cryptocom.exchange as cro
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_account_get_balance(account: cro.Account): async def test_account_get_balance(account: cro.Account):
balances = await account.get_balance() balances = await account.get_balance()
if balances[cro.coins.CRO].available < 1: while balances[cro.coins.CRO].available < 5:
await account.buy_market(cro.pairs.CRO_USDT, 1) await account.buy_market(cro.pairs.CRO_USDT, 1)
if balances[cro.coins.USDT].available < 1: balances = await account.get_balance()
await account.sell_market(cro.pairs.CRO_USDT, 2) while balances[cro.coins.USDT].available < 1:
await account.sell_market(cro.pairs.CRO_USDT, 1)
balances = await account.get_balance()
balances = await account.get_balance() balances = await account.get_balance()
local_coins = cro.coins.all() local_coins = cro.coins.all()
assert balances[cro.coins.CRO].available > 2 assert balances[cro.coins.CRO].available > 5
assert balances[cro.coins.USDT].available > 1 assert balances[cro.coins.USDT].available > 1
for coin in balances: for coin in balances:
assert coin in local_coins assert coin in local_coins
@ -56,7 +58,7 @@ async def test_account_limit_orders(
for i in range(25) for i in range(25)
]) ])
order_ids += await asyncio.gather(*[ order_ids += await asyncio.gather(*[
account.sell_limit(cro.pairs.CRO_USDT, 0.01, round(buy_price * 2, 4)) account.sell_limit(cro.pairs.CRO_USDT, 0.001, round(buy_price * 2, 4))
for i in range(25) for i in range(25)
]) ])
all_orders = await account.get_orders_history( all_orders = await account.get_orders_history(

Loading…
Cancel
Save