IQ.Pilot Prebuilt Release @ ab07000

system/updated/tests/test_branch_migrations.py

@@ -0,0 +1,54 @@
+from openpilot.common.params import Params
+from openpilot.system.version import BuildMetadata, OpenpilotMetadata
+from openpilot.system.updated.updated import Updater
+
+
+def test_tici_branch_migration(mocker):
+  params = Params()
+  params.put("UpdaterTargetBranch", "master-dev")
+  mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value="tici")
+
+  try:
+    assert Updater().target_branch == "master-mici-tici"
+  finally:
+    params.remove("UpdaterTargetBranch")
+
+
+def test_non_tici_branch_unchanged(mocker):
+  params = Params()
+  params.put("UpdaterTargetBranch", "master-dev")
+  mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value="tizi")
+
+  try:
+    assert Updater().target_branch == "master-dev"
+  finally:
+    params.remove("UpdaterTargetBranch")
+
+
+def test_non_git_baked_deployment_uses_build_metadata(mocker):
+  mocker.patch("openpilot.system.updated.updated.has_git_repo", return_value=False)
+  mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value="tici")
+  mocker.patch(
+    "openpilot.system.updated.updated.get_build_metadata",
+    return_value=BuildMetadata(
+      "release3",
+      OpenpilotMetadata(
+        version="1.2.3",
+        release_notes="notes",
+        git_commit="abcdef1234567890",
+        git_origin="github.com/IQLvbs/openpilot",
+        git_commit_date="Jul 02",
+        build_style="release",
+        is_dirty=False,
+      ),
+    ),
+  )
+
+  updater = Updater()
+
+  assert updater.git_mode is False
+  assert updater.get_branch() == "release3"
+  assert updater.get_commit_hash() == "abcdef1234567890"
+  assert updater.target_branch == "release3"
+  assert updater.update_available is False
+  assert updater.update_ready is False