verify OnRunCompleted waits for pipeline to complete

bilal-fazlani · Nov 26, 2023 · a64a72f · a64a72f
1 parent 6b4bb01
commit a64a72f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/CommandDotNet.Tests/FeatureTests/OnRunCompletedTests.cs b/CommandDotNet.Tests/FeatureTests/OnRunCompletedTests.cs
@@ -1,3 +1,4 @@
+using System.Threading.Tasks;
 using FluentAssertions;
 using Xunit;
 using Xunit.Abstractions;
@@ -18,16 +19,24 @@ public void OnRunCompleted_IsCalled_AfterPipelineIsExecuted()
             new AppRunner<App>()
                 .Configure(b => b.OnRunCompleted += args =>
                 {
+                    // ensure OnRunCompleted waits for the pipeline to complete
+                    App.Executed.Should().BeTrue();
                     wasCalled = true;
                 })
-                .RunInMem("");
+                .RunInMem("Do");
 
             wasCalled.Should().BeTrue();
         }
 
         private class App
         {
-            public void Do() { }
+            public static bool Executed { get; set; }
+
+            public void Do()
+            {
+                Task.Delay(1000);
+                Executed = true;
+            }
         }
     }
 }