def setUp(self) -> None:
super().setUp()
throttler = AsyncThrottler(CONSTANTS.RATE_LIMITS)
self.tracker: MexcOrderBookTracker = MexcOrderBookTracker(throttler=throttler,
trading_pairs=[self.trading_pair])
self.tracking_task = None
# Simulate start()
self.tracker._order_books[self.trading_pair] = MexcOrderBook()
self.tracker._tracking_message_queues[self.trading_pair] = asyncio.Queue()
self.tracker._order_books_initialized.set()
def test_get_order_book_for_valid_trading_pair(self):
dummy_order_book = MexcOrderBook()
self.exchange.order_book_tracker.order_books["BTC-USDT"] = dummy_order_book
self.assertEqual(dummy_order_book, self.exchange.get_order_book("BTC-USDT"))