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)