updated: complete branch migration dict for mici and tizi (#37830)

system/updated/tests/test_updated.py

@@ -0,0 +1,38 @@
+import pytest
+
+from openpilot.common.params import Params
+from openpilot.system.updated.updated import Updater
+
+
+@pytest.mark.parametrize(("device_type", "branch", "expected"), [
+  ("tizi", "release3", "release-tizi"),
+  ("tizi", "release3-staging", "release-tizi-staging"),
+  ("mici", "release3", "release-mici"),
+  ("mici", "release3-staging", "release-mici-staging"),
+])
+def test_target_branch_migration_from_current_branch(mocker, device_type, branch, expected):
+  params = Params()
+  params.remove("UpdaterTargetBranch")
+
+  mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value=device_type)
+  mocker.patch.object(Updater, "get_branch", return_value=branch)
+
+  assert Updater().target_branch == expected
+
+
+@pytest.mark.parametrize(("device_type", "branch", "expected"), [
+  ("tizi", "release3", "release-tizi"),
+  ("tizi", "release3-staging", "release-tizi-staging"),
+  ("mici", "release3", "release-mici"),
+  ("mici", "release3-staging", "release-mici-staging"),
+])
+def test_target_branch_migration_from_param(mocker, device_type, branch, expected):
+  params = Params()
+  params.put("UpdaterTargetBranch", branch)
+
+  mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value=device_type)
+
+  try:
+    assert Updater().target_branch == expected
+  finally:
+    params.remove("UpdaterTargetBranch")

system/updated/updated.py

@@ -244,6 +244,9 @@ class Updater:
       b = self.get_branch(BASEDIR)
     b = {
       ("tizi", "release3"): "release-tizi",
+      ("tizi", "release3-staging"): "release-tizi-staging",
+      ("mici", "release3"): "release-mici",
+      ("mici", "release3-staging"): "release-mici-staging",
     }.get((HARDWARE.get_device_type(), b), b)
     return b