Code coverage

MarkCiliaVincenti · Dec 30, 2023 · abb7d24 · abb7d24
1 parent 2fc35e1
commit abb7d24
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 3 deletions.
diff --git a/AsyncKeyedLock.Tests/AsyncKeyedLocker/OriginalTests.cs b/AsyncKeyedLock.Tests/AsyncKeyedLocker/OriginalTests.cs
@@ -373,6 +373,19 @@ public void TestTimeoutWithZeroTimeSpanAndCancelledToken()
             asyncKeyedLocker.IsInUse("test").Should().BeFalse();
         }
 
+        [Fact]
+        public void TestTimeoutTryLock()
+        {
+            var asyncKeyedLocker = new AsyncKeyedLocker<string>();
+            using (asyncKeyedLocker.Lock("test"))
+            {
+                Assert.True(asyncKeyedLocker.IsInUse("test"));
+                Assert.False(asyncKeyedLocker.TryLock("test", () => { }, 0, CancellationToken.None));
+                Assert.False(asyncKeyedLocker.TryLock("test", () => { }, TimeSpan.Zero, CancellationToken.None));
+            }
+            Assert.False(asyncKeyedLocker.IsInUse("test"));
+        }
+
         [Fact]
         public async Task BasicTest()
         {

diff --git a/AsyncKeyedLock.Tests/StripedAsyncKeyedLocker/OriginalTests.cs b/AsyncKeyedLock.Tests/StripedAsyncKeyedLocker/OriginalTests.cs
@@ -77,7 +77,7 @@ public async Task TestTimeoutBasic()
         public void TestTimeoutBasicWithOutParameter()
         {
             var asyncKeyedLocker = new StripedAsyncKeyedLocker<string>();
-            using (var myLock = asyncKeyedLocker.Lock("test", 0, out var entered))
+            using (var myLock = asyncKeyedLocker.Lock("test", 0, out bool entered))
             {
                 Assert.True(entered);
                 Assert.True(asyncKeyedLocker.IsInUse("test"));
@@ -89,7 +89,7 @@ public void TestTimeoutBasicWithOutParameter()
         public async Task TestTimeout()
         {
             var asyncKeyedLocker = new StripedAsyncKeyedLocker<string>();
-            using (await asyncKeyedLocker.LockAsync("test"))
+            using (await asyncKeyedLocker.LockAsync("test", 0))
             {
                 using (var myLock = await asyncKeyedLocker.LockAsync("test", 0))
                 {
@@ -102,11 +102,27 @@ public async Task TestTimeout()
 
         [Fact]
         public void TestTimeoutWithTimeSpanSynchronous()
+        {
+            var asyncKeyedLocker = new StripedAsyncKeyedLocker<string>();
+            using (asyncKeyedLocker.Lock("test", TimeSpan.Zero, out bool entered))
+            {
+                Assert.True(entered);
+                using (asyncKeyedLocker.Lock("test", TimeSpan.Zero, out entered))
+                {
+                    Assert.False(entered);
+                }
+                Assert.True(asyncKeyedLocker.IsInUse("test"));
+            }
+            Assert.False(asyncKeyedLocker.IsInUse("test"));
+        }
+
+        [Fact]
+        public void TestTimeoutWithTimeoutSynchronous()
         {
             var asyncKeyedLocker = new StripedAsyncKeyedLocker<string>();
             using (asyncKeyedLocker.Lock("test"))
             {
-                using (asyncKeyedLocker.Lock("test", TimeSpan.Zero, out bool entered))
+                using (asyncKeyedLocker.Lock("test", 0, out bool entered))
                 {
                     Assert.False(entered);
                 }