Fix test

microsoft · Jun 22, 2020 · 303cee8 · 303cee8
1 parent e635ec5
commit 303cee8
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...rossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs b/...rossPlatEngine.UnitTests/MultiTestRunFinalization/MultiTestRunFinalizationManagerTests.cs
@@ -533,6 +533,18 @@ public async Task FinalizeMultiTestRunAsync_ShouldReturnProperlySendProgressEven
                 new AttachmentSet(new Uri(uri2), "uri2_output")
             };
 
+            var innerTaskCompletionSource = new TaskCompletionSource<object>();
+
+            int counter = 0;
+            mockEventsHandler.Setup(h => h.HandleMultiTestRunFinalizationProgress(It.IsAny<MultiTestRunFinalizationProgressEventArgs>())).Callback(() => 
+            { 
+                counter++;
+                if(counter == 6)
+                {
+                    innerTaskCompletionSource.TrySetResult(null);
+                }
+            });
+
             mockAttachmentHandler1.Setup(h => h.HandleDataCollectionAttachmentSets(It.IsAny<ICollection<AttachmentSet>>(), It.IsAny<IProgress<int>>(), It.IsAny<IMessageLogger>(), It.IsAny<CancellationToken>())).Returns((ICollection<AttachmentSet> i1, IProgress<int> progress, IMessageLogger logger, CancellationToken cancellation) =>
             {
                 progress.Report(25);
@@ -555,6 +567,7 @@ public async Task FinalizeMultiTestRunAsync_ShouldReturnProperlySendProgressEven
             await manager.FinalizeMultiTestRunAsync(mockRequestData.Object, inputAttachments, mockEventsHandler.Object, CancellationToken.None);
 
             // assert
+            await innerTaskCompletionSource.Task;
             VerifyCompleteEvent(false, false, outputAttachments1[0], outputAttachments2[0]);
             mockEventsHandler.Verify(h => h.HandleMultiTestRunFinalizationProgress(It.IsAny<MultiTestRunFinalizationProgressEventArgs>()), Times.Exactly(6));
             mockEventsHandler.Verify(h => h.HandleMultiTestRunFinalizationProgress(It.Is<MultiTestRunFinalizationProgressEventArgs>(a => VerifyProgressArgsForTwoHandlers(a, 1, 25, uri1))));