[69-74] KeepGoingStageSpec (#6617)

* [69-74] `KeepGoingStageSpec`

* Changes

---------

Co-authored-by: Aaron Stannard <[email protected]>

Author: eaba

src/core/Akka.Streams.Tests/Implementation/Fusing/KeepGoingStageSpec.cs

@@ -200,143 +200,137 @@ public KeepGoingStageSpec(ITestOutputHelper helper = null) : base(helper)
         }
 
         [Fact]
-        public void A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_explicity_closed()
+        public async Task A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_explicity_closed()
         {
-            this.AssertAllStagesStopped(() =>
-            {
+            await this.AssertAllStagesStoppedAsync(async() => {
                 var t = Source.Maybe<int>().ToMaterialized(new PingableSink(true), Keep.Both).Run(Materializer);
                 var maybePromise = t.Item1;
                 var pingerFuture = t.Item2;
                 pingerFuture.Wait(TimeSpan.FromSeconds(3)).Should().BeTrue();
-                var pinger = pingerFuture.Result;
+                var pinger = await pingerFuture;
 
                 pinger.Register(TestActor);
 
                 //Before completion
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 maybePromise.TrySetResult(0);
-                ExpectMsg<UpstreamCompleted>();
+                await ExpectMsgAsync<UpstreamCompleted>();
 
-                ExpectNoMsg(200);
+                await ExpectNoMsgAsync(200);
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Stop();
                 // PostStop should not be concurrent with the event handler. This event here tests this.
-                ExpectMsg<EndOfEventHandler>();
-                ExpectMsg<PostStop>();
+                await ExpectMsgAsync<EndOfEventHandler>();
+                await ExpectMsgAsync<PostStop>();
             }, Materializer);
         }
 
         [Fact]
-        public void A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_explicitly_failed()
+        public async Task A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_explicitly_failed()
         {
-            this.AssertAllStagesStopped(() =>
-            {
+            await this.AssertAllStagesStoppedAsync(async() => {
                 var t = Source.Maybe<int>().ToMaterialized(new PingableSink(true), Keep.Both).Run(Materializer);
                 var maybePromise = t.Item1;
                 var pingerFuture = t.Item2;
                 pingerFuture.Wait(TimeSpan.FromSeconds(3)).Should().BeTrue();
-                var pinger = pingerFuture.Result;
+                var pinger = await pingerFuture;
 
                 pinger.Register(TestActor);
 
                 //Before completion
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 maybePromise.TrySetResult(0);
-                ExpectMsg<UpstreamCompleted>();
+                await ExpectMsgAsync<UpstreamCompleted>();
 
-                ExpectNoMsg(200);
+                await ExpectNoMsgAsync(200);
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Fail();
                 // PostStop should not be concurrent with the event handler. This event here tests this.
-                ExpectMsg<EndOfEventHandler>();
-                ExpectMsg<PostStop>();
-
+                await ExpectMsgAsync<EndOfEventHandler>();
+                await ExpectMsgAsync<PostStop>();
             }, Materializer);
         }
 
         [Fact]
-        public void A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_implicity_failed_via_exception()
+        public async Task A_stage_with_keep_going_must_still_be_alive_after_all_ports_have_been_closed_until_implicity_failed_via_exception()
         {
-            this.AssertAllStagesStopped(() =>
-            {
+            await this.AssertAllStagesStoppedAsync(async() => {
                 var t = Source.Maybe<int>().ToMaterialized(new PingableSink(true), Keep.Both).Run(Materializer);
                 var maybePromise = t.Item1;
                 var pingerFuture = t.Item2;
                 pingerFuture.Wait(TimeSpan.FromSeconds(3)).Should().BeTrue();
-                var pinger = pingerFuture.Result;
+                var pinger = await pingerFuture;
 
                 pinger.Register(TestActor);
 
                 //Before completion
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 maybePromise.TrySetResult(0);
-                ExpectMsg<UpstreamCompleted>();
+                await ExpectMsgAsync<UpstreamCompleted>();
 
                 ExpectNoMsg(200);
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 // We need to catch the exception otherwise the test fails
                 // ReSharper disable once EmptyGeneralCatchClause
-                try { pinger.ThrowEx();} catch { }
+                try { pinger.ThrowEx(); } catch { }
                 // PostStop should not be concurrent with the event handler. This event here tests this.
-                ExpectMsg<EndOfEventHandler>();
-                ExpectMsg<PostStop>();
-
+                await ExpectMsgAsync<EndOfEventHandler>();
+                await ExpectMsgAsync<PostStop>();
             }, Materializer);
         }
 
         [Fact]
-        public void A_stage_with_keep_going_must_close_down_earls_if_keepAlive_is_not_requested()
+        public async Task A_stage_with_keep_going_must_close_down_earls_if_keepAlive_is_not_requested()
         {
-            this.AssertAllStagesStopped(() =>
-            {
+            await this.AssertAllStagesStoppedAsync(async() => {
                 var t = Source.Maybe<int>().ToMaterialized(new PingableSink(false), Keep.Both).Run(Materializer);
                 var maybePromise = t.Item1;
                 var pingerFuture = t.Item2;
                 pingerFuture.Wait(TimeSpan.FromSeconds(3)).Should().BeTrue();
-                var pinger = pingerFuture.Result;
+                var pinger = await pingerFuture;
 
                 pinger.Register(TestActor);
 
                 //Before completion
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 pinger.Ping();
-                ExpectMsg<Pong>();
+                await ExpectMsgAsync<Pong>();
 
                 maybePromise.TrySetResult(0);
-                ExpectMsg<UpstreamCompleted>();
-                ExpectMsg<PostStop>();
+                await ExpectMsgAsync<UpstreamCompleted>();
+                await ExpectMsgAsync<PostStop>();
             }, Materializer);
         }
     }