mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-04 07:03:44 +08:00
firestar5683
78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
from openpilot.selfdrive.ui.layouts.settings import device
|
|
from openpilot.system.ui.widgets import DialogResult
|
|
|
|
|
|
class FakeParams:
|
|
def __init__(self):
|
|
self.writes = []
|
|
|
|
def put_bool(self, key, value):
|
|
self.writes.append((key, value))
|
|
|
|
def put_bool_nonblocking(self, *_args, **_kwargs):
|
|
raise AssertionError("reboot command must be written synchronously")
|
|
|
|
|
|
def _device_layout():
|
|
layout = device.DeviceLayout.__new__(device.DeviceLayout)
|
|
layout._params = FakeParams()
|
|
return layout
|
|
|
|
|
|
class FakeHardware:
|
|
def __init__(self, error=None):
|
|
self.error = error
|
|
self.reboot_calls = 0
|
|
|
|
def reboot(self):
|
|
self.reboot_calls += 1
|
|
if self.error is not None:
|
|
raise self.error
|
|
|
|
|
|
def test_confirmed_user_reboot_is_written_synchronously(monkeypatch):
|
|
layout = _device_layout()
|
|
hardware = FakeHardware()
|
|
monkeypatch.setattr(device, "HARDWARE", hardware)
|
|
|
|
layout._perform_reboot(DialogResult.CONFIRM)
|
|
|
|
assert layout._params.writes == [("DoUserReboot", True)]
|
|
assert hardware.reboot_calls == 1
|
|
|
|
|
|
def test_confirmed_user_reboot_is_allowed_onroad(monkeypatch):
|
|
layout = _device_layout()
|
|
hardware = FakeHardware()
|
|
|
|
monkeypatch.setattr(device, "HARDWARE", hardware)
|
|
monkeypatch.setattr(device.ui_state, "started", True)
|
|
monkeypatch.setattr(device.ui_state, "ignition", True)
|
|
layout._perform_reboot(DialogResult.CONFIRM)
|
|
|
|
assert layout._params.writes == [("DoUserReboot", True)]
|
|
assert hardware.reboot_calls == 1
|
|
|
|
|
|
def test_cancelled_user_reboot_is_ignored(monkeypatch):
|
|
layout = _device_layout()
|
|
hardware = FakeHardware()
|
|
monkeypatch.setattr(device, "HARDWARE", hardware)
|
|
|
|
layout._perform_reboot(DialogResult.CANCEL)
|
|
|
|
assert layout._params.writes == []
|
|
assert hardware.reboot_calls == 0
|
|
|
|
|
|
def test_failed_direct_reboot_keeps_manager_fallback(monkeypatch):
|
|
layout = _device_layout()
|
|
hardware = FakeHardware(RuntimeError("reboot failed"))
|
|
monkeypatch.setattr(device, "HARDWARE", hardware)
|
|
monkeypatch.setattr(device.cloudlog, "exception", lambda *_args, **_kwargs: None)
|
|
|
|
layout._perform_reboot(DialogResult.CONFIRM)
|
|
|
|
assert layout._params.writes == [("DoUserReboot", True)]
|
|
assert hardware.reboot_calls == 1
|