Unit test PeriodicScheduler

Author: James Hagborg

src/main/java/org/usfirst/frc/team69/util/PeriodicScheduler.java

@@ -50,6 +50,13 @@ public synchronized void addEvent(Runnable event) {
         events.add(event);
     }
 
+    /**
+     * Remove all scheduled tasks.  This is mainly useful for testing.
+     */
+    public synchronized void clear() {
+        events.clear();
+    }
+    
     /**
      * Run all of the events which have been added to the scheduler.  If you
      * are using {@link HYPERRobot}, you do not need to call this manually.

src/test/java/org/usfirst/frc/team69/util/oi/test/MockPreferencesListener.java

@@ -0,0 +1,18 @@
+package org.usfirst.frc.team69.util.oi.test;
+
+import org.usfirst.frc.team69.util.pref.PreferencesListener;
+
+public class MockPreferencesListener implements PreferencesListener {
+
+    private int m_count;
+    
+    public int getCount() {
+        return m_count;
+    }
+    
+    @Override
+    public void onPreferencesUpdated() {
+        m_count++;
+    }
+
+}

src/test/java/org/usfirst/frc/team69/util/oi/test/MockRunnable.java

@@ -0,0 +1,16 @@
+package org.usfirst.frc.team69.util.oi.test;
+
+public class MockRunnable implements Runnable {
+
+    private int m_count = 0;
+    
+    public int getCount() {
+        return m_count;
+    }
+    
+    @Override
+    public void run() {
+        m_count++;
+    }
+
+}

src/test/java/org/usfirst/frc/team69/util/oi/test/PeroidicSchedulerTest.java

@@ -0,0 +1,83 @@
+package org.usfirst.frc.team69.util.oi.test;
+
+import static org.junit.Assert.*;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.usfirst.frc.team69.util.PeriodicScheduler;
+
+import edu.wpi.first.wpilibj.UnitTestUtility;
+
+public class PeroidicSchedulerTest {
+    
+    private PeriodicScheduler m_sched;
+    
+    @BeforeClass
+    public static void setUpBeforeClass() throws Exception {
+        UnitTestUtility.setupMockBase();
+    }
+
+    @Before
+    public void setUp() {
+        m_sched = PeriodicScheduler.getInstance();
+    }
+    
+    @After
+    public void tearDown() {
+        m_sched.clear();
+    }
+    
+    @Test
+    public void testClearOnEmpty() {
+        // just check that nothing crashes
+        m_sched.clear();
+        m_sched.run();
+    }
+    
+    @Test
+    public void testClearWithElement() {
+        MockRunnable event = new MockRunnable();
+        m_sched.addEvent(event);
+        m_sched.clear();
+        m_sched.run();
+        assertEquals(0, event.getCount());
+    }
+    
+    @Test
+    public void testRunOnEmpty() {
+        // just check nothing crashes
+        m_sched.run();
+    }
+    
+    @Test
+    public void testRunEvent() {
+        MockRunnable event = new MockRunnable();
+        m_sched.addEvent(event);
+        m_sched.run();
+        assertEquals(1, event.getCount());
+    }
+    
+    @Test
+    public void testAddFromEvent() {
+        MockRunnable event = new MockRunnable();
+        
+        Runnable adder = new Runnable() {
+            private boolean hasRun = false;
+            @Override public void run() {
+                if (!hasRun) {
+                    m_sched.addEvent(event);
+                    hasRun = true;
+                }
+            }
+        };
+        
+        m_sched.addEvent(adder);
+        m_sched.run();
+        assertEquals(1, event.getCount());
+        m_sched.run();
+        assertEquals(2, event.getCount());
+    }
+
+}