Add two more unit tests for covering pair that are in a blacklist, and unknown pairs in the conf

This commit is contained in:
Jean-Baptiste LE STANG 2018-01-02 13:42:10 +01:00
parent a3e827c144
commit 5344b711ea

View File

@ -16,6 +16,9 @@ def whitelist_conf():
"BTC_SWT",
"BTC_BCC"
],
"pair_blacklist": [
"BTC_BLK"
],
},
}
@ -28,15 +31,27 @@ def get_health():
{'Currency': 'TKN',
'IsActive': True,
'BaseVolume': 1664
}]
},
{'Currency': 'BLK',
'IsActive': True,
'BaseVolume': 4096
}
]
def get_health_empty():
return []
# below three test could be merged into a single
# test that ran randomlly generated health lists
def test_refresh_market_pair_not_in_whitelist(mocker):
conf = whitelist_conf()
mocker.patch.dict('freqtrade.main._CONF', conf)
mocker.patch.multiple('freqtrade.main.exchange',
get_wallet_health=get_health)
refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist']+ ['BTC_XXX'])
# List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN']
# Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist
def test_refresh_whitelist(mocker):
conf = whitelist_conf()
@ -44,6 +59,7 @@ def test_refresh_whitelist(mocker):
mocker.patch.multiple('freqtrade.main.exchange',
get_wallet_health=get_health)
refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist'])
# List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN']
# Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist