improve test coverage

MarkCiliaVincenti · Nov 10, 2024 · 82bcc4e · 82bcc4e
1 parent 0386195
commit 82bcc4e
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/AsyncKeyedLock.Tests/AsyncKeyedLocker/OriginalTests.cs b/AsyncKeyedLock.Tests/AsyncKeyedLocker/OriginalTests.cs
@@ -490,6 +490,22 @@ public async Task TestTimeoutOrNullBasic()
             Assert.False(asyncKeyedLocker.IsInUse("test"));
         }
 
+        [Fact]
+        public void TestTimeoutBasicWithOutParameter()
+        {
+            var asyncKeyedLocker = new AsyncKeyedLocker<string>(o => { o.PoolSize = 0; });
+            using (var myLock = asyncKeyedLocker.Lock("test", 0, out var entered))
+            {
+                Assert.True(entered);
+                Assert.True(asyncKeyedLocker.IsInUse("test"));
+                asyncKeyedLocker.Lock("test", 0, out entered);
+                Assert.False(entered);
+                asyncKeyedLocker.Lock("test", TimeSpan.Zero, out entered);
+                Assert.False(entered);
+            }
+            Assert.False(asyncKeyedLocker.IsInUse("test"));
+        }
+
         [Fact]
         public void TestTimeoutOrNullBasicWithOutParameter()
         {
@@ -498,6 +514,10 @@ public void TestTimeoutOrNullBasicWithOutParameter()
             {
                 Assert.NotNull(myLock);
                 Assert.True(asyncKeyedLocker.IsInUse("test"));
+                var entered = asyncKeyedLocker.LockOrNull("test", 0);
+                Assert.Null(entered);
+                entered = asyncKeyedLocker.LockOrNull("test", TimeSpan.Zero);
+                Assert.Null(entered);
             }
             Assert.False(asyncKeyedLocker.IsInUse("test"));
         }