From af2976c2cba4d40f3f08d9b8ba3c6def8295e0b9 Mon Sep 17 00:00:00 2001
From: Nelson Tavares de Sousa <stu103017@mail.uni-kiel.de>
Date: Wed, 20 May 2015 15:35:44 +0200
Subject: [PATCH] added test for roundrobin2

---
 .../basic/distributor/DistributorTest.java    | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/test/java/teetime/stage/basic/distributor/DistributorTest.java b/src/test/java/teetime/stage/basic/distributor/DistributorTest.java
index ecdeb41c..3d637d25 100644
--- a/src/test/java/teetime/stage/basic/distributor/DistributorTest.java
+++ b/src/test/java/teetime/stage/basic/distributor/DistributorTest.java
@@ -74,6 +74,29 @@ public class DistributorTest {
 		assertThat(this.secondIntegers, is(empty()));
 	}
 
+	@Test
+	public void roundRobin2ShouldWork() {
+		distributor.setStrategy(new RoundRobinStrategy2());
+
+		test(distributor).and().send(1, 2, 3, 4, 5).to(distributor.getInputPort()).and().receive(firstIntegers).from(distributor.getNewOutputPort()).and()
+				.receive(secondIntegers).from(distributor.getNewOutputPort()).start();
+
+		assertThat(this.firstIntegers, contains(1, 3, 5));
+		assertThat(this.secondIntegers, contains(2, 4));
+	}
+
+	@Test
+	public void singleElementRoundRobin2ShouldWork() {
+		distributor.setStrategy(new RoundRobinStrategy2());
+
+		test(distributor).and().send(1).to(distributor.getInputPort()).and().receive(firstIntegers).from(distributor.getNewOutputPort()).and()
+				.receive(secondIntegers)
+				.from(distributor.getNewOutputPort()).start();
+
+		assertThat(this.firstIntegers, contains(1));
+		assertThat(this.secondIntegers, is(empty()));
+	}
+
 	@Test
 	public void copyByReferenceShouldWork() {
 		distributor.setStrategy(new CopyByReferenceStrategy());
-- 
GitLab