mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-08 15:12:10 +08:00
firestar5683
54 lines
2.5 KiB
Python
54 lines
2.5 KiB
Python
import importlib
|
|
import sys
|
|
import types
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def updater_module(monkeypatch):
|
|
hardware = types.ModuleType("openpilot.system.hardware")
|
|
hardware.HARDWARE = types.SimpleNamespace(get_device_type=lambda: "tici")
|
|
monkeypatch.setitem(sys.modules, "openpilot.system.hardware", hardware)
|
|
sys.modules.pop("openpilot.system.ui.updater", None)
|
|
module = importlib.import_module("openpilot.system.ui.updater")
|
|
yield module
|
|
sys.modules.pop("openpilot.system.ui.updater", None)
|
|
|
|
|
|
def test_device_tree_tici_uses_big_ui(updater_module):
|
|
with patch.object(updater_module, "_device_tree_device_type", return_value="tici"), \
|
|
patch.object(updater_module, "_framebuffer_size", return_value=(536, 240)), \
|
|
patch.object(updater_module, "_reported_device_type", return_value="mici"):
|
|
assert updater_module._ui_device_type() == "tici"
|
|
|
|
|
|
def test_device_tree_tizi_uses_big_ui(updater_module):
|
|
with patch.object(updater_module, "_device_tree_device_type", return_value="tizi"), \
|
|
patch.object(updater_module, "_framebuffer_size", return_value=(536, 240)), \
|
|
patch.object(updater_module, "_reported_device_type", return_value="mici"):
|
|
assert updater_module._ui_device_type() == "tizi"
|
|
|
|
|
|
@pytest.mark.parametrize("device_tree_type", ["mici", "comma 4", "comma four"])
|
|
def test_non_tici_device_tree_uses_small_ui(updater_module, device_tree_type):
|
|
with patch.object(updater_module, "_device_tree_device_type", return_value=device_tree_type), \
|
|
patch.object(updater_module, "_framebuffer_size", return_value=(2160, 1080)), \
|
|
patch.object(updater_module, "_reported_device_type", return_value="tici"):
|
|
assert updater_module._ui_device_type() == "mici"
|
|
|
|
|
|
def test_small_framebuffer_uses_small_ui_without_device_tree(updater_module):
|
|
with patch.object(updater_module, "_device_tree_device_type", return_value=None), \
|
|
patch.object(updater_module, "_framebuffer_size", return_value=(536, 240)), \
|
|
patch.object(updater_module, "_reported_device_type", return_value="tici"):
|
|
assert updater_module._ui_device_type() == "mici"
|
|
|
|
|
|
def test_reported_tici_used_when_no_other_signal(updater_module):
|
|
with patch.object(updater_module, "_device_tree_device_type", return_value=None), \
|
|
patch.object(updater_module, "_framebuffer_size", return_value=None), \
|
|
patch.object(updater_module, "_reported_device_type", return_value="tici"):
|
|
assert updater_module._ui_device_type() == "tici"
|