131 lines
4.3 KiB
Python
131 lines
4.3 KiB
Python
# pragma pylint: disable=missing-docstring
|
|
import copy
|
|
from unittest.mock import MagicMock, call
|
|
|
|
import pytest
|
|
from jsonschema import validate
|
|
|
|
from freqtrade.exchange import Exchanges
|
|
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": {
|
|
"bid_ask_balance": 1.0
|
|
},
|
|
"exchange": {
|
|
"name": "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, mocker):
|
|
mocker.patch.dict('freqtrade.main._CONF', conf)
|
|
buy_signal = 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=MagicMock(return_value={
|
|
'bid': 0.07256061,
|
|
'ask': 0.072661,
|
|
}),
|
|
buy=MagicMock(return_value='mocked_order_id'))
|
|
# Save state of current whitelist
|
|
whitelist = copy.deepcopy(conf['exchange']['pair_whitelist'])
|
|
|
|
init(conf, 'sqlite://')
|
|
for pair in ['BTC_ETH', 'BTC_TKN', 'BTC_TRST', 'BTC_SWT']:
|
|
trade = create_trade(15.0)
|
|
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 == Exchanges.BITTREX.name
|
|
assert trade.amount == 206.43811673387373
|
|
assert trade.stake_amount == 15.0
|
|
assert trade.is_open
|
|
assert trade.open_date is not None
|
|
assert whitelist == conf['exchange']['pair_whitelist']
|
|
|
|
buy_signal.assert_has_calls(
|
|
[call('BTC_ETH'), call('BTC_TKN'), call('BTC_TRST'), call('BTC_SWT')]
|
|
)
|
|
|
|
|
|
def test_handle_trade(conf, mocker):
|
|
mocker.patch.dict('freqtrade.main._CONF', conf)
|
|
mocker.patch.multiple('freqtrade.main.telegram', init=MagicMock(), send_msg=MagicMock())
|
|
mocker.patch.multiple('freqtrade.main.exchange',
|
|
validate_pairs=MagicMock(),
|
|
get_ticker=MagicMock(return_value={
|
|
'bid': 0.17256061,
|
|
'ask': 0.172661,
|
|
}),
|
|
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, mocker):
|
|
mocker.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 not trade.is_open
|
|
|
|
|
|
def test_balance_fully_bid_side(mocker):
|
|
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'bid_ask_balance': 0.0}})
|
|
assert get_target_bid({'bid': 10, 'ask': 20}) == 10
|
|
|
|
|
|
def test_balance_fully_ask_side(mocker):
|
|
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'bid_ask_balance': 1.0}})
|
|
assert get_target_bid({'bid': 10, 'ask': 20}) == 20
|
|
|
|
|
|
def test_balance_half(mocker):
|
|
mocker.patch.dict('freqtrade.main._CONF', {'bid_strategy': {'bid_ask_balance': 0.5}})
|
|
assert get_target_bid({'bid': 10, 'ask': 20}) == 15
|