Test all table lock modes

SectorLabs · Apr 6, 2023 · c79a8ca · c79a8ca
1 parent 0bb392d
commit c79a8ca
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/psqlextra/locking.py b/psqlextra/locking.py
@@ -19,6 +19,13 @@ class PostgresTableLockMode(Enum):
     EXCLUSIVE = "EXCLUSIVE"
     ACCESS_EXCLUSIVE = "ACCESS EXCLUSIVE"
 
+    @property
+    def alias(self) -> str:
+        return (
+            "".join([word.title() for word in self.name.lower().split("_")])
+            + "Lock"
+        )
+
 
 def postgres_lock_table(
     table_name: str,

diff --git a/tests/test_locking.py b/tests/test_locking.py
@@ -66,18 +66,17 @@ def test_postgres_lock_table_in_schema():
     assert lock_signature not in get_table_locks()
 
 
+@pytest.mark.parametrize("lock_mode", list(PostgresTableLockMode))
 @pytest.mark.django_db(transaction=True)
-def test_postgres_lock_mode(mocked_model):
+def test_postgres_lock_model(mocked_model, lock_mode):
     lock_signature = (
         "public",
         mocked_model._meta.db_table,
-        "AccessExclusiveLock",
+        lock_mode.alias,
     )
 
     with transaction.atomic():
-        postgres_lock_model(
-            mocked_model, PostgresTableLockMode.ACCESS_EXCLUSIVE
-        )
+        postgres_lock_model(mocked_model, lock_mode)
         assert lock_signature in get_table_locks()
 
     assert lock_signature not in get_table_locks()