1 files changed, 163 insertions, 0 deletions
diff --git a/watchmaker/swing/src/java/test/org/uncommons/watchmaker/swing/ProbabilityParameterControlTest.java b/watchmaker/swing/src/java/test/org/uncommons/watchmaker/swing/ProbabilityParameterControlTest.java
new file mode 100644
index 0000000..1da22ea
--- /dev/null
+++ b/watchmaker/swing/src/java/test/org/uncommons/watchmaker/swing/ProbabilityParameterControlTest.java
@@ -0,0 +1,163 @@
+//=============================================================================
+// Copyright 2006-2010 Daniel W. Dyer
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//=============================================================================
+package org.uncommons.watchmaker.swing;
+
+import java.awt.BorderLayout;
+import javax.swing.JFrame;
+import javax.swing.JSlider;
+import org.fest.swing.core.BasicRobot;
+import org.fest.swing.core.Robot;
+import org.fest.swing.fixture.FrameFixture;
+import org.testng.annotations.AfterMethod;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+import org.uncommons.maths.random.Probability;
+
+/**
+ * Unit test for the {@link ProbabilityParameterControl} component.
+ * @author Daniel Dyer
+ */
+public class ProbabilityParameterControlTest
+{
+    private Robot robot;
+
+    @BeforeMethod(groups = "display-required")
+    public void prepare()
+    {
+        robot = BasicRobot.robotWithNewAwtHierarchy();
+    }
+
+
+    @AfterMethod(groups = "display-required")
+    public void cleanUp()
+    {
+        robot.cleanUp();
+        robot = null;
+    }
+
+    
+    @Test
+    public void testDefaultValue()
+    {
+        Probability defaultValue = new Probability(0.75d);
+        ProbabilityParameterControl control = new ProbabilityParameterControl(defaultValue);
+        assert control.getNumberGenerator().nextValue().equals(defaultValue) : "Wrong initial value.";
+    }
+
+
+    /**
+     * Initial value must not be less than the minimum.
+     */
+    @Test(expectedExceptions = IllegalArgumentException.class)
+    public void testDefaultValueTooLow()
+    {
+        new ProbabilityParameterControl(Probability.EVENS,
+                                        Probability.ONE,
+                                        2,
+                                        new Probability(0.45d)); // Should throw an IllegalArgumentException.
+    }
+
+
+    /**
+     * Initial value must not be less than the minimum.
+     */
+    @Test(expectedExceptions = IllegalArgumentException.class)
+    public void testDefaultValueTooHigh()
+    {
+        new ProbabilityParameterControl(Probability.ZERO,
+                                        Probability.EVENS,
+                                        2,
+                                        new Probability(0.55d)); // Should throw an IllegalArgumentException.
+    }
+
+
+    /**
+     * Minimum must be less than maximum.
+     */
+    @Test(expectedExceptions = IllegalArgumentException.class)
+    public void testMinimumHigherThanMaximum()
+    {
+        new ProbabilityParameterControl(new Probability(0.7d),
+                                        new Probability(0.6d),
+                                        2,
+                                        new Probability(0.7d)); // Should throw an IllegalArgumentException.
+    }
+
+
+    @Test(expectedExceptions = IllegalArgumentException.class)
+    public void testInvalidDecimalPlaces()
+    {
+        new ProbabilityParameterControl(Probability.ZERO,
+                                        Probability.ONE,
+                                        0, // Invalid, should trigger IllegalArgumentException.
+                                        Probability.EVENS);
+    }
+
+
+    @Test(dependsOnMethods = "testDefaultValue",
+          groups = "display-required")
+    public void testSlider()
+    {
+        Probability defaultValue = new Probability(0.75d);
+        ProbabilityParameterControl control = new ProbabilityParameterControl(defaultValue);
+
+        JFrame frame = new JFrame();
+        frame.add(control.getControl(), BorderLayout.CENTER);
+        FrameFixture frameFixture = new FrameFixture(robot, frame);
+        frame.setSize(300, 50);
+        frame.validate();
+        frame.setVisible(true);
+
+        JSlider slider = frameFixture.slider().component();
+        assert slider.getValue() == 75 : "Wrong slider position: " + slider.getValue();
+        String displayedValue = frameFixture.label().text();
+        assert displayedValue.equals("0.75") : "Wrong value displayed: " + displayedValue;
+
+
+        frameFixture.slider().slideTo(80); // 80 ticks is a probability of 0.8.
+        robot.waitForIdle();
+        double probability = control.getNumberGenerator().nextValue().doubleValue();
+        assert probability == 0.8 : "Wrong probability: " + probability;
+        displayedValue = frameFixture.label().text();
+        assert displayedValue.equals("0.80") : "Wrong value displayed: " + displayedValue;
+    }
+
+
+    @Test(dependsOnMethods = "testSlider",
+          groups = "display-required")
+    public void testReset()
+    {
+        Probability defaultValue = new Probability(0.75d);
+        ProbabilityParameterControl control = new ProbabilityParameterControl(defaultValue);
+
+        JFrame frame = new JFrame();
+        frame.add(control.getControl(), BorderLayout.CENTER);
+        FrameFixture frameFixture = new FrameFixture(robot, frame);
+        frame.setSize(300, 50);
+        frame.validate();
+        frame.setVisible(true);
+
+        JSlider slider = frameFixture.slider().component();
+        frameFixture.slider().slideTo(80); // 80 ticks is a probability of 0.8.
+
+        control.reset();
+        assert control.getNumberGenerator().nextValue().equals(defaultValue) : "NumberGenerator reset failed.";
+        assert slider.getValue() == 75 : "JSlider reset failed.";
+
+        String displayedValue = frameFixture.label().text();
+        assert displayedValue.equals("0.75") : "Wrong value displayed: " + displayedValue;
+    }
+}