private TestSinkStage(GraphStageWithMaterializedValue <SinkShape <T>, TMat> stageUnderTest, TestProbe probe)
{
_stageUnderTest = stageUnderTest;
_probe = probe;
Shape = new SinkShape <T>(_in);
}
private TestSourceStage(GraphStageWithMaterializedValue <SourceShape <T>, TMat> stageUnderTest, TestProbe probe)
{
_stageUnderTest = stageUnderTest;
_probe = probe;
Shape = new SourceShape <T>(_out);
}
public static TestSinkStage <T, TMat> Create(GraphStageWithMaterializedValue <SinkShape <T>, TMat> stageUnderTest,
TestProbe probe) => new TestSinkStage <T, TMat>(stageUnderTest, probe);
public static Source <T, TMat> Create(GraphStageWithMaterializedValue <SourceShape <T>, TMat> stageUnderTest,
TestProbe probe) => Source.FromGraph(new TestSourceStage <T, TMat>(stageUnderTest, probe));
