Skip to content

Commit 29b56ab

Browse files
authored
[3-74]BidiFlowSpec: Change AssertAllStagesStopped to AssertAllStagesStoppedAsync (#6542)
1 parent 29da910 commit 29b56ab

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

src/core/Akka.Streams.Tests/Dsl/BidiFlowSpec.cs

+15-13
Original file line numberDiff line numberDiff line change
@@ -162,19 +162,20 @@ public void A_BidiFlow_must_materialize_its_value()
162162
}
163163

164164
[Fact]
165-
public void A_BidiFlow_must_combine_materialization_values()
165+
public async Task A_BidiFlow_must_combine_materialization_values()
166166
{
167-
this.AssertAllStagesStopped(() =>
168-
{
169-
var left = Flow.FromGraph(GraphDsl.Create(Sink.First<int>(), (b, sink) =>
170-
{
171-
var broadcast = b.Add(new Broadcast<int>(2));
172-
var merge = b.Add(new Merge<int>(2));
173-
var flow = b.Add(Flow.Create<string>().Select(int.Parse));
174-
b.From(broadcast).To(sink);
175-
b.From(Source.Single(1).MapMaterializedValue(_ => Task.FromResult(0))).Via(broadcast).To(merge);
176-
b.From(flow).To(merge);
177-
return new FlowShape<string, int>(flow.Inlet, merge.Out);
167+
await this.AssertAllStagesStoppedAsync(() => {
168+
var left =
169+
Flow.FromGraph(GraphDsl.Create(Sink.First<int>(),
170+
(b, sink) =>
171+
{
172+
var broadcast = b.Add(new Broadcast<int>(2));
173+
var merge = b.Add(new Merge<int>(2));
174+
var flow = b.Add(Flow.Create<string>().Select(int.Parse));
175+
b.From(broadcast).To(sink);
176+
b.From(Source.Single(1).MapMaterializedValue(_ => Task.FromResult(0))).Via(broadcast).To(merge);
177+
b.From(flow).To(merge);
178+
return new FlowShape<string, int>(flow.Inlet, merge.Out);
178179
}));
179180

180181
var right = Flow.FromGraph(GraphDsl.Create(Sink.First<List<long>>(), (b, sink) =>
@@ -197,7 +198,8 @@ public void A_BidiFlow_must_combine_materialization_values()
197198
Task.WhenAll(l, m, r).Wait(TimeSpan.FromSeconds(3)).Should().BeTrue();
198199
l.Result.Should().Be(1);
199200
m.Result.Should().Be(42);
200-
r.Result.Should().BeEquivalentTo(new [] {3L, 12L});
201+
r.Result.Should().BeEquivalentTo(new[] { 3L, 12L });
202+
return Task.CompletedTask;
201203
}, Materializer);
202204
}
203205

0 commit comments

Comments
 (0)