Examples with CompileNode org.talend.dataprep.transformation.pipeline.node.CompileNode used on opensource projects

Example 6 with CompileNode

use of org.talend.dataprep.transformation.pipeline.node.CompileNode in project data-prep by Talend.

the class StepNodeTransformerTest method shouldCreateStepNodesWhenSurrounded.

@Test
public void shouldCreateStepNodesWhenSurrounded() throws Exception {
    // given
    Node node = // 
    NodeBuilder.from(// 
    new TestNode()).to(// 
    new CompileNode(null, null)).to(// 
    new ActionNode(null, null)).to(// 
    new BasicNode()).to(// 
    new CompileNode(null, null)).to(// 
    new ActionNode(null, null)).build();
    // when
    final Node processed = StepNodeTransformer.transform(node, asList(ROOT, STEP, STEP), s -> null);
    // then
    final AtomicInteger stepNodeCount = new AtomicInteger();
    processed.accept(new Visitor() {

        @Override
        public void visitStepNode(StepNode stepNode) {
            stepNodeCount.incrementAndGet();
            super.visitStepNode(stepNode);
        }
    });
    assertEquals(2, stepNodeCount.get());
}