mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-14 12:13:57 +00:00
117 lines
4.9 KiB
Python
117 lines
4.9 KiB
Python
from random import randint
|
|
from unittest.mock import MagicMock
|
|
|
|
import ccxt
|
|
import pytest
|
|
|
|
from freqtrade.exceptions import DependencyException, InvalidOrderException
|
|
from tests.conftest import EXMS, get_patched_exchange
|
|
from tests.exchange.test_exchange import ccxt_exceptionhandlers
|
|
|
|
|
|
@pytest.mark.parametrize('limitratio,expected,side', [
|
|
(None, 220 * 0.99, "sell"),
|
|
(0.99, 220 * 0.99, "sell"),
|
|
(0.98, 220 * 0.98, "sell"),
|
|
])
|
|
def test_create_stoploss_order_huobi(default_conf, mocker, limitratio, expected, side):
|
|
api_mock = MagicMock()
|
|
order_id = 'test_prod_buy_{}'.format(randint(0, 10 ** 6))
|
|
order_type = 'stop-limit'
|
|
|
|
api_mock.create_order = MagicMock(return_value={
|
|
'id': order_id,
|
|
'info': {
|
|
'foo': 'bar'
|
|
}
|
|
})
|
|
default_conf['dry_run'] = False
|
|
mocker.patch(f'{EXMS}.amount_to_precision', lambda s, x, y: y)
|
|
mocker.patch(f'{EXMS}.price_to_precision', lambda s, x, y, **kwargs: y)
|
|
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, 'huobi')
|
|
|
|
with pytest.raises(InvalidOrderException):
|
|
order = exchange.create_stoploss(pair='ETH/BTC', amount=1, stop_price=190,
|
|
order_types={'stoploss_on_exchange_limit_ratio': 1.05},
|
|
side=side,
|
|
leverage=1.0)
|
|
|
|
api_mock.create_order.reset_mock()
|
|
order_types = {} if limitratio is None else {'stoploss_on_exchange_limit_ratio': limitratio}
|
|
order = exchange.create_stoploss(
|
|
pair='ETH/BTC', amount=1, stop_price=220, order_types=order_types, side=side, leverage=1.0)
|
|
|
|
assert 'id' in order
|
|
assert 'info' in order
|
|
assert order['id'] == order_id
|
|
assert api_mock.create_order.call_args_list[0][1]['symbol'] == 'ETH/BTC'
|
|
assert api_mock.create_order.call_args_list[0][1]['type'] == order_type
|
|
assert api_mock.create_order.call_args_list[0][1]['side'] == 'sell'
|
|
assert api_mock.create_order.call_args_list[0][1]['amount'] == 1
|
|
# Price should be 1% below stopprice
|
|
assert api_mock.create_order.call_args_list[0][1]['price'] == expected
|
|
assert api_mock.create_order.call_args_list[0][1]['params'] == {"stopPrice": 220,
|
|
"operator": "lte",
|
|
}
|
|
|
|
# test exception handling
|
|
with pytest.raises(DependencyException):
|
|
api_mock.create_order = MagicMock(side_effect=ccxt.InsufficientFunds("0 balance"))
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, 'huobi')
|
|
exchange.create_stoploss(pair='ETH/BTC', amount=1, stop_price=220,
|
|
order_types={}, side=side, leverage=1.0)
|
|
|
|
with pytest.raises(InvalidOrderException):
|
|
api_mock.create_order = MagicMock(
|
|
side_effect=ccxt.InvalidOrder("binance Order would trigger immediately."))
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, 'binance')
|
|
exchange.create_stoploss(pair='ETH/BTC', amount=1, stop_price=220,
|
|
order_types={}, side=side, leverage=1.0)
|
|
|
|
ccxt_exceptionhandlers(mocker, default_conf, api_mock, "huobi",
|
|
"create_stoploss", "create_order", retries=1,
|
|
pair='ETH/BTC', amount=1, stop_price=220, order_types={},
|
|
side=side, leverage=1.0)
|
|
|
|
|
|
def test_create_stoploss_order_dry_run_huobi(default_conf, mocker):
|
|
api_mock = MagicMock()
|
|
order_type = 'stop-limit'
|
|
default_conf['dry_run'] = True
|
|
mocker.patch(f'{EXMS}.amount_to_precision', lambda s, x, y: y)
|
|
mocker.patch(f'{EXMS}.price_to_precision', lambda s, x, y, **kwargs: y)
|
|
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, 'huobi')
|
|
|
|
with pytest.raises(InvalidOrderException):
|
|
order = exchange.create_stoploss(pair='ETH/BTC', amount=1, stop_price=190,
|
|
order_types={'stoploss_on_exchange_limit_ratio': 1.05},
|
|
side='sell', leverage=1.0)
|
|
|
|
api_mock.create_order.reset_mock()
|
|
|
|
order = exchange.create_stoploss(pair='ETH/BTC', amount=1, stop_price=220,
|
|
order_types={}, side='sell', leverage=1.0)
|
|
|
|
assert 'id' in order
|
|
assert 'info' in order
|
|
assert 'type' in order
|
|
|
|
assert order['type'] == order_type
|
|
assert order['price'] == 220
|
|
assert order['amount'] == 1
|
|
|
|
|
|
def test_stoploss_adjust_huobi(mocker, default_conf):
|
|
exchange = get_patched_exchange(mocker, default_conf, id='huobi')
|
|
order = {
|
|
'type': 'stop',
|
|
'price': 1500,
|
|
'stopPrice': '1500',
|
|
}
|
|
assert exchange.stoploss_adjust(1501, order, 'sell')
|
|
assert not exchange.stoploss_adjust(1499, order, 'sell')
|
|
# Test with invalid order case
|
|
assert exchange.stoploss_adjust(1501, order, 'sell')
|