Search in sources :

Example 6 with PInput

use of org.apache.beam.sdk.values.PInput in project beam by apache.

the class DataflowPipelineJobTest method appliedPTransform.

private AppliedPTransform<?, ?, ?> appliedPTransform(String fullName, PTransform<PInput, POutput> transform, Pipeline p) {
    PInput input = mock(PInput.class);
    when(input.getPipeline()).thenReturn(p);
    return AppliedPTransform.of(fullName, Collections.<TupleTag<?>, PValue>emptyMap(), Collections.<TupleTag<?>, PValue>emptyMap(), transform, p);
}
Also used : PInput(org.apache.beam.sdk.values.PInput)

Aggregations

PInput (org.apache.beam.sdk.values.PInput)6 PValue (org.apache.beam.sdk.values.PValue)3 Test (org.junit.Test)2 Read (org.apache.beam.sdk.io.Read)1 ReplacementOutput (org.apache.beam.sdk.runners.PTransformOverrideFactory.ReplacementOutput)1 Node (org.apache.beam.sdk.runners.TransformHierarchy.Node)1 Create (org.apache.beam.sdk.transforms.Create)1 DoFn (org.apache.beam.sdk.transforms.DoFn)1 PTransform (org.apache.beam.sdk.transforms.PTransform)1 PDone (org.apache.beam.sdk.values.PDone)1 TaggedPValue (org.apache.beam.sdk.values.TaggedPValue)1