Fix race condition in TypicalLoop_WithLogsQueueTrigger

DataDog · Jan 8, 2025 · 0e0de0e · 0e0de0e
1 parent c0655f6
commit 0e0de0e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tracer/test/Datadog.Trace.Tests/Telemetry/TelemetryControllerSchedulerTests.cs b/tracer/test/Datadog.Trace.Tests/Telemetry/TelemetryControllerSchedulerTests.cs
@@ -1,4 +1,4 @@
-// <copyright file="TelemetryControllerSchedulerTests.cs" company="Datadog">
+// <copyright file="TelemetryControllerSchedulerTests.cs" company="Datadog">
 // Unless explicitly stated otherwise all files in this repository are licensed under the Apache 2 License.
 // This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2017 Datadog, Inc.
 // </copyright>
@@ -162,6 +162,8 @@ public async Task TypicalLoop_WithLogsQueueTrigger()
         delayMutex.Wait();
         queueTcs.SetResult(true); // this triggers the queue task
 
+        await waitTask;
+
         // t = 15s;
         _scheduler.ShouldFlushTelemetry.Should().BeFalse(); // not a complete interval
         _scheduler.ShouldFlushRedactedErrorLogs.Should().BeTrue(); // triggered by queue
@@ -180,7 +182,7 @@ public async Task TypicalLoop_WithLogsQueueTrigger()
         // t = 125s;
         _scheduler.ShouldFlushTelemetry.Should().BeTrue();
         _scheduler.ShouldFlushRedactedErrorLogs.Should().BeTrue();
-     }
+    }
 
     [Fact]
     public async Task DoesNotFlushTelemetryUntilInitialized()