Fix InMemoryConnector handling of queueing_lock

Éric Lemoine · Éric Lemoine · commit 81674acd81f1 · 2020-12-18T17:10:27.000+01:00
diff --git a/procrastinate/testing.py b/procrastinate/testing.py
@@ -93,7 +93,7 @@ async def close_async(self) -> None:
     def defer_job_one(
         self, task_name, lock, queueing_lock, args, scheduled_at, queue
     ) -> JobRow:
-        if any(
+        if queueing_lock is not None and any(
             job for job in self.jobs.values() if job["queueing_lock"] == queueing_lock
         ):
             raise exceptions.UniqueViolation(
diff --git a/tests/unit/test_testing.py b/tests/unit/test_testing.py
@@ -84,6 +84,26 @@ def test_defer_job_one(connector):
     assert connector.jobs[1] == job
 
 
+def test_defer_job_one_multiple_times(connector):
+    connector.defer_job_one(
+        task_name="mytask",
+        lock=None,
+        queueing_lock=None,
+        args={},
+        scheduled_at=None,
+        queue="default",
+    )
+    connector.defer_job_one(
+        task_name="mytask",
+        lock=None,
+        queueing_lock=None,
+        args={},
+        scheduled_at=None,
+        queue="default",
+    )
+    assert len(connector.jobs) == 2
+
+
 def test_current_locks(connector):
     connector.jobs = {
         1: {"status": "todo", "lock": "foo"},
