@Test
@Category(NeedsRunner.class)
public void testReadToImpulseOverride() {
BoundedSource<Long> source = CountingSource.upTo(10L);
PCollection<Long> input = p.apply(Read.from(source));
PAssert.that(input).containsInAnyOrder(0L, 1L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9L);
p.replaceAll(Collections.singletonList(JavaReadViaImpulse.boundedOverride()));
p.traverseTopologically(
new Pipeline.PipelineVisitor() {
@Override
public void enterPipeline(Pipeline p) {}
@Override
public CompositeBehavior enterCompositeTransform(TransformHierarchy.Node node) {
assertNotReadTransform(node.getTransform());
return CompositeBehavior.ENTER_TRANSFORM;
}
@Override
public void leaveCompositeTransform(TransformHierarchy.Node node) {}
@Override
public void visitPrimitiveTransform(TransformHierarchy.Node node) {
assertNotReadTransform(node.getTransform());
}
@Override
public void visitValue(PValue value, TransformHierarchy.Node producer) {}
@Override
public void leavePipeline(Pipeline pipeline) {}
});
p.run();
}