From be9537ba5109bfe993ac913717f700725555f0fe Mon Sep 17 00:00:00 2001
From: Jakub Chocholowicz <jakub.chocholowicz@gmail.com>
Date: Tue, 16 Jun 2020 23:04:08 +0200
Subject: [PATCH] fix next test

---
 .../MultiTestRunFinalizationManagerTests.cs                | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/Microsoft.TestPlatform.CrossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs b/test/Microsoft.TestPlatform.CrossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs
index 094d26fa05..ae911a6feb 100644
--- a/test/Microsoft.TestPlatform.CrossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs
+++ b/test/Microsoft.TestPlatform.CrossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs
@@ -452,14 +452,15 @@ public async Task FinalizeMultiTestRunAsync_ShouldReturnNull_IfOperationCancelle
             {
                 for (int i = 0; i < 1000; ++i)
                 {
-                    Task.Delay(100);
+                    Task.Delay(100).Wait();
                     Console.WriteLine($"Iteration: {i}");
 
-                    if (cancellation.IsCancellationRequested) break;
+                    cancellation.ThrowIfCancellationRequested();
 
                     if (i == 3)
-                    {
+                    {                        
                         cancellationTokenSource.Cancel();
+                        Task.Delay(500).Wait();
                     }
                 }