From cf5d691950c99c8c055048e1b5aa4150b136a765 Mon Sep 17 00:00:00 2001 From: Anton Date: Fri, 25 May 2018 00:46:08 +0300 Subject: [PATCH] Clean the tests --- freqtrade/tests/test_freqtradebot.py | 24 ++++++------------------ 1 file changed, 6 insertions(+), 18 deletions(-) diff --git a/freqtrade/tests/test_freqtradebot.py b/freqtrade/tests/test_freqtradebot.py index d9d009a47..e47bfd007 100644 --- a/freqtrade/tests/test_freqtradebot.py +++ b/freqtrade/tests/test_freqtradebot.py @@ -224,23 +224,18 @@ def test_get_trade_stake_amount(default_conf, ticker, limit_buy_order, fee, mock """ Test get_trade_stake_amount() method """ - patch_get_signal(mocker) + patch_RPCManager(mocker) - patch_coinmarketcap(mocker) mocker.patch.multiple( 'freqtrade.freqtradebot.exchange', validate_pairs=MagicMock(), - get_ticker=ticker, - buy=MagicMock(return_value={'id': limit_buy_order['id']}), - get_fee=fee, + get_balance=MagicMock(return_value=default_conf['stake_amount'] * 2) ) - conf = deepcopy(default_conf) - - freqtrade = FreqtradeBot(conf, create_engine('sqlite://')) + freqtrade = FreqtradeBot(default_conf, create_engine('sqlite://')) result = freqtrade._get_trade_stake_amount() - assert(result == conf['stake_amount']) + assert(result == default_conf['stake_amount']) def test_get_trade_stake_amount_no_stake_amount(default_conf, @@ -251,21 +246,14 @@ def test_get_trade_stake_amount_no_stake_amount(default_conf, """ Test get_trade_stake_amount() method """ - patch_get_signal(mocker) patch_RPCManager(mocker) - patch_coinmarketcap(mocker) mocker.patch.multiple( 'freqtrade.freqtradebot.exchange', validate_pairs=MagicMock(), - get_ticker=ticker, - buy=MagicMock(return_value={'id': limit_buy_order['id']}), - get_balance=MagicMock(return_value=default_conf['stake_amount'] * 0.5), - get_fee=fee, + get_balance=MagicMock(return_value=default_conf['stake_amount'] * 0.5) ) - conf = deepcopy(default_conf) - - freqtrade = FreqtradeBot(conf, create_engine('sqlite://')) + freqtrade = FreqtradeBot(default_conf, create_engine('sqlite://')) with pytest.raises(DependencyException, match=r'.*stake amount.*'): freqtrade._get_trade_stake_amount()