From 3229571141b638d9d5d6985e98b7e61642229b77 Mon Sep 17 00:00:00 2001 From: schneefux Date: Sun, 5 Mar 2017 10:36:14 +0100 Subject: support job batching (#12) * support job batching * worker: pull more frequently --- test_joblib.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'test_joblib.py') diff --git a/test_joblib.py b/test_joblib.py index 865cfab..1a760f8 100644 --- a/test_joblib.py +++ b/test_joblib.py @@ -48,6 +48,16 @@ class TestJoblib: # there should not be another job assert None == (await queue.acquire(jobtype="testing"))[1] + @pytest.mark.asyncio + async def test_request_and_acquire_batch(self, queue, payload): + # request should succeed + await queue.request(jobtype="testing", payload=[payload]*5) + # acquire should return same payload + for _ in range(5): + assert payload == (await queue.acquire(jobtype="testing"))[1] + # there should not be another job + assert None == (await queue.acquire(jobtype="testing"))[1] + @pytest.mark.asyncio async def test_cleanup(self, queue, payload): await queue.request(jobtype="testing", payload=payload) -- cgit v1.3.1