def test_binance_query_trade_history_custom_markets(function_scope_binance,
user_data_dir):
"""Test that custom pairs are queried correctly"""
msg_aggregator = MessagesAggregator()
db = DBHandler(user_data_dir, '123', msg_aggregator, None)
binance_api_key = ApiKey('binance_api_key')
binance_api_secret = ApiSecret(b'binance_api_secret')
db.add_exchange('binance', Location.BINANCE, binance_api_key,
binance_api_secret)
binance = function_scope_binance
markets = ['ETHBTC', 'BNBBTC', 'BTCUSDC']
db.set_binance_pairs('binance', markets, Location.BINANCE)
count = 0
p = re.compile(r'symbol=[A-Z]*')
seen = set()
def mock_my_trades(url, timeout): # pylint: disable=unused-argument
nonlocal count
count += 1
market = p.search(url).group()[7:]
assert market in markets and market not in seen
seen.add(market)
text = '[]'
return MockResponse(200, text)
with patch.object(binance.session, 'get', side_effect=mock_my_trades):
binance.query_trade_history(start_ts=0,
end_ts=1564301134,
only_cache=False)
assert count == len(markets)
def test_binance_pairs(user_data_dir):
msg_aggregator = MessagesAggregator()
db = DBHandler(user_data_dir, '123', msg_aggregator, None)
binance_api_key = ApiKey('binance_api_key')
binance_api_secret = ApiSecret(b'binance_api_secret')
db.add_exchange('binance', Location.BINANCE, binance_api_key,
binance_api_secret)
db.set_binance_pairs('binance', ['ETHUSDC', 'ETHBTC', 'BNBBTC'],
Location.BINANCE)
query = db.get_binance_pairs('binance', Location.BINANCE)
assert query == ['ETHUSDC', 'ETHBTC', 'BNBBTC']
db.set_binance_pairs('binance', [], Location.BINANCE)
query = db.get_binance_pairs('binance', Location.BINANCE)
assert query == []
def test_add_remove_exchange(user_data_dir):
"""Tests that adding and removing an exchange in the DB works.
Also unknown exchanges should fail
"""
msg_aggregator = MessagesAggregator()
db = DBHandler(user_data_dir, '123', msg_aggregator, None)
# Test that an unknown exchange fails
with pytest.raises(InputError):
db.add_exchange('non_existing_exchange', 'api_key', 'api_secret')
credentials = db.get_exchange_credentials()
assert len(credentials) == 0
kraken_api_key = ApiKey('kraken_api_key')
kraken_api_secret = ApiSecret(b'kraken_api_secret')
binance_api_key = ApiKey('binance_api_key')
binance_api_secret = ApiSecret(b'binance_api_secret')
# add mock kraken and binance
db.add_exchange('kraken', kraken_api_key, kraken_api_secret)
db.add_exchange('binance', binance_api_key, binance_api_secret)
# and check the credentials can be retrieved
credentials = db.get_exchange_credentials()
assert len(credentials) == 2
assert credentials['kraken'].api_key == kraken_api_key
assert credentials['kraken'].api_secret == kraken_api_secret
assert credentials['binance'].api_key == binance_api_key
assert credentials['binance'].api_secret == binance_api_secret
# remove an exchange and see it works
db.remove_exchange('kraken')
credentials = db.get_exchange_credentials()
assert len(credentials) == 1
def test_add_remove_exchange(data_dir, username):
"""Tests that adding and removing an exchange in the DB works.
Also unknown exchanges should fail
"""
msg_aggregator = MessagesAggregator()
username = '******'
userdata_dir = os.path.join(data_dir, username)
os.mkdir(userdata_dir)
db = DBHandler(userdata_dir, '123', msg_aggregator)
# Test that an unknown exchange fails
with pytest.raises(InputError):
db.add_exchange('non_existing_exchange', 'api_key', 'api_secret')
credentials = db.get_exchange_credentials()
assert len(credentials) == 0
# add mock kraken and binance
db.add_exchange('kraken', 'kraken_api_key', 'kraken_api_secret')
db.add_exchange('binance', 'binance_api_key', 'binance_api_secret')
# and check the credentials can be retrieved
credentials = db.get_exchange_credentials()
assert len(credentials) == 2
assert credentials['kraken'].api_key == b'kraken_api_key'
assert credentials['kraken'].api_secret == b'kraken_api_secret'
assert credentials['binance'].api_key == b'binance_api_key'
assert credentials['binance'].api_secret == b'binance_api_secret'
# remove an exchange and see it works
db.remove_exchange('kraken')
credentials = db.get_exchange_credentials()
assert len(credentials) == 1
def test_add_remove_exchange(user_data_dir):
"""Tests that adding and removing an exchange in the DB works.
Also unknown exchanges should fail
"""
msg_aggregator = MessagesAggregator()
db = DBHandler(user_data_dir, '123', msg_aggregator, None)
# Test that an unknown exchange fails
with pytest.raises(InputError):
db.add_exchange('foo', Location.EXTERNAL, 'api_key', 'api_secret')
credentials = db.get_exchange_credentials()
assert len(credentials) == 0
kraken_api_key1 = ApiKey('kraken_api_key')
kraken_api_secret1 = ApiSecret(b'kraken_api_secret')
kraken_api_key2 = ApiKey('kraken_api_key2')
kraken_api_secret2 = ApiSecret(b'kraken_api_secret2')
binance_api_key = ApiKey('binance_api_key')
binance_api_secret = ApiSecret(b'binance_api_secret')
# add mock kraken and binance
db.add_exchange('kraken1', Location.KRAKEN, kraken_api_key1,
kraken_api_secret1)
db.add_exchange('kraken2', Location.KRAKEN, kraken_api_key2,
kraken_api_secret2)
db.add_exchange('binance', Location.BINANCE, binance_api_key,
binance_api_secret)
# and check the credentials can be retrieved
credentials = db.get_exchange_credentials()
assert len(credentials) == 2
assert len(credentials[Location.KRAKEN]) == 2
kraken1 = credentials[Location.KRAKEN][0]
assert kraken1.name == 'kraken1'
assert kraken1.api_key == kraken_api_key1
assert kraken1.api_secret == kraken_api_secret1
kraken2 = credentials[Location.KRAKEN][1]
assert kraken2.name == 'kraken2'
assert kraken2.api_key == kraken_api_key2
assert kraken2.api_secret == kraken_api_secret2
assert len(credentials[Location.BINANCE]) == 1
binance = credentials[Location.BINANCE][0]
assert binance.name == 'binance'
assert binance.api_key == binance_api_key
assert binance.api_secret == binance_api_secret
# remove an exchange and see it works
db.remove_exchange('kraken1', Location.KRAKEN)
credentials = db.get_exchange_credentials()
assert len(credentials) == 2
assert len(credentials[Location.KRAKEN]) == 1
kraken2 = credentials[Location.KRAKEN][0]
assert kraken2.name == 'kraken2'
assert kraken2.api_key == kraken_api_key2
assert kraken2.api_secret == kraken_api_secret2
assert len(credentials[Location.BINANCE]) == 1
binance = credentials[Location.BINANCE][0]
assert binance.name == 'binance'
assert binance.api_key == binance_api_key
assert binance.api_secret == binance_api_secret
# remove last exchange of a location and see nothing is returned
db.remove_exchange('kraken2', Location.KRAKEN)
credentials = db.get_exchange_credentials()
assert len(credentials) == 1
assert len(credentials[Location.BINANCE]) == 1
binance = credentials[Location.BINANCE][0]
assert binance.name == 'binance'
assert binance.api_key == binance_api_key
assert binance.api_secret == binance_api_secret