Matthias
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from freqtrade.exceptions import RetryableOrderError
|
|
from freqtrade.exchange.common import API_RETRY_COUNT
|
|
from tests.conftest import EXMS, get_patched_exchange
|
|
from tests.exchange.test_exchange import ccxt_exceptionhandlers
|
|
|
|
|
|
@pytest.mark.usefixtures("init_persistence")
|
|
def test_fetch_stoploss_order_bitget(default_conf, mocker):
|
|
default_conf["dry_run"] = False
|
|
mocker.patch("freqtrade.exchange.common.time.sleep")
|
|
api_mock = MagicMock()
|
|
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, exchange="bitget")
|
|
|
|
api_mock.fetch_open_orders = MagicMock(return_value=[])
|
|
api_mock.fetch_canceled_and_closed_orders = MagicMock(return_value=[])
|
|
|
|
with pytest.raises(RetryableOrderError):
|
|
exchange.fetch_stoploss_order("1234", "ETH/BTC")
|
|
assert api_mock.fetch_open_orders.call_count == API_RETRY_COUNT + 1
|
|
assert api_mock.fetch_canceled_and_closed_orders.call_count == API_RETRY_COUNT + 1
|
|
|
|
api_mock.fetch_open_orders.reset_mock()
|
|
api_mock.fetch_canceled_and_closed_orders.reset_mock()
|
|
|
|
api_mock.fetch_canceled_and_closed_orders = MagicMock(
|
|
return_value=[{"id": "1234", "status": "closed", "info": {"ordId": "123455"}}]
|
|
)
|
|
mocker.patch(f"{EXMS}.fetch_order", MagicMock(return_value={"id": "123455"}))
|
|
resp = exchange.fetch_stoploss_order("1234", "ETH/BTC")
|
|
assert api_mock.fetch_open_orders.call_count == 1
|
|
assert api_mock.fetch_canceled_and_closed_orders.call_count == 1
|
|
|
|
assert resp["id"] == "1234"
|
|
assert resp["id_stop"] == "123455"
|
|
assert resp["type"] == "stoploss"
|
|
|
|
default_conf["dry_run"] = True
|
|
exchange = get_patched_exchange(mocker, default_conf, api_mock, exchange="bitget")
|
|
dro_mock = mocker.patch(f"{EXMS}.fetch_dry_run_order", MagicMock(return_value={"id": "123455"}))
|
|
|
|
api_mock.fetch_open_orders.reset_mock()
|
|
api_mock.fetch_canceled_and_closed_orders.reset_mock()
|
|
resp = exchange.fetch_stoploss_order("1234", "ETH/BTC")
|
|
|
|
assert api_mock.fetch_open_orders.call_count == 0
|
|
assert api_mock.fetch_canceled_and_closed_orders.call_count == 0
|
|
assert dro_mock.call_count == 1
|
|
|
|
|
|
def test_fetch_stoploss_order_bitget_exceptions(default_conf_usdt, mocker):
|
|
default_conf_usdt["dry_run"] = False
|
|
api_mock = MagicMock()
|
|
|
|
# Test emulation of the stoploss getters
|
|
api_mock.fetch_canceled_and_closed_orders = MagicMock(return_value=[])
|
|
|
|
ccxt_exceptionhandlers(
|
|
mocker,
|
|
default_conf_usdt,
|
|
api_mock,
|
|
"bitget",
|
|
"fetch_stoploss_order",
|
|
"fetch_open_orders",
|
|
retries=API_RETRY_COUNT + 1,
|
|
order_id="12345",
|
|
pair="ETH/USDT",
|
|
)
|