Fix tests to support ccxt

This commit is contained in:
Matthias Voppichler 2018-03-04 20:08:05 +01:00
parent 2b5d61c9e4
commit 858a52dc07

View File

@ -12,22 +12,23 @@ def whitelist_conf():
'stake_currency': 'BTC', 'stake_currency': 'BTC',
'exchange': { 'exchange': {
'pair_whitelist': [ 'pair_whitelist': [
'BTC_ETH', 'ETH/BTC',
'BTC_TKN', 'TKN/BTC',
'BTC_TRST', 'TRST/BTC',
'BTC_SWT', 'SWT/BTC',
'BTC_BCC' 'BCC/BTC'
], ],
'pair_blacklist': [ 'pair_blacklist': [
'BTC_BLK' 'BLK/BTC'
], ],
}, },
} }
def get_market_summaries(): def get_market_summaries():
return [{ return {'TKN/BTC': {
'MarketName': 'BTC-TKN', 'symbol': 'TKN/BTC',
'MarketName': 'TKN/BTC',
'High': 0.00000919, 'High': 0.00000919,
'Low': 0.00000820, 'Low': 0.00000820,
'Volume': 74339.61396015, 'Volume': 74339.61396015,
@ -40,9 +41,11 @@ def get_market_summaries():
'OpenSellOrders': 15, 'OpenSellOrders': 15,
'PrevDay': 0.00000821, 'PrevDay': 0.00000821,
'Created': '2014-03-20T06:00:00', 'Created': '2014-03-20T06:00:00',
'DisplayMarketName': '' 'DisplayMarketName': '',
}, { 'info': {},
'MarketName': 'BTC-ETH', }, 'ETH/BTC': {
'symbol': 'ETH/BTC',
'MarketName': 'ETH/BTC',
'High': 0.00000072, 'High': 0.00000072,
'Low': 0.00000001, 'Low': 0.00000001,
'Volume': 166340678.42280999, 'Volume': 166340678.42280999,
@ -55,9 +58,11 @@ def get_market_summaries():
'OpenSellOrders': 18, 'OpenSellOrders': 18,
'PrevDay': 0.00000002, 'PrevDay': 0.00000002,
'Created': '2014-05-30T07:57:49.637', 'Created': '2014-05-30T07:57:49.637',
'DisplayMarketName': '' 'DisplayMarketName': '',
}, { 'info': {},
'MarketName': 'BTC-BLK', }, 'BLK/BTC': {
'symbol': 'BLK/BTC',
'MarketName': 'BLK/BTC',
'High': 0.00000072, 'High': 0.00000072,
'Low': 0.00000001, 'Low': 0.00000001,
'Volume': 166340678.42280999, 'Volume': 166340678.42280999,
@ -70,18 +75,21 @@ def get_market_summaries():
'OpenSellOrders': 18, 'OpenSellOrders': 18,
'PrevDay': 0.00000002, 'PrevDay': 0.00000002,
'Created': '2014-05-30T07:57:49.637', 'Created': '2014-05-30T07:57:49.637',
'DisplayMarketName': '' 'DisplayMarketName': '',
}] 'info': {},
}}
def get_health(): def get_health():
return [{'Currency': 'ETH', 'IsActive': True}, return {
{'Currency': 'TKN', 'IsActive': True}, 'ETH/BTC': {'Currency': 'ETH', 'base': 'ETH', 'quote': 'BTC', 'active': True},
{'Currency': 'BLK', 'IsActive': True}] 'TKN/BTC': {'Currency': 'TKN', 'base': 'TKN', 'quote': 'BTC', 'active': True},
'BLK/BTC': {'Currency': 'BLK', 'base': 'BLK', 'quote': 'BTC', 'active': True}
}
def get_health_empty(): def get_health_empty():
return [] return {}
def test_refresh_market_pair_not_in_whitelist(mocker): def test_refresh_market_pair_not_in_whitelist(mocker):
@ -90,9 +98,9 @@ def test_refresh_market_pair_not_in_whitelist(mocker):
mocker.patch.multiple('freqtrade.main.exchange', mocker.patch.multiple('freqtrade.main.exchange',
get_wallet_health=get_health) get_wallet_health=get_health)
refreshedwhitelist = refresh_whitelist( refreshedwhitelist = refresh_whitelist(
conf['exchange']['pair_whitelist'] + ['BTC_XXX']) conf['exchange']['pair_whitelist'] + ['XXX/BTC'])
# List ordered by BaseVolume # List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN'] whitelist = ['ETH/BTC', 'TKN/BTC']
# Ensure all except those in whitelist are removed # Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist assert whitelist == refreshedwhitelist
@ -104,7 +112,7 @@ def test_refresh_whitelist(mocker):
get_wallet_health=get_health) get_wallet_health=get_health)
refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist']) refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist'])
# List ordered by BaseVolume # List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN'] whitelist = ['ETH/BTC', 'TKN/BTC']
# Ensure all except those in whitelist are removed # Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist assert whitelist == refreshedwhitelist
@ -117,7 +125,7 @@ def test_refresh_whitelist_dynamic(mocker):
mocker.patch.multiple('freqtrade.main.exchange', mocker.patch.multiple('freqtrade.main.exchange',
get_market_summaries=get_market_summaries) get_market_summaries=get_market_summaries)
# argument: use the whitelist dynamically by exchange-volume # argument: use the whitelist dynamically by exchange-volume
whitelist = ['BTC_TKN', 'BTC_ETH'] whitelist = ['TKN/BTC', 'ETH/BTC']
refreshedwhitelist = refresh_whitelist( refreshedwhitelist = refresh_whitelist(
gen_pair_whitelist(conf['stake_currency'])) gen_pair_whitelist(conf['stake_currency']))
assert whitelist == refreshedwhitelist assert whitelist == refreshedwhitelist