[C#] Fix timing issue in testcase

microsoft · May 13, 2021 · 36de988 · 36de988
1 parent 79eaa00
commit 36de988
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cs/test/FasterLogRecoverReadOnlyTests.cs b/cs/test/FasterLogRecoverReadOnlyTests.cs
@@ -22,6 +22,7 @@ public class FasterLogRecoverReadOnlyTests
         string path;
         string deviceName;
         CancellationTokenSource cts;
+        SemaphoreSlim done;
 
         [SetUp]
         public void Setup()
@@ -31,6 +32,7 @@ public void Setup()
             if (Directory.Exists(path))
                 TestUtils.DeleteDirectory(path);
             cts = new CancellationTokenSource();
+            done = new SemaphoreSlim(0);
         }
 
         [TearDown]
@@ -61,7 +63,8 @@ private async Task ProducerAsync(FasterLog log, CancellationTokenSource cts)
                 log.RefreshUncommitted();
                 await Task.Delay(TimeSpan.FromMilliseconds(ProducerPauseMs));
             }
-            await Task.Delay(TimeSpan.FromMilliseconds(CommitPeriodMs * 4));
+            // Ensure the reader had time to see all data
+            await done.WaitAsync();
             cts.Cancel();
         }
 
@@ -99,6 +102,8 @@ private async Task ReadOnlyConsumerAsync(string deviceName, CancellationToken ca
                     Assert.AreEqual(prevValue + 1, value);
                     prevValue = value;
                     iter.CompleteUntil(nextAddress);
+                    if (prevValue == NumElements - 1)
+                        done.Release();
                 }
             } catch (OperationCanceledException) { }
             Assert.AreEqual(NumElements - 1, prevValue);