Linters & Create Tests

openweathermap · Dec 18, 2023 · 64acb85 · 64acb85
1 parent b9dd392
commit 64acb85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/deker/locks.py b/deker/locks.py
@@ -456,6 +456,10 @@ def get_result(self, func: Callable, args: Any, kwargs: Any) -> Any:
         return result
 
     def release(self, e: Optional[Exception] = None) -> None:
+        """Release Flock.
+
+        :param e: exception that might have been raised
+        """
         self.path.unlink(missing_ok=True)
         super().release(e)
 

diff --git a/tests/test_cases/test_concurrency/test_in_processes.py b/tests/test_cases/test_concurrency/test_in_processes.py
@@ -36,6 +36,7 @@
     UpdateMetaAttributeLock,
     WriteArrayLock,
     WriteVarrayLock,
+    CreateArrayLock,
 )
 from deker.schemas import ArraySchema, DimensionSchema, VArraySchema
 from deker.tools import get_paths
@@ -104,7 +105,9 @@ def call_array_method(
     # Get Array object
     if method == "create":
         with patch.object(
-            Flock, "release", wait_unlock(Flock.release, lock_set, funcs_finished, wait)
+            CreateArrayLock,
+            "release",
+            wait_unlock(CreateArrayLock.release, lock_set, funcs_finished, wait),
         ):
             with patch("deker.ABC.base_array.get_id", lambda *a: id_):
                 try: