diff options
Diffstat (limited to 'test_joblib.py')
| -rw-r--r-- | test_joblib.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/test_joblib.py b/test_joblib.py index 35ae90e..47eedec 100644 --- a/test_joblib.py +++ b/test_joblib.py @@ -51,17 +51,22 @@ class TestJoblib: async def test_cleanup(self, queue, payload): await queue.request(jobtype="testing", payload=payload) # mark job as processing - jobid_1, payload_1 = await queue.acquire(jobtype="testing") + jobid_1, payload_1, _ = await queue.acquire(jobtype="testing") assert payload_1 == payload await queue.cleanup() # same job should be available again - jobid_2, payload_2 = await queue.acquire(jobtype="testing") + jobid_2, payload_2, _ = await queue.acquire(jobtype="testing") assert jobid_1 == jobid_2 and payload_1 == payload_2 @pytest.mark.asyncio + async def test_priority(self, queue, payload): + await queue.request(jobtype="testing", payload=payload, priority=9) + assert 9 == (await queue.acquire(jobtype="testing"))[2] + + @pytest.mark.asyncio async def test_finish(self, queue, payload): await queue.request(jobtype="testing", payload=payload) - jobid, _ = await queue.acquire(jobtype="testing") + jobid, _, _ = await queue.acquire(jobtype="testing") await queue.finish(jobid) # job should not be available again assert None == (await queue.acquire(jobtype="testing"))[1] |