From 80845807e113527fcbf2f716db62a4beb2ddae89 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sat, 23 Jul 2022 15:14:38 +0200 Subject: [PATCH] Improve some test resiliance --- tests/test_freqtradebot.py | 14 ++++++++------ tests/test_persistence.py | 8 ++++++-- 2 files changed, 14 insertions(+), 8 deletions(-) diff --git a/tests/test_freqtradebot.py b/tests/test_freqtradebot.py index e431e7ac3..a4b10fbcd 100644 --- a/tests/test_freqtradebot.py +++ b/tests/test_freqtradebot.py @@ -2139,8 +2139,6 @@ def test_handle_trade( assert trade time.sleep(0.01) # Race condition fix - oobj = Order.parse_from_ccxt_object(enter_order, enter_order['symbol'], entry_side(is_short)) - trade.update_trade(oobj) assert trade.is_open is True freqtrade.wallets.update() @@ -2150,11 +2148,15 @@ def test_handle_trade( assert trade.open_order_id == exit_order['id'] # Simulate fulfilled LIMIT_SELL order for trade - oobj = Order.parse_from_ccxt_object(exit_order, exit_order['symbol'], exit_side(is_short)) - trade.update_trade(oobj) + trade.orders[-1].ft_is_open = False + trade.orders[-1].status = 'closed' + trade.orders[-1].filled = trade.orders[-1].remaining + trade.orders[-1].remaining = 0.0 - assert trade.close_rate == 2.0 if is_short else 2.2 - assert trade.close_profit == close_profit + trade.update_trade(trade.orders[-1]) + + assert trade.close_rate == (2.0 if is_short else 2.2) + assert pytest.approx(trade.close_profit) == close_profit assert trade.calc_profit(trade.close_rate) == 5.685 assert trade.close_date is not None assert trade.exit_reason == 'sell_signal1' diff --git a/tests/test_persistence.py b/tests/test_persistence.py index a09711048..529f6dc31 100644 --- a/tests/test_persistence.py +++ b/tests/test_persistence.py @@ -481,6 +481,7 @@ def test_update_limit_order(fee, caplog, limit_buy_order_usdt, limit_sell_order_ trade.open_order_id = 'something' oobj = Order.parse_from_ccxt_object(enter_order, 'ADA/USDT', entry_side) + trade.orders.append(oobj) trade.update_trade(oobj) assert trade.open_order_id is None assert trade.open_rate == open_rate @@ -496,11 +497,12 @@ def test_update_limit_order(fee, caplog, limit_buy_order_usdt, limit_sell_order_ trade.open_order_id = 'something' time_machine.move_to("2022-03-31 21:45:05 +00:00") oobj = Order.parse_from_ccxt_object(exit_order, 'ADA/USDT', exit_side) + trade.orders.append(oobj) trade.update_trade(oobj) assert trade.open_order_id is None assert trade.close_rate == close_rate - assert trade.close_profit == profit + assert pytest.approx(trade.close_profit) == profit assert trade.close_date is not None assert log_has_re(f"LIMIT_{exit_side.upper()} has been fulfilled for " r"Trade\(id=2, pair=ADA/USDT, amount=30.00000000, " @@ -529,6 +531,7 @@ def test_update_market_order(market_buy_order_usdt, market_sell_order_usdt, fee, trade.open_order_id = 'something' oobj = Order.parse_from_ccxt_object(market_buy_order_usdt, 'ADA/USDT', 'buy') + trade.orders.append(oobj) trade.update_trade(oobj) assert trade.open_order_id is None assert trade.open_rate == 2.0 @@ -543,10 +546,11 @@ def test_update_market_order(market_buy_order_usdt, market_sell_order_usdt, fee, trade.is_open = True trade.open_order_id = 'something' oobj = Order.parse_from_ccxt_object(market_sell_order_usdt, 'ADA/USDT', 'sell') + trade.orders.append(oobj) trade.update_trade(oobj) assert trade.open_order_id is None assert trade.close_rate == 2.2 - assert trade.close_profit == round(0.0945137157107232, 8) + assert pytest.approx(trade.close_profit) == 0.094513715710723 assert trade.close_date is not None assert log_has_re(r"MARKET_SELL has been fulfilled for Trade\(id=1, " r"pair=ADA/USDT, amount=30.00000000, is_short=False, leverage=1.0, "