def test__get_event_ids(self): mock_markets = [ mock.Mock(event_id=1, event_type_id="1", closed=False), mock.Mock(event_id=1, event_type_id="1", closed=False), mock.Mock(event_id=2, event_type_id="1", closed=False), mock.Mock(event_id=3, event_type_id="1", closed=True), mock.Mock(event_id=4, event_type_id="7", closed=False), ] self.assertEqual(utils.get_event_ids(mock_markets, "1"), [1, 2])
def poll_in_play_service(context: dict, flumine, event_type_id: str) -> None: trading = flumine.client.betting_client event_ids = get_event_ids(flumine.markets, event_type_id=event_type_id) for event_id in event_ids: response = trading.in_play_service.get_scores(event_ids=[event_id]) if response is None: logger.warning( "poll_in_play_service", extra={"response": response, "event_id": event_id}, ) continue for score in response: flumine.handler_queue.put(CustomEvent(score, callback))