Merge branch 'master' of https://github.com/vainglorygame/joblib

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]