IQ.Pilot Release Commit @ bec7652

iqpilot/selfdrive/selfdrived/tests/test_longitudinal_pref_persistence.py

@@ -0,0 +1,40 @@
+from iqpilot.cereal import car
+
+from iqpilot.selfdrive.selfdrived.selfdrived import _cleanup_startup_params, get_sanitize_int_param
+
+
+class DummyParams:
+  def __init__(self):
+    self.removed: list[str] = []
+
+  def remove(self, key: str) -> None:
+    self.removed.append(key)
+
+
+class TestLongitudinalPrefPersistence:
+  def test_startup_cleanup_preserves_persistent_longitudinal_preferences(self):
+    params = DummyParams()
+    cp = car.CarParams()
+    cp.alphaLongitudinalAvailable = False
+    cp.openpilotLongitudinalControl = False
+
+    _cleanup_startup_params(cp, params)
+
+    assert params.removed == []
+
+  def test_invalid_personality_is_clamped_before_use(self):
+    class ParamsWithInvalidPersonality:
+      def __init__(self):
+        self.value = 3
+
+      def get(self, key: str, return_default: bool = False) -> int:
+        assert key == "LongitudinalPersonality"
+        return self.value
+
+      def put(self, key: str, value: int) -> None:
+        assert key == "LongitudinalPersonality"
+        self.value = value
+
+    params = ParamsWithInvalidPersonality()
+    assert get_sanitize_int_param("LongitudinalPersonality", 0, 2, params) == 2
+    assert params.value == 2