def test_get_conversation_ids_to_process_error( requested_ids: Optional[List[Text]], available_ids: List[Text], exception: Exception): # create and mock tracker store containing `available_ids` as keys tracker_store = Mock() tracker_store.keys.return_value = available_ids exporter = MockExporter(tracker_store) exporter.requested_conversation_ids = requested_ids with pytest.raises(exception): # noinspection PyProtectedMember exporter._get_conversation_ids_to_process()
def test_get_conversation_ids_to_process( requested_ids: Optional[List[Text]], available_ids: Optional[List[Text]], expected: Optional[List[Text]], ): # create and mock tracker store containing `available_ids` as keys tracker_store = Mock() tracker_store.keys.return_value = available_ids exporter = MockExporter(tracker_store) exporter.requested_conversation_ids = requested_ids # noinspection PyProtectedMember assert exporter._get_conversation_ids_to_process() == set(expected)