Files
sunnypilot-byd/openpilot/sunnypilot/system/updated/tests/test_sp_branch_migrations.py
T
github-actions[bot] de77c89cf6 sunnypilot v2026.002.000
version: sunnypilot v2026.002.000 (staging)
date: 2026-07-21T07:44:02
master commit: 2c334ede443d7391d27575af8c854a095ba702a8
2026-07-21 07:44:02 +00:00

74 lines
2.6 KiB
Python

"""
Copyright (c) 2021-, Haibin Wen, sunnypilot, and a number of other contributors.
This file is part of sunnypilot and is licensed under the MIT License.
See the LICENSE.md file in the root directory for more details.
"""
import pytest
from openpilot.common.params import Params
from openpilot.system.updated.updated import Updater
@pytest.mark.parametrize(("device_type", "branch", "expected"), [
("tici", "staging-c3-new", "staging-tici"),
("tici", "dev-c3-new", "staging-tici"),
("tici", "master", "master-tici"),
("tici", "master-dev-c3-new", "master-tici"),
("tizi", "staging-c3-new", "staging"),
("tizi", "dev-c3-new", "dev"),
("tizi", "master-dev-c3-new", "master-dev"),
("tizi", "release3", "release-tizi"),
("tizi", "release3-staging", "release-tizi-staging"),
("mici", "release3", "release-mici"),
("mici", "release3-staging", "release-mici-staging"),
])
def test_sp_branch_migrations_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"), [
("tici", "staging-c3-new", "staging-tici"),
("tici", "dev-c3-new", "staging-tici"),
("tici", "master", "master-tici"),
("tici", "master-dev-c3-new", "master-tici"),
("tizi", "staging-c3-new", "staging"),
("tizi", "dev-c3-new", "dev"),
("tizi", "master-dev-c3-new", "master-dev"),
("tizi", "release3", "release-tizi"),
("tizi", "release3-staging", "release-tizi-staging"),
("mici", "release3", "release-mici"),
("mici", "release3-staging", "release-mici-staging"),
])
def test_sp_branch_migrations_from_param(mocker, device_type, branch, expected):
params = Params()
params.put("UpdaterTargetBranch", branch, block=True)
mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value=device_type)
try:
assert Updater().target_branch == expected
finally:
params.remove("UpdaterTargetBranch")
@pytest.mark.parametrize(("device_type", "branch"), [
("tici", "unknown"),
("tizi", "unknown"),
("mici", "unknown"),
])
def test_sp_branch_migrations_passthrough(mocker, device_type, branch):
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 == branch