From 596ee7f64662dd94f1a21967f15d451bad969a2e Mon Sep 17 00:00:00 2001 From: schneefux Date: Mon, 6 Mar 2017 20:15:51 +0100 Subject: support acquiring in batches --- test_joblib.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'test_joblib.py') diff --git a/test_joblib.py b/test_joblib.py index 1a760f8..fc9640b 100644 --- a/test_joblib.py +++ b/test_joblib.py @@ -58,6 +58,15 @@ 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_batched(self, queue, payload): + # request should succeed + await queue.request(jobtype="testing", payload=[payload]*5) + # acquire should return same payload + assert payload == (await queue.acquire(jobtype="testing", length=5))[0][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