add job fails (#4)

1 files changed, 14 insertions, 0 deletions

diff --git a/test_joblib.py b/test_joblib.py
index 47eedec..f9aa875 100644
--- a/test_joblib.py
+++ b/test_joblib.py
@@ -2,6 +2,7 @@
 
 import os
 import asyncio
+import json
 import asyncpg
 import pytest
 import joblib
@@ -64,6 +65,19 @@ class TestJoblib:
         assert 9 == (await queue.acquire(jobtype="testing"))[2]
 
     @pytest.mark.asyncio
+    async def test_fail(self, queue, payload):
+        err = "testing errors"
+        await queue.request(jobtype="testing", payload=payload)
+        jobid, _, _ = await queue.acquire(jobtype="testing")
+        await queue.fail(jobid, err)
+        async with queue._pool.acquire() as con:
+            jid, pl = await con.fetchrow(
+                "SELECT id, payload FROM jobs WHERE status='failed'")
+            assert jid == jobid
+            payload["error"] = err
+            assert json.loads(pl) == payload
+
+    @pytest.mark.asyncio
     async def test_finish(self, queue, payload):
         await queue.request(jobtype="testing", payload=payload)
         jobid, _, _ = await queue.acquire(jobtype="testing")