From 1347fb57f858add6cf3e9e78ba16f82231e51798 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Thu, 10 Mar 2022 13:05:42 +0100 Subject: [PATCH] test refac --- pyinfra/test/unit_tests/consumer_test.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/pyinfra/test/unit_tests/consumer_test.py b/pyinfra/test/unit_tests/consumer_test.py index 547137e..e93599b 100644 --- a/pyinfra/test/unit_tests/consumer_test.py +++ b/pyinfra/test/unit_tests/consumer_test.py @@ -29,14 +29,18 @@ def test_consuming_empty_input_queue_does_not_put_anything_on_output_queue(consu def test_consuming_nonempty_input_queue_puts_messages_on_output_queue_in_fifo_order(consumer, queue_manager, callback): + + def produce_items(): + return range(3) + queue_manager.clear() - queue_manager.publish_request(1) - queue_manager.publish_request(2) + + for item in produce_items(): + queue_manager.publish_request(item) + requests = consumer.consume() - for i, r in enumerate(requests): + for _, r in zip(produce_items(), requests): queue_manager.publish_response(r, callback) - if i == 1: - break - assert queue_manager.output_queue.to_list() == [1, 4] + assert queue_manager.output_queue.to_list() == [0, 1, 4]