6e2ccc8b15
version: sunnypilot v2026.002.000 (staging) date: 2026-05-27T04:05:25 master commit: dfc3c98b226da57a653daf57131a8a3d66166fcb
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)
|
|
|
|
mocker.patch("openpilot.system.updated.updated.HARDWARE.get_device_type", return_value=device_type)
|
|
|
|
try:
|
|
assert Updater().target_branch == expected
|
|
finally:
|
|
params.remove("UpdaterTargetBranch")
|