update unittest

Polygant · Sep 25, 2023 · 30ec16b · 30ec16b
1 parent 9fc7e7b
commit 30ec16b
Showing 1 changed file with 93 additions and 1 deletion.
diff --git a/test/hummingbot/connector/derivative/kucoin_perpetual/test_kucoin_perpetual_derivative.py b/test/hummingbot/connector/derivative/kucoin_perpetual/test_kucoin_perpetual_derivative.py
@@ -15,10 +15,11 @@
 from hummingbot.client.config.client_config_map import ClientConfigMap
 from hummingbot.client.config.config_helpers import ClientConfigAdapter
 from hummingbot.connector.derivative.kucoin_perpetual.kucoin_perpetual_derivative import KucoinPerpetualDerivative
+from hummingbot.connector.derivative.position import Position
 from hummingbot.connector.test_support.perpetual_derivative_test import AbstractPerpetualDerivativeTests
 from hummingbot.connector.trading_rule import TradingRule
 from hummingbot.connector.utils import combine_to_hb_trading_pair
-from hummingbot.core.data_type.common import OrderType, PositionAction, PositionMode, TradeType
+from hummingbot.core.data_type.common import OrderType, PositionAction, PositionMode, TradeType, PositionSide
 from hummingbot.core.data_type.funding_info import FundingInfo
 from hummingbot.core.data_type.in_flight_order import InFlightOrder
 from hummingbot.core.data_type.trade_fee import AddedToCostTradeFee, TokenAmount, TradeFeeBase
@@ -1555,3 +1556,94 @@ def test_start_network_update_trading_rules(self, mock_api):
         self.assertEqual(1, len(self.exchange.trading_rules))
         self.assertIn(self.trading_pair, self.exchange.trading_rules)
         self.assertEqual(repr(self.expected_trading_rule), repr(self.exchange.trading_rules[self.trading_pair]))
+
+    @aioresponses()
+    def test_user_stream_update_for_order_full_fill(self, mock_api):
+        self.exchange._set_current_timestamp(1640780000)
+        leverage = 2
+        self.exchange._perpetual_trading.set_leverage(self.trading_pair, leverage)
+        self.exchange.start_tracking_order(
+            order_id=self.client_order_id_prefix + "1",
+            exchange_order_id=self.exchange_order_id_prefix + "1",
+            trading_pair=self.trading_pair,
+            order_type=OrderType.LIMIT,
+            trade_type=TradeType.SELL,
+            price=Decimal("10000"),
+            amount=Decimal("1"),
+            position_action=PositionAction.OPEN,
+        )
+        order = self.exchange.in_flight_orders[self.client_order_id_prefix + "1"]
+
+        order_event = self.order_event_for_full_fill_websocket_update(order=order)
+        trade_event = self.trade_event_for_full_fill_websocket_update(order=order)
+        expected_unrealized_pnl = 12
+        position_event = self.position_event_for_full_fill_websocket_update(
+            order=order, unrealized_pnl=expected_unrealized_pnl
+        )
+
+        mock_queue = AsyncMock()
+        event_messages = []
+        if trade_event:
+            event_messages.append(trade_event)
+        if order_event:
+            event_messages.append(order_event)
+        if position_event:
+            event_messages.append(position_event)
+        event_messages.append(asyncio.CancelledError)
+        mock_queue.get.side_effect = event_messages
+        self.exchange._user_stream_tracker._user_stream = mock_queue
+
+        if self.is_order_fill_http_update_executed_during_websocket_order_event_processing:
+            self.configure_full_fill_trade_response(
+                order=order,
+                mock_api=mock_api)
+
+        try:
+            self.async_run_with_timeout(self.exchange._user_stream_event_listener())
+        except asyncio.CancelledError:
+            pass
+        # Execute one more synchronization to ensure the async task that processes the update is finished
+        self.async_run_with_timeout(order.wait_until_completely_filled())
+
+        fill_event = self.order_filled_logger.event_log[0]
+        self.assertEqual(self.exchange.current_timestamp, fill_event.timestamp)
+        self.assertEqual(order.client_order_id, fill_event.order_id)
+        self.assertEqual(order.trading_pair, fill_event.trading_pair)
+        self.assertEqual(order.trade_type, fill_event.trade_type)
+        self.assertEqual(order.order_type, fill_event.order_type)
+        self.assertEqual(order.price, fill_event.price)
+        self.assertEqual(order.amount, fill_event.amount)
+        expected_fee = self.expected_fill_fee
+        self.assertEqual(expected_fee, fill_event.trade_fee)
+        self.assertEqual(leverage, fill_event.leverage)
+        self.assertEqual(PositionAction.OPEN.value, fill_event.position)
+
+        sell_event = self.sell_order_completed_logger.event_log[0]
+        self.assertEqual(self.exchange.current_timestamp, sell_event.timestamp)
+        self.assertEqual(order.client_order_id, sell_event.order_id)
+        self.assertEqual(order.base_asset, sell_event.base_asset)
+        self.assertEqual(order.quote_asset, sell_event.quote_asset)
+        self.assertEqual(order.amount, sell_event.base_asset_amount)
+        self.assertEqual(order.amount * fill_event.price, sell_event.quote_asset_amount)
+        self.assertEqual(order.order_type, sell_event.order_type)
+        self.assertEqual(order.exchange_order_id, sell_event.exchange_order_id)
+        self.assertNotIn(order.client_order_id, self.exchange.in_flight_orders)
+        self.assertTrue(order.is_filled)
+        self.assertTrue(order.is_done)
+
+        self.assertTrue(
+            self.is_logged(
+                "INFO",
+                f"SELL order {order.client_order_id} completely filled."
+            )
+        )
+
+        self.assertEqual(1, len(self.exchange.account_positions))
+
+        position: Position = self.exchange.account_positions[self.trading_pair]
+        self.assertEqual(self.trading_pair, position.trading_pair)
+        self.assertEqual(PositionSide.SHORT, position.position_side)
+        self.assertEqual(expected_unrealized_pnl, position.unrealized_pnl)
+        self.assertEqual(fill_event.price, position.entry_price)
+        self.assertEqual(-fill_event.amount, (self.exchange.get_quantity_of_contracts(self.trading_pair, position.amount)))
+        self.assertEqual(leverage, position.leverage)