Update HostTests.cs (#51495)

dotnet · Apr 19, 2021 · 4bcc1c1 · 4bcc1c1
1 parent db3c25d
commit 4bcc1c1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/libraries/Microsoft.Extensions.Hosting/tests/UnitTests/Internal/HostTests.cs b/src/libraries/Microsoft.Extensions.Hosting/tests/UnitTests/Internal/HostTests.cs
@@ -710,11 +710,12 @@ public async Task HostStopsApplicationWithOneBackgroundServiceErrorAndOthersWith
             var wasStoppingCalled = false;
             lifetime.ApplicationStopping.Register(() =>
             {
-                otherTcs.SetResult(true);
                 wasStoppingCalled = true;
+                otherTcs.SetResult(true);
             });
 
-            await host.StartAsync();
+            // Ensure all completions have been signaled before continuing
+            await Task.WhenAll(host.StartAsync(), throwingTcs.Task, otherTcs.Task);
 
             Assert.True(wasStartedCalled);
             Assert.True(wasStoppingCalled);