def test_run( self, mock__process_market_books, mock__process_raw_data, mock__process_market_catalogues, mock__process_end_flumine, mock__process_current_orders, mock__process_order_package, mock__process_close_market, mock__process_cleared_markets, mock__process_cleared_orders, mock__process_custom_event, mock__add_default_workers, ): mock_events = [ events.MarketCatalogueEvent(None), events.MarketBookEvent(None), events.RawDataEvent(None), events.CurrentOrdersEvent(None), BaseOrderPackage(None, "1.123", [], "12", None), events.ClearedMarketsEvent(None), events.ClearedOrdersEvent(None), events.CloseMarketEvent(None), events.CustomEvent(None, None), events.NewDayEvent(None), events.TerminationEvent(None), ] for i in mock_events: self.flumine.handler_queue.put(i) self.flumine.run() mock__process_market_books.assert_called_with(mock_events[1]) mock__process_raw_data.assert_called_with(mock_events[2]) mock__process_market_catalogues.assert_called_with(mock_events[0]) mock__process_end_flumine.assert_called_with() mock__process_current_orders.assert_called_with(mock_events[3]) mock__process_order_package.assert_called_with(mock_events[4]) mock__process_close_market.assert_called_with(mock_events[7]) mock__process_cleared_markets.assert_called_with(mock_events[5]) mock__process_cleared_orders.assert_called_with(mock_events[6]) mock__process_custom_event.assert_called_with(mock_events[8]) mock__add_default_workers.assert_called()
def test_str(self): self.base_event = events.MarketBookEvent(None) self.assertEqual(str(self.base_event), "<MARKET_BOOK [HANDLER]>")