[WIP][RestExchange] add test for replace order

Author: techfreaque

tests/exchanges/traders/test_trader.py

@@ -915,6 +915,59 @@ async def test_edit_order(self):
 
         await self.stop(exchange_manager)
 
+    async def test_edit_order_by_replacing(self):
+        # test with binance usdm as it supports edit order by replacing
+        _, exchange_manager, trader_inst = await self.init_default(is_future=True)
+        portfolio_manager = exchange_manager.exchange_personal_data.portfolio_manager
+        currency, market = commons_symbols.parse_symbol(self.DEFAULT_SYMBOL).base_and_quote()
+        assert portfolio_manager.portfolio.portfolio[currency].available == decimal.Decimal(10)
+        assert portfolio_manager.portfolio.portfolio[market].available == decimal.Decimal(1000)
+
+        market_buy = BuyMarketOrder(trader_inst)
+        market_buy.update(order_type=TraderOrderType.BUY_MARKET,
+                          symbol=self.DEFAULT_SYMBOL,
+                          current_price=decimal.Decimal("70"),
+                          quantity=decimal.Decimal("10"),
+                          price=decimal.Decimal("70"))
+        with pytest.raises(OrderEditError):
+            # market orders can't be edited
+            await trader_inst.edit_order(market_buy, edited_price=decimal.Decimal("100"))
+        assert market_buy.origin_quantity == decimal.Decimal("10")
+        assert market_buy.origin_price == decimal.Decimal("70")
+        assert market_buy.origin_stop_price == decimal.Decimal("0")
+        assert market_buy.created_last_price == decimal.Decimal("70")
+
+        limit_buy = BuyLimitOrder(trader_inst)
+        limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
+                         symbol=self.DEFAULT_SYMBOL,
+                         current_price=decimal.Decimal("70"),
+                         quantity=decimal.Decimal("10"),
+                         price=decimal.Decimal("70"))
+        assert portfolio_manager.portfolio.portfolio[currency].available == decimal.Decimal(10)
+        assert portfolio_manager.portfolio.portfolio[market].available == decimal.Decimal(1000)
+        exchange_manager.exchange_personal_data.portfolio_manager.portfolio.update_portfolio_available(
+            limit_buy, is_new_order=True
+        )
+        assert portfolio_manager.portfolio.portfolio[currency].available == decimal.Decimal(10)
+        assert portfolio_manager.portfolio.portfolio[market].available == decimal.Decimal(300)
+        await trader_inst.edit_order(
+            limit_buy,
+            edited_quantity=decimal.Decimal("4"),
+            edited_price=decimal.Decimal("42"),
+            edited_stop_price=decimal.Decimal("424"),
+            edited_current_price=decimal.Decimal("4242"),
+        )
+        assert limit_buy.origin_quantity == decimal.Decimal("4")
+        assert limit_buy.origin_price == decimal.Decimal("42")
+        assert limit_buy.origin_stop_price == decimal.Decimal("424")
+        assert limit_buy.created_last_price == decimal.Decimal("4242")
+        # also updated portfolio
+        assert portfolio_manager.portfolio.portfolio[currency].available == decimal.Decimal(10)
+        assert portfolio_manager.portfolio.portfolio[market].available == decimal.Decimal(832)
+        limit_buy.clear()
+
+        await self.stop(exchange_manager)
+
     async def test_bundle_chained_order_with_uncreated_order(self):
         _, exchange_manager, trader_inst = await self.init_default()