From e8a9534fa85022c87ef99dd7fd1569b69261bc40 Mon Sep 17 00:00:00 2001
From: David Waroquiers <david.waroquiers@matgenix.com>
Date: Thu, 18 Jul 2024 15:33:33 +0200
Subject: [PATCH] Added test for MongoLock.

---
 tests/db/utils/test_db.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 tests/db/utils/test_db.py

diff --git a/tests/db/utils/test_db.py b/tests/db/utils/test_db.py
new file mode 100644
index 00000000..7ce84c7f
--- /dev/null
+++ b/tests/db/utils/test_db.py
@@ -0,0 +1,26 @@
+from jobflow_remote.utils.db import MongoLock
+
+
+class TestMongoLock:
+    def test_context(self, mongoclient):
+        db = mongoclient["mytestdb"]
+        collection = db["mytestcollection"]
+        collection.drop()
+        collection.insert_one({"a": 1})
+        collection.insert_one({"b": 2})
+        with MongoLock(collection=collection, filter={"a": 1}) as lock:
+            assert lock.locked_document is not None
+            lock_id = lock.lock_id
+            assert lock_id == lock.get_lock_id(lock.locked_document)
+            assert lock.is_locked is False
+            with MongoLock(collection=collection, filter={"a": 1}) as lock2:
+                assert lock2.locked_document is None
+                assert lock2.unavailable_document is None
+                assert lock2.is_locked is True
+            with MongoLock(
+                collection=collection, filter={"a": 1}, get_locked_doc=True
+            ) as lock3:
+                assert lock3.locked_document is None
+                assert lock3.unavailable_document is not None
+                assert lock_id == lock3.get_lock_id(lock3.unavailable_document)
+                assert lock3.is_locked is True