Merge pull request #24 from Joshuaalbert/f23

* Do #23
Joshuaalbert · Oct 16, 2024 · d1299ec · d1299ec
2 parents 424be8b + 19006b1
commit d1299ec
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/fair_async_rlock/fair_async_rlock.py b/fair_async_rlock/fair_async_rlock.py
@@ -22,6 +22,9 @@ def is_owner(self, task=None):
             task = asyncio.current_task()
         return self._owner == task
 
+    def locked(self) -> bool:
+        return self._owner is not None
+
     async def acquire(self):
         """Acquire the lock."""
         me = asyncio.current_task()

diff --git a/fair_async_rlock/tests/test_fair_async_rlock.py b/fair_async_rlock/tests/test_fair_async_rlock.py
@@ -652,3 +652,13 @@ async def task3():
     t3 = asyncio.create_task(task3())
 
     await asyncio.gather(t1, t2, t3)
+
+@pytest.mark.asyncio
+def test_locked():
+    lock = FairAsyncRLock()
+    assert not lock.locked()
+    async def task():
+        async with lock:
+            assert lock.locked()
+    asyncio.run(task())
+    assert not lock.locked()