From 117d5cee4ff336874f529838cbb747debcbd023a Mon Sep 17 00:00:00 2001
From: Ethan Reish <Ereish02@gmail.com>
Date: Wed, 15 Apr 2026 12:46:15 -0500
Subject: [PATCH] updated: complete branch migration dict for mici and tizi
 (#37830)

---
 system/updated/tests/test_updated.py | 38 ++++++++++++++++++++++++++++
 system/updated/updated.py            |  3 +++
 2 files changed, 41 insertions(+)
 create mode 100644 system/updated/tests/test_updated.py

diff --git a/system/updated/tests/test_updated.py b/system/updated/tests/test_updated.py
new file mode 100644
index 000000000..d36d4dd4e
--- /dev/null
+++ b/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")
diff --git a/system/updated/updated.py b/system/updated/updated.py
index ffd10e038..693a31c9f 100755
--- a/system/updated/updated.py
+++ b/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