def test_exchange_query_balances_async(rotkehlchen_api_server_with_exchanges):
"""Test that using the exchange balances query endpoint works fine for async calls"""
# async query balances of one specific exchange
server = rotkehlchen_api_server_with_exchanges
binance = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'binance']
binance_patch = patch_binance_balances_query(binance)
with binance_patch:
response = requests.get(api_url_for(
server,
"named_exchanges_balances_resource",
name='binance',
),
json={'async_query': True})
task_id = assert_ok_async_response(response)
outcome = wait_for_async_task(server, task_id)
assert_binance_balances_result(outcome['result'])
# async query of one exchange with querystring parameters
poloniex = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'poloniex']
poloniex_patch = patch_poloniex_balances_query(poloniex)
with poloniex_patch:
response = requests.get(
api_url_for(
server,
"named_exchanges_balances_resource",
name='poloniex',
) + '?async_query=true')
task_id = assert_ok_async_response(response)
outcome = wait_for_async_task(server, task_id)
assert_poloniex_balances_result(outcome['result'])
# async query balances of all setup exchanges
with binance_patch, poloniex_patch:
response = requests.get(
api_url_for(server, "exchangebalancesresource"),
json={'async_query': True},
)
task_id = assert_ok_async_response(response)
outcome = wait_for_async_task(server, task_id)
result = outcome['result']
assert_binance_balances_result(result['binance'])
assert_poloniex_balances_result(result['poloniex'])
def test_exchange_query_balances(rotkehlchen_api_server_with_exchanges):
"""Test that using the exchange balances query endpoint works fine"""
async_query = random.choice([False, True])
# query balances of one specific exchange
server = rotkehlchen_api_server_with_exchanges
binance = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'binance']
binance_patch = patch_binance_balances_query(binance)
with binance_patch:
response = requests.get(api_url_for(
server,
"named_exchanges_balances_resource",
name='binance',
),
json={'async_query': async_query})
if async_query:
task_id = assert_ok_async_response(response)
outcome = wait_for_async_task_with_result(server, task_id)
else:
outcome = assert_proper_response_with_result(response)
assert_binance_balances_result(outcome)
# query balances of all setup exchanges
poloniex = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'poloniex']
poloniex_patch = patch_poloniex_balances_query(poloniex)
with binance_patch, poloniex_patch:
response = requests.get(
api_url_for(server, "exchangebalancesresource"),
json={'async_query': async_query},
)
if async_query:
task_id = assert_ok_async_response(response)
result = wait_for_async_task_with_result(server, task_id)
else:
result = assert_proper_response_with_result(response)
assert_binance_balances_result(result['binance'])
assert_poloniex_balances_result(result['poloniex'])
def test_exchange_query_balances(rotkehlchen_api_server_with_exchanges):
"""Test that using the exchange balances query endpoint works fine"""
# query balances of one specific exchange
server = rotkehlchen_api_server_with_exchanges
binance = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'binance']
binance_patch = patch_binance_balances_query(binance)
with binance_patch:
response = requests.get(
api_url_for(
server,
"named_exchanges_balances_resource",
name='binance',
))
assert_proper_response(response)
json_data = response.json()
assert json_data['message'] == ''
result = json_data['result']
assert_binance_balances_result(json_data['result'])
# query balances of all setup exchanges
poloniex = server.rest_api.rotkehlchen.exchange_manager.connected_exchanges[
'poloniex']
poloniex_patch = patch_poloniex_balances_query(poloniex)
with binance_patch, poloniex_patch:
response = requests.get(api_url_for(server,
"exchangebalancesresource"))
assert_proper_response(response)
json_data = response.json()
assert json_data['message'] == ''
result = json_data['result']
assert_binance_balances_result(result['binance'])
assert_poloniex_balances_result(result['poloniex'])
