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_SWT",
"BTC_BCC" "BTC_BCC"
], ],
"pair_blacklist": [
"BTC_BLK"
],
}, },
} }
@ -28,15 +31,27 @@ def get_health():
{'Currency': 'TKN', {'Currency': 'TKN',
'IsActive': True, 'IsActive': True,
'BaseVolume': 1664 'BaseVolume': 1664
}] },
{'Currency': 'BLK',
'IsActive': True,
'BaseVolume': 4096
}
]
def get_health_empty(): def get_health_empty():
return [] return []
# below three test could be merged into a single def test_refresh_market_pair_not_in_whitelist(mocker):
# test that ran randomlly generated health lists 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): def test_refresh_whitelist(mocker):
conf = whitelist_conf() conf = whitelist_conf()
@ -44,6 +59,7 @@ def test_refresh_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(conf['exchange']['pair_whitelist']) refreshedwhitelist = refresh_whitelist(conf['exchange']['pair_whitelist'])
# List ordered by BaseVolume
whitelist = ['BTC_ETH', 'BTC_TKN'] whitelist = ['BTC_ETH', 'BTC_TKN']
# Ensure all except those in whitelist are removed # Ensure all except those in whitelist are removed
assert whitelist == refreshedwhitelist assert whitelist == refreshedwhitelist