[64-74] SourceSpec (#6612)

akkadotnet · Mar 27, 2023 · 5e2b211 · 5e2b211
1 parent fab1288
commit 5e2b211
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/src/core/Akka.Streams.Tests/Dsl/SourceSpec.cs b/src/core/Akka.Streams.Tests/Dsl/SourceSpec.cs
@@ -92,9 +92,9 @@ public void Failed_Source_must_emit_error_immediately()
         }
 
         [Fact]
-        public void Maybe_Source_must_complete_materialized_future_with_None_when_stream_cancels()
+        public async Task Maybe_Source_must_complete_materialized_future_with_None_when_stream_cancels()
         {
-            this.AssertAllStagesStopped(async() =>
+            await this.AssertAllStagesStoppedAsync(async() =>
             {
                 var neverSource = Source.Maybe<object>();
                 var pubSink = Sink.AsPublisher<object>(false);
@@ -117,9 +117,9 @@ public void Maybe_Source_must_complete_materialized_future_with_None_when_stream
         }
 
         [Fact]
-        public void Maybe_Source_must_allow_external_triggering_of_empty_completion()
+        public async Task Maybe_Source_must_allow_external_triggering_of_empty_completion()
         {
-            this.AssertAllStagesStopped(async() =>
+            await this.AssertAllStagesStoppedAsync(async() =>
             {
                 var neverSource = Source.Maybe<int>().Where(_ => false);
                 var counterSink = Sink.Aggregate<int, int>(0, (acc, _) => acc + 1);
@@ -137,9 +137,9 @@ public void Maybe_Source_must_allow_external_triggering_of_empty_completion()
         }
 
         [Fact]
-        public void Maybe_Source_must_allow_external_triggering_of_non_empty_completion()
+        public async Task Maybe_Source_must_allow_external_triggering_of_non_empty_completion()
         {
-            this.AssertAllStagesStopped(async() =>
+            await this.AssertAllStagesStoppedAsync(async() =>
             {
                 var neverSource = Source.Maybe<int>();
                 var counterSink = Sink.First<int>();
@@ -156,10 +156,9 @@ public void Maybe_Source_must_allow_external_triggering_of_non_empty_completion(
         }
 
         [Fact]
-        public void Maybe_Source_must_allow_external_triggering_of_OnError()
+        public async Task Maybe_Source_must_allow_external_triggering_of_OnError()
         {
-            this.AssertAllStagesStopped(() =>
-            {
+            await this.AssertAllStagesStoppedAsync(() => {
                 var neverSource = Source.Maybe<int>();
                 var counterSink = Sink.First<int>();
 
@@ -172,6 +171,7 @@ public void Maybe_Source_must_allow_external_triggering_of_OnError()
 
                 counterFuture.Invoking(f => f.Wait(TimeSpan.FromSeconds(3))).Should().Throw<Exception>()
                     .WithMessage("Boom");
+                return Task.CompletedTask;
             }, Materializer);
         }