Fix racing condition

microsoft · Oct 25, 2023 · 39ac400 · 39ac400
1 parent 0dc3a16
commit 39ac400
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/code/DeltaKustoAdxIntegrationTest/AdxDbTestHelper.cs b/code/DeltaKustoAdxIntegrationTest/AdxDbTestHelper.cs
@@ -116,6 +116,7 @@ public async Task<IImmutableList<string>> GetDbsAsync(int dbCount)
                 }
                 //  Not enough db available:  let's wait for some!
                 await newDbEvent.Task;
+                //  Change the event for a fresh one
                 Interlocked.CompareExchange(
                     ref _newDbEvent,
                     new TaskCompletionSource(),
@@ -142,7 +143,7 @@ public void ReleaseDbs(IEnumerable<string> dbNames)
                 _availableDbNames.AddRange(dbNames);
             }
             //  Pop event for waiting threads
-            _newDbEvent.SetResult();
+            _newDbEvent.TrySetResult();
         }
 
         void IDisposable.Dispose()