package com.acuitra.pipeline;
import java.util.LinkedList;
import org.junit.Assert;
import org.junit.Test;
public class PipelineTest {
@Test
public void test() {
Pipeline<LinkedList<Integer>, String> pipeline = new Pipeline<>();
pipeline.addStage(new TestStage(1));
pipeline.addStage(new TestStage(2));
pipeline.addStage(new TestStage(3));
pipeline.addStage(new TestStage(4));
pipeline.addStage(new TestStage(5));
Context<LinkedList<Integer>, String> context = new Context<>();
context.setInput(new LinkedList<Integer>());
pipeline.execute(context);
LinkedList<Integer> stageCounts = context.getInput();
for (int i = 0; i < stageCounts.size(); i++) {
Assert.assertEquals(i + 1, stageCounts.get(i).intValue());
}
}
}