Add trigger.onchange test

wpilibsuite · Jul 18, 2024 · 08ca37b · 08ca37b
1 parent 15001af
commit 08ca37b
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/wpilibNewCommands/src/test/java/edu/wpi/first/wpilibj2/command/button/TriggerTest.java b/wpilibNewCommands/src/test/java/edu/wpi/first/wpilibj2/command/button/TriggerTest.java
@@ -62,6 +62,32 @@ void onFalseTest() {
     assertFalse(command1.isScheduled());
   }
 
+  @Test
+  void onChangeTest() {
+    CommandScheduler scheduler = CommandScheduler.getInstance();
+    AtomicBoolean finished = new AtomicBoolean(false);
+    Command command1 = new WaitUntilCommand(finished::get);
+
+    InternalButton button = new InternalButton();
+    button.setPressed(true);
+    button.onChange(command1);
+    scheduler.run();
+    assertFalse(command1.isScheduled());
+    button.setPressed(false);
+    scheduler.run();
+    assertTrue(command1.isScheduled());
+    finished.set(true);
+    scheduler.run();
+    assertFalse(command1.isScheduled());
+    finished.set(false);
+    button.setPressed(true);
+    scheduler.run();
+    assertTrue(command1.isScheduled());
+    finished.set(true);
+    scheduler.run();
+    assertFalse(command1.isScheduled());
+  }
+
   @Test
   void whileTrueRepeatedlyTest() {
     CommandScheduler scheduler = CommandScheduler.getInstance();