diff --git a/tests/unit_test/queue_test.py b/tests/unit_test/queue_test.py index 492b24f..21d3a32 100644 --- a/tests/unit_test/queue_test.py +++ b/tests/unit_test/queue_test.py @@ -18,7 +18,21 @@ def make_callback(process_time): return callback +def file_not_found_callback(x): + raise FileNotFoundError("File not found") + + class TestQueueManager: + def test_not_available_file_leads_to_message_rejection_without_crashing( + self, queue_manager, input_message, stop_message + ): + queue_manager.purge_queues() + + queue_manager.publish_message_to_input_queue(input_message) + queue_manager.publish_message_to_input_queue(stop_message) + + queue_manager.start_consuming(file_not_found_callback) + def test_processing_of_several_messages(self, queue_manager, input_message, stop_message): queue_manager.purge_queues()