From 84e9251fcd3d2f5bbab7404f3337791f6bf5da72 Mon Sep 17 00:00:00 2001 From: Matthias Date: Mon, 15 Dec 2025 07:01:23 +0100 Subject: [PATCH] test: use object patching instead of direct assignment --- tests/exchange/test_gate.py | 4 +--- tests/exchange/test_okx.py | 11 +++++------ 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/tests/exchange/test_gate.py b/tests/exchange/test_gate.py index ee3d70c84..9c3246f46 100644 --- a/tests/exchange/test_gate.py +++ b/tests/exchange/test_gate.py @@ -42,9 +42,7 @@ def test_fetch_stoploss_order_gate(default_conf, mocker): def test_cancel_stoploss_order_gate(default_conf, mocker): exchange = get_patched_exchange(mocker, default_conf, exchange="gate") - - cancel_order_mock = MagicMock() - exchange.cancel_order = cancel_order_mock + cancel_order_mock = mocker.patch.object(exchange, "cancel_order") exchange.cancel_stoploss_order("1234", "ETH/BTC") assert cancel_order_mock.call_count == 1 diff --git a/tests/exchange/test_okx.py b/tests/exchange/test_okx.py index 750afbd40..43e14a51a 100644 --- a/tests/exchange/test_okx.py +++ b/tests/exchange/test_okx.py @@ -661,14 +661,13 @@ def test_stoploss_adjust_okx(mocker, default_conf, sl1, sl2, sl3, side): def test_stoploss_cancel_okx(mocker, default_conf): exchange = get_patched_exchange(mocker, default_conf, exchange="okx") - - exchange.cancel_order = MagicMock() + co_mock = mocker.patch.object(exchange, "cancel_order") exchange.cancel_stoploss_order("1234", "ETH/USDT") - assert exchange.cancel_order.call_count == 1 - assert exchange.cancel_order.call_args_list[0][1]["order_id"] == "1234" - assert exchange.cancel_order.call_args_list[0][1]["pair"] == "ETH/USDT" - assert exchange.cancel_order.call_args_list[0][1]["params"] == {"stop": True} + assert co_mock.call_count == 1 + assert co_mock.call_args_list[0][1]["order_id"] == "1234" + assert co_mock.call_args_list[0][1]["pair"] == "ETH/USDT" + assert co_mock.call_args_list[0][1]["params"] == {"stop": True} def test__get_stop_params_okx(mocker, default_conf):