Extend TaskQueueTest

src/main/java/teetime/stage/IncStage.java

+package teetime.stage;
+
+import teetime.framework.AbstractStage;
+import teetime.framework.ITaskQueueDuplicable;
+import teetime.stage.basic.AbstractFilter;
+
+/**
+ * Created by nilsziermann on 06.01.17.
+ */
+public class IncStage extends AbstractFilter<Integer> implements ITaskQueueDuplicable {
+
+    @Override
+    protected void execute(Integer element) {
+        this.outputPort.send(element+1);
+    }
+
+    @Override
+    public AbstractStage duplicate() {
+        IncStage inc = new IncStage();
+        inc.getInputPort().setPipe(this.getInputPort().getPipe());
+        inc.getOutputPort().setPipe(this.getOutputPort().getPipe());
+        inc.setExceptionHandler(this.getExceptionListener());
+
+        return inc;
+    }
+}

src/test/java/teetime/examples/directory/TaskQueueTestConfiguration.java

@@ -25,10 +25,12 @@ public class TaskQueueTestConfiguration extends Configuration {
         final VaryingVariableWorkloadStage varyingVariableWorkloadStage = new VaryingVariableWorkloadStage();
         final VaryingVariableWorkloadStage varyingVariableWorkloadStage2 = new VaryingVariableWorkloadStage();
         final VaryingVariableWorkloadStage varyingVariableWorkloadStage3 = new VaryingVariableWorkloadStage();
+        final IncStage inc = new IncStage();
 
         connectPorts(init.getOutputPort(), varyingVariableWorkloadStage.getInputPort());
         connectPorts(varyingVariableWorkloadStage.getOutputPort(), varyingVariableWorkloadStage2.getInputPort());
-        connectPorts(varyingVariableWorkloadStage2.getOutputPort(), counter.getInputPort());
+        connectPorts(varyingVariableWorkloadStage2.getOutputPort(), inc.getInputPort());
+        connectPorts(inc.getOutputPort(), counter.getInputPort());
         connectPorts(counter.getOutputPort(), collectorSink.getInputPort());
     }
 }

src/test/java/teetime/framework/TaskQueueTest.java

@@ -50,9 +50,12 @@ public class TaskQueueTest {
         final List<Integer> output = configuration.collectorSink.getElements();
 
         int last = Integer.MIN_VALUE;
-        for(Integer element : output) {
-            Assert.assertTrue(element > last);
-            last = element;
+
+        int n = output.size();
+        for(int i = 0; i < n; i++) {
+            int test = output.remove(0);
+            System.out.println(test);
+            Assert.assertEquals(i+1, test );
         }
     }
 }