From 30ec16b8767fc9e3fbcc044a84f156a8c1a4a3c6 Mon Sep 17 00:00:00 2001 From: bczhang Date: Mon, 25 Sep 2023 13:30:31 +0800 Subject: [PATCH] update unittest --- .../test_kucoin_perpetual_derivative.py | 94 ++++++++++++++++++- 1 file changed, 93 insertions(+), 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 index a2d3aeea2c..233a3e6eb0 100644 --- 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) \ No newline at end of file