From 152079de6b1d3fb195acd800a40f4f973e3d5d0c Mon Sep 17 00:00:00 2001 From: Christian Wulf <chw@informatik.uni-kiel.de> Date: Mon, 6 Jul 2015 15:01:04 +0200 Subject: [PATCH] refactored DynamicDistributorTest --- .../dynamic/DynamicDistributorTest.java | 22 +++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/src/test/java/teetime/stage/basic/distributor/dynamic/DynamicDistributorTest.java b/src/test/java/teetime/stage/basic/distributor/dynamic/DynamicDistributorTest.java index c62002f7..c252c8c0 100644 --- a/src/test/java/teetime/stage/basic/distributor/dynamic/DynamicDistributorTest.java +++ b/src/test/java/teetime/stage/basic/distributor/dynamic/DynamicDistributorTest.java @@ -71,11 +71,11 @@ public class DynamicDistributorTest { analysis.executeBlocking(); assertThat(config.getOutputElements(), contains(0)); - assertValuesForIndex(inputActions, Arrays.asList(1), 0); - assertValuesForIndex(inputActions, Arrays.asList(2), 1); - assertValuesForIndex(inputActions, Arrays.asList(3), 2); - assertValuesForIndex(inputActions, Arrays.asList(4), 3); - assertValuesForIndex(inputActions, Collections.<Integer> emptyList(), 4); + assertValuesForIndex(inputActions[0], Arrays.asList(1)); + assertValuesForIndex(inputActions[1], Arrays.asList(2)); + assertValuesForIndex(inputActions[2], Arrays.asList(3)); + assertValuesForIndex(inputActions[3], Arrays.asList(4)); + assertValuesForIndex(inputActions[4], Collections.<Integer> emptyList()); } @Test @@ -103,9 +103,9 @@ public class DynamicDistributorTest { analysis.executeBlocking(); assertThat(config.getOutputElements(), contains(0, 1, 2, 4, 5)); - assertValuesForIndex(inputActions, Collections.<Integer> emptyList(), 0); - assertValuesForIndex(inputActions, Arrays.asList(3), 2); - assertValuesForIndex(inputActions, Collections.<Integer> emptyList(), 3); + assertValuesForIndex(inputActions[0], Collections.<Integer> emptyList()); + assertValuesForIndex(inputActions[2], Arrays.asList(3)); + assertValuesForIndex(inputActions[3], Collections.<Integer> emptyList()); } private CreatePortAction<Integer> createPortCreateAction(final PortContainer<Integer> portContainer) { @@ -120,12 +120,12 @@ public class DynamicDistributorTest { return portAction; } - private void assertValuesForIndex(final PortAction<DynamicDistributor<Integer>>[] inputActions, - final List<Integer> values, final int index) { - PortAction<DynamicDistributor<Integer>> ia = inputActions[index]; + private void assertValuesForIndex(final PortAction<DynamicDistributor<Integer>> ia, final List<Integer> values) { Stage stage = ((CreatePortAction<Integer>) ia).getInputPort().getOwningStage(); + @SuppressWarnings("unchecked") CollectorSink<Integer> collectorSink = (CollectorSink<Integer>) stage; + assertThat(collectorSink.getElements(), is(values)); } -- GitLab