Add isLocked property fix #3

mattmassicotte · Sep 30, 2024 · 9ba857c · 9ba857c
1 parent ffb00a2
commit 9ba857c
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/Sources/Lock/AsyncLock.swift b/Sources/Lock/AsyncLock.swift
@@ -71,5 +71,15 @@ public final class AsyncLock {
 			throw error
 		}
 	}
-}
 
+	public var isLocked: Bool {
+		get {
+			switch state {
+			case .unlocked:
+				false
+			case .locked:
+				true
+			}
+		}
+	}
+}
diff --git a/Tests/LockTests/LockTests.swift b/Tests/LockTests/LockTests.swift
@@ -63,4 +63,26 @@ struct LockTests {
 			try await task.value
 		}
 	}
+
+	@Test
+	func checkLock() async throws {
+		let lock = AsyncLock()
+
+		#expect(lock.isLocked == false)
+		await lock.lock()
+		#expect(lock.isLocked == true)
+		lock.unlock()
+		#expect(lock.isLocked == false)
+	}
+
+	@Test
+	func checkLockWithLocked() async throws {
+		let lock = AsyncLock()
+
+		#expect(lock.isLocked == false)
+		await lock.withLock {
+			#expect(lock.isLocked == true)
+		}
+		#expect(lock.isLocked == false)
+	}
 }