Vehicle Researcher
6adb63b915
date: 2026-06-04T09:49:56 master commit: c0ab3550eca2e9daf197c46b7e4b24aa9637cf2e
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
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, 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")
|