From ce6b869f84d8d0dc0ef5d0b5c28b4bec792c36d1 Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 31 Oct 2019 07:11:57 +0100 Subject: [PATCH] Cleanup test --- tests/test_integration.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/tests/test_integration.py b/tests/test_integration.py index 857e0a2e3..c3a3ada07 100644 --- a/tests/test_integration.py +++ b/tests/test_integration.py @@ -2,10 +2,9 @@ from unittest.mock import MagicMock, PropertyMock -from freqtrade.freqtradebot import FreqtradeBot from freqtrade.persistence import Trade from freqtrade.strategy.interface import SellCheckTuple, SellType -from tests.conftest import (patch_exchange, +from tests.conftest import (patch_exchange, get_patched_freqtradebot, patch_get_signal) @@ -22,7 +21,6 @@ def test_may_execute_sell_stoploss_on_exchange_multi(default_conf, """ default_conf['max_open_trades'] = 3 default_conf['exchange']['name'] = 'binance' - patch_exchange(mocker) stoploss_limit = { 'id': 123, @@ -62,14 +60,12 @@ def test_may_execute_sell_stoploss_on_exchange_multi(default_conf, 'freqtrade.exchange.Exchange', get_ticker=ticker, get_fee=fee, - markets=PropertyMock(return_value=markets), symbol_amount_prec=lambda s, x, y: y, symbol_price_prec=lambda s, x, y: y, get_order=stoploss_order_mock, cancel_order=cancel_order_mock, ) - wallets_mock = MagicMock() mocker.patch.multiple( 'freqtrade.freqtradebot.FreqtradeBot', create_stoploss_order=MagicMock(return_value=True), @@ -77,9 +73,9 @@ def test_may_execute_sell_stoploss_on_exchange_multi(default_conf, _notify_sell=MagicMock(), ) mocker.patch("freqtrade.strategy.interface.IStrategy.should_sell", should_sell_mock) - mocker.patch("freqtrade.wallets.Wallets.update", wallets_mock) + wallets_mock = mocker.patch("freqtrade.wallets.Wallets.update", MagicMock()) - freqtrade = FreqtradeBot(default_conf) + freqtrade = get_patched_freqtradebot(mocker, default_conf) freqtrade.strategy.order_types['stoploss_on_exchange'] = True # Switch ordertype to market to close trade immediately freqtrade.strategy.order_types['sell'] = 'market'