From 0335ac60a6e81645e619cc8cb72355e34c3e8b7c Mon Sep 17 00:00:00 2001
From: Safia Abdalla <safia@microsoft.com>
Date: Mon, 3 Aug 2020 13:40:29 -0700
Subject: [PATCH] Check completed task in circuit termination tests (#24450)

* Check completed task in circuit termination tests
* Remove extraneous whitespace
* Update CircuitGracefulTerminationTests.cs
---
 .../ServerExecutionTests/CircuitGracefulTerminationTests.cs    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Components/test/E2ETest/ServerExecutionTests/CircuitGracefulTerminationTests.cs b/src/Components/test/E2ETest/ServerExecutionTests/CircuitGracefulTerminationTests.cs
index 8183142196..69c20cb970 100644
--- a/src/Components/test/E2ETest/ServerExecutionTests/CircuitGracefulTerminationTests.cs
+++ b/src/Components/test/E2ETest/ServerExecutionTests/CircuitGracefulTerminationTests.cs
@@ -83,9 +83,10 @@ namespace Microsoft.AspNetCore.Components.E2ETest.ServerExecutionTests
         {
             // Arrange & Act
             Browser.Navigate().GoToUrl("about:blank");
-            await Task.WhenAny(Task.Delay(10000), GracefulDisconnectCompletionSource.Task);
+            var task = await Task.WhenAny(Task.Delay(10000), GracefulDisconnectCompletionSource.Task);
 
             // Assert
+            Assert.Equal(GracefulDisconnectCompletionSource.Task, task);
             Assert.Contains((Extensions.Logging.LogLevel.Debug, "CircuitTerminatedGracefully"), Messages);
             Assert.Contains((Extensions.Logging.LogLevel.Debug, "CircuitDisconnectedPermanently"), Messages);
         }