chore: add test for 'wait_until_state'

rust-dd · Oct 8, 2024 · 5f72b72 · 5f72b72
1 parent 6ab2f48
commit 5f72b72
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/dht.rs b/src/dht.rs
@@ -136,4 +136,23 @@ mod tests {
 
         dht.pin.done();
     }
+
+    #[test]
+    fn test_wait_until_state() {
+        let expectations = [
+            PinTransaction::get(State::Low),
+            PinTransaction::get(State::Low),
+            PinTransaction::get(State::High),
+        ];
+
+        let mock_pin = Mock::new(&expectations);
+        let mock_delay = MockNoop::new();
+
+        let mut dht = Dht::new(mock_pin, mock_delay);
+
+        let result = dht.wait_until_state(PinState::High);
+        assert!(result.is_ok());
+
+        dht.pin.done();
+    }
 }