From eb3c8495629451c037de3a91b02b695736da55fa Mon Sep 17 00:00:00 2001
From: Nelson Tavares de Sousa <ntd@informatik.uni-kiel.de>
Date: Tue, 21 Apr 2015 16:46:59 +0200
Subject: [PATCH] reactived test case and corrected ObjectProducer's behavior
---
src/main/java/teetime/stage/ObjectProducer.java | 11 ++++-------
src/test/java/teetime/stage/ObjectProducerTest.java | 1 +
2 files changed, 5 insertions(+), 7 deletions(-)
diff --git a/src/main/java/teetime/stage/ObjectProducer.java b/src/main/java/teetime/stage/ObjectProducer.java
index 5e0c272a..a5267aed 100644
--- a/src/main/java/teetime/stage/ObjectProducer.java
+++ b/src/main/java/teetime/stage/ObjectProducer.java
@@ -54,14 +54,11 @@ public final class ObjectProducer<T> extends AbstractProducerStage<T> {
@Override
protected void execute() {
- T newObject = this.inputObjectCreator.create();
- this.numInputObjects--;
-
- outputPort.send(newObject);
-
- if (this.numInputObjects == 0) {
- this.terminate();
+ for (int i = 0; i < numInputObjects; i++) {
+ T newObject = this.inputObjectCreator.create();
+ outputPort.send(newObject);
}
+ this.terminate();
}
}
diff --git a/src/test/java/teetime/stage/ObjectProducerTest.java b/src/test/java/teetime/stage/ObjectProducerTest.java
index 242060a6..fc909882 100644
--- a/src/test/java/teetime/stage/ObjectProducerTest.java
+++ b/src/test/java/teetime/stage/ObjectProducerTest.java
@@ -43,6 +43,7 @@ public class ObjectProducerTest {
assertThat(results, contains(1, 2, 3));
}
+ @Test
public void producerShouldSendNothingIfSpecified() {
final List<Integer> results = new ArrayList<Integer>();
final ObjectProducer<Integer> producer = new ObjectProducer<Integer>(0, new Generator());
--
GitLab