Add tests for Hold

projectharmonia · Oct 16, 2024 · 7275ed4 · 7275ed4
1 parent 041b727
commit 7275ed4
Showing 1 changed file with 49 additions and 0 deletions.
diff --git a/src/input_context/input_condition/hold.rs b/src/input_context/input_condition/hold.rs
@@ -89,3 +89,52 @@ impl InputCondition for Hold {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn hold() {
+        let world = World::new();
+        let actions_data = ActionsData::default();
+
+        let mut hold = Hold::new(1.0);
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 0.0, 1.0.into()),
+            ActionState::Ongoing,
+        );
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 1.0, 1.0.into()),
+            ActionState::Fired,
+        );
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 1.0, 1.0.into()),
+            ActionState::Fired,
+        );
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 1.0, 0.0.into()),
+            ActionState::None,
+        );
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 0.0, 1.0.into()),
+            ActionState::Ongoing,
+        );
+    }
+
+    #[test]
+    fn one_shot() {
+        let world = World::new();
+        let actions_data = ActionsData::default();
+
+        let mut hold = Hold::new(1.0).one_shot(true);
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 1.0, 1.0.into()),
+            ActionState::Fired,
+        );
+        assert_eq!(
+            hold.evaluate(&world, &actions_data, 1.0, 1.0.into()),
+            ActionState::None,
+        );
+    }
+}