import copy from unittest.mock import patch, MagicMock, call import pytest from jsonschema import validate from freqtrade import exchange from freqtrade.main import create_trade, handle_trade, close_trade_if_fulfilled, init, \ get_target_bid from freqtrade.misc import CONF_SCHEMA from freqtrade.persistence import Trade @pytest.fixture def conf(): configuration = { "max_open_trades": 3, "stake_currency": "BTC", "stake_amount": 0.05, "dry_run": True, "minimal_roi": { "2880": 0.005, "720": 0.01, "0": 0.02 }, "bid_strategy": { "ask_last_balance": 0.0 }, "bittrex": { "enabled": True, "key": "key", "secret": "secret", "pair_whitelist": [ "BTC_ETH", "BTC_TKN", "BTC_TRST", "BTC_SWT", ] }, "telegram": { "enabled": True, "token": "token", "chat_id": "chat_id" } } validate(configuration, CONF_SCHEMA) return configuration def test_create_trade(conf): with patch.dict('freqtrade.main._CONF', conf): with patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True) as buy_signal: with patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock()): with patch.multiple('freqtrade.main.exchange', get_ticker=MagicMock(return_value={ 'bid': 0.07256061, 'ask': 0.072661, 'last': 0.07256061 }), buy=MagicMock(return_value='mocked_order_id')): # Save state of current whitelist whitelist = copy.deepcopy(conf['bittrex']['pair_whitelist']) init(conf, 'sqlite://') for pair in ['BTC_ETH', 'BTC_TKN', 'BTC_TRST', 'BTC_SWT']: trade = create_trade(15.0, exchange.Exchange.BITTREX) Trade.session.add(trade) Trade.session.flush() assert trade is not None assert trade.open_rate == 0.072661 assert trade.pair == pair assert trade.exchange == exchange.Exchange.BITTREX assert trade.amount == 206.43811673387373 assert trade.stake_amount == 15.0 assert trade.is_open == True assert trade.open_date is not None assert whitelist == conf['bittrex']['pair_whitelist'] buy_signal.assert_has_calls( [call('BTC_ETH'), call('BTC_TKN'), call('BTC_TRST'), call('BTC_SWT')] ) def test_handle_trade(conf): with patch.dict('freqtrade.main._CONF', conf): with patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock()): with patch.multiple('freqtrade.main.exchange', get_ticker=MagicMock(return_value={ 'bid': 0.17256061, 'ask': 0.172661, 'last': 0.17256061 }), buy=MagicMock(return_value='mocked_order_id')): trade = Trade.query.filter(Trade.is_open.is_(True)).first() assert trade handle_trade(trade) assert trade.close_rate == 0.17256061 assert trade.close_profit == 137.4872490056564 assert trade.close_date is not None assert trade.open_order_id == 'dry_run' def test_close_trade(conf): with patch.dict('freqtrade.main._CONF', conf): trade = Trade.query.filter(Trade.is_open.is_(True)).first() assert trade # Simulate that there is no open order trade.open_order_id = None closed = close_trade_if_fulfilled(trade) assert closed assert trade.is_open == False def test_balance_fully_ask_side(): with patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 0.0}}): assert get_target_bid({'ask': 20, 'last': 10}) == 20 def test_balance_fully_last_side(): with patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}}): assert get_target_bid({'ask': 20, 'last': 10}) == 10 def test_balance_when_last_bigger_than_ask(): with patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}}): assert get_target_bid({'ask': 5, 'last': 10}) == 5