Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

test refac

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-10 13:05:42 +01:00
parent ce1217c191
commit 1347fb57f8
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyinfra/test/unit_tests/consumer_test.py
View File

@ -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]