stable/freqtrade/tests/test_main.py

238 lines
9.9 KiB
Python
Raw Normal View History

# pragma pylint: disable=missing-docstring
import copy
2017-11-07 21:27:44 +00:00
from unittest.mock import MagicMock
import pytest
import requests
2017-11-09 22:45:22 +00:00
from sqlalchemy import create_engine
2017-10-06 10:22:04 +00:00
from freqtrade.exchange import Exchanges
2017-09-28 21:26:28 +00:00
from freqtrade.main import create_trade, handle_trade, close_trade_if_fulfilled, init, \
2017-11-07 21:27:44 +00:00
get_target_bid, _process
from freqtrade.misc import get_state, State
2017-09-28 21:26:28 +00:00
from freqtrade.persistence import Trade
def test_process_trade_creation(default_conf, ticker, health, mocker):
2017-11-07 21:27:44 +00:00
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
get_wallet_health=health,
2017-11-07 21:27:44 +00:00
buy=MagicMock(return_value='mocked_limit_buy'))
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
trades = Trade.query.filter(Trade.is_open.is_(True)).all()
assert len(trades) == 0
result = _process()
assert result is True
trades = Trade.query.filter(Trade.is_open.is_(True)).all()
assert len(trades) == 1
trade = trades[0]
assert trade is not None
assert trade.stake_amount == default_conf['stake_amount']
assert trade.is_open
assert trade.open_date is not None
assert trade.exchange == Exchanges.BITTREX.name
assert trade.open_rate == 0.072661
assert trade.amount == 0.6864067381401302
def test_process_exchange_failures(default_conf, ticker, health, mocker):
2017-11-07 21:27:44 +00:00
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
sleep_mock = mocker.patch('time.sleep', side_effect=lambda _: None)
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
get_wallet_health=health,
2017-11-07 21:27:44 +00:00
buy=MagicMock(side_effect=requests.exceptions.RequestException))
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
result = _process()
assert result is False
assert sleep_mock.has_calls()
def test_process_runtime_error(default_conf, ticker, health, mocker):
2017-11-07 21:27:44 +00:00
msg_mock = MagicMock()
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=msg_mock)
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
get_wallet_health=health,
2017-11-07 21:27:44 +00:00
buy=MagicMock(side_effect=RuntimeError))
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
assert get_state() == State.RUNNING
result = _process()
assert result is False
assert get_state() == State.STOPPED
assert 'RuntimeError' in msg_mock.call_args_list[-1][0][0]
def test_process_trade_handling(default_conf, ticker, limit_buy_order, health, mocker):
2017-11-07 21:27:44 +00:00
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
get_wallet_health=health,
2017-11-07 21:27:44 +00:00
buy=MagicMock(return_value='mocked_limit_buy'),
2017-11-08 21:44:32 +00:00
get_order=MagicMock(return_value=limit_buy_order))
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
2017-11-08 21:44:32 +00:00
trades = Trade.query.filter(Trade.is_open.is_(True)).all()
assert len(trades) == 0
2017-11-07 21:27:44 +00:00
result = _process()
assert result is True
2017-11-08 21:44:32 +00:00
trades = Trade.query.filter(Trade.is_open.is_(True)).all()
assert len(trades) == 1
2017-11-07 21:27:44 +00:00
result = _process()
assert result is False
2017-11-07 19:12:56 +00:00
def test_create_trade(default_conf, ticker, limit_buy_order, mocker):
mocker.patch.dict('freqtrade.main._CONF', default_conf)
2017-11-07 21:27:44 +00:00
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch.multiple('freqtrade.main.exchange',
2017-10-01 21:28:09 +00:00
validate_pairs=MagicMock(),
2017-11-07 19:12:56 +00:00
get_ticker=ticker,
2017-10-31 23:26:32 +00:00
buy=MagicMock(return_value='mocked_limit_buy'))
# Save state of current whitelist
2017-11-07 19:12:56 +00:00
whitelist = copy.deepcopy(default_conf['exchange']['pair_whitelist'])
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
trade = create_trade(15.0)
Trade.session.add(trade)
Trade.session.flush()
assert trade is not None
assert trade.stake_amount == 15.0
assert trade.is_open
assert trade.open_date is not None
assert trade.exchange == Exchanges.BITTREX.name
# Simulate fulfilled LIMIT_BUY order for trade
trade.update(limit_buy_order)
2017-10-31 23:26:32 +00:00
2017-11-07 21:27:44 +00:00
assert trade.open_rate == 0.07256061
assert trade.amount == 206.43811673387373
assert whitelist == default_conf['exchange']['pair_whitelist']
def test_create_trade_no_stake_amount(default_conf, ticker, mocker):
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
buy=MagicMock(return_value='mocked_limit_buy'),
get_balance=MagicMock(return_value=default_conf['stake_amount'] * 0.5))
with pytest.raises(ValueError, match=r'.*stake amount.*'):
create_trade(default_conf['stake_amount'])
2017-11-07 19:12:56 +00:00
2017-10-31 23:26:32 +00:00
2017-11-07 21:27:44 +00:00
def test_create_trade_no_pairs(default_conf, ticker, mocker):
mocker.patch.dict('freqtrade.main._CONF', default_conf)
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
buy=MagicMock(return_value='mocked_limit_buy'))
2017-11-07 21:27:44 +00:00
with pytest.raises(ValueError, match=r'.*No pair in whitelist.*'):
conf = copy.deepcopy(default_conf)
conf['exchange']['pair_whitelist'] = []
mocker.patch.dict('freqtrade.main._CONF', conf)
create_trade(default_conf['stake_amount'])
2017-10-30 23:36:35 +00:00
2017-11-07 21:27:44 +00:00
def test_handle_trade(default_conf, limit_buy_order, limit_sell_order, mocker):
2017-11-07 19:12:56 +00:00
mocker.patch.dict('freqtrade.main._CONF', default_conf)
2017-11-07 21:27:44 +00:00
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch.multiple('freqtrade.main.exchange',
2017-10-01 21:28:09 +00:00
validate_pairs=MagicMock(),
get_ticker=MagicMock(return_value={
'bid': 0.17256061,
'ask': 0.172661,
'last': 0.17256061
}),
2017-11-07 21:27:44 +00:00
buy=MagicMock(return_value='mocked_limit_buy'),
2017-10-31 23:26:32 +00:00
sell=MagicMock(return_value='mocked_limit_sell'))
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
trade = create_trade(15.0)
trade.update(limit_buy_order)
Trade.session.add(trade)
Trade.session.flush()
trade = Trade.query.filter(Trade.is_open.is_(True)).first()
assert trade
2017-10-31 23:26:32 +00:00
handle_trade(trade)
2017-10-31 23:26:32 +00:00
assert trade.open_order_id == 'mocked_limit_sell'
2017-11-07 21:27:44 +00:00
assert close_trade_if_fulfilled(trade) is False
2017-10-31 23:26:32 +00:00
# Simulate fulfilled LIMIT_SELL order for trade
2017-11-07 19:12:56 +00:00
trade.update(limit_sell_order)
assert trade.close_rate == 0.0802134
assert trade.close_profit == 0.10046755
assert trade.close_date is not None
2017-10-30 23:36:35 +00:00
2017-11-07 21:27:44 +00:00
def test_close_trade(default_conf, ticker, limit_buy_order, limit_sell_order, mocker):
2017-11-07 19:12:56 +00:00
mocker.patch.dict('freqtrade.main._CONF', default_conf)
2017-11-07 21:27:44 +00:00
mocker.patch('freqtrade.main.get_buy_signal', side_effect=lambda _: True)
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
mocker.patch.multiple('freqtrade.main.exchange',
validate_pairs=MagicMock(),
get_ticker=ticker,
buy=MagicMock(return_value='mocked_limit_buy'))
# Create trade and sell it
2017-11-09 22:45:22 +00:00
init(default_conf, create_engine('sqlite://'))
2017-11-07 21:27:44 +00:00
trade = create_trade(15.0)
trade.update(limit_buy_order)
trade.update(limit_sell_order)
Trade.session.add(trade)
Trade.session.flush()
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 not trade.is_open
2017-11-07 21:27:44 +00:00
with pytest.raises(ValueError, match=r'.*closed trade.*'):
handle_trade(trade)
2017-10-30 23:36:35 +00:00
def test_balance_fully_ask_side(mocker):
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 0.0}})
assert get_target_bid({'ask': 20, 'last': 10}) == 20
2017-10-30 23:36:35 +00:00
def test_balance_fully_last_side(mocker):
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}})
assert get_target_bid({'ask': 20, 'last': 10}) == 10
2017-10-30 23:36:35 +00:00
2017-11-07 19:12:56 +00:00
def test_balance_bigger_last_ask(mocker):
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'ask_last_balance': 1.0}})
assert get_target_bid({'ask': 5, 'last': 10}) == 5