mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-28 02:13:45 +08:00
AngusBell97
121 lines
4.7 KiB
Python
121 lines
4.7 KiB
Python
"""Requested standby contract, exercised through the actual Device class."""
|
|
import json
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from openpilot.starpilot.common import screen_settings as screen
|
|
from test_screen_device_runtime import make_device
|
|
|
|
|
|
EXPECTED = {
|
|
'StandbyWakeEngage', 'StandbyWakeDisengage', 'StandbyWakeInfoAlert',
|
|
'StandbyWakeWarningAlert', 'StandbyWakeCriticalAlert', 'StandbyWakeTurnSignal', 'StandbyWakeButton',
|
|
}
|
|
|
|
|
|
def disabled():
|
|
return dict.fromkeys(EXPECTED, False)
|
|
|
|
|
|
def test_only_requested_wake_options_are_exposed():
|
|
assert screen.SCREEN_WAKE_KEYS == EXPECTED
|
|
root = Path(__file__).resolve().parents[3]
|
|
layout = json.loads((root / 'starpilot/common/assets/device_settings_layout.json').read_text())
|
|
keys = {p['key'] for section in layout for p in section['params'] if p['key'].startswith('StandbyWake')}
|
|
assert keys == EXPECTED
|
|
|
|
|
|
def test_touch_always_wakes_with_all_options_disabled():
|
|
device, state, app = make_device(**disabled(), StandbyWakeTouch=False)
|
|
device._update_wakefulness()
|
|
app.mouse_events = [SimpleNamespace(left_down=True)]
|
|
device._update_wakefulness()
|
|
assert device.awake
|
|
assert device._calculate_brightness() > 0
|
|
|
|
|
|
@pytest.mark.parametrize('enabled', [False, True])
|
|
@pytest.mark.parametrize('brightness', [0, 101])
|
|
def test_unassigned_buttons_only_wake_when_button_toggle_enabled(enabled, brightness):
|
|
device, state, _ = make_device(**{**disabled(), 'StandbyWakeButton': enabled}, ScreenBrightnessOnroad=brightness)
|
|
device._update_wakefulness()
|
|
state.params_memory.values['StandbyButtonPressTime'] = 99_500_000_000
|
|
device._update_wakefulness()
|
|
assert device.awake is enabled
|
|
assert (device._calculate_brightness() > 0) is enabled
|
|
|
|
|
|
@pytest.mark.parametrize('field,value,old_key', [
|
|
('brakePressed', True, 'StandbyWakeBrake'),
|
|
('gasPressed', True, 'StandbyWakeAccelerator'),
|
|
('gearShifter', 'reverse', 'StandbyWakeDriveState'),
|
|
])
|
|
def test_removed_driver_triggers_do_not_wake_even_with_old_enabled_settings(field, value, old_key):
|
|
device, state, _ = make_device(**disabled(), **{old_key: True})
|
|
device._update_wakefulness()
|
|
setattr(state.sm['carState'], field, value)
|
|
device._update_wakefulness()
|
|
assert device._calculate_brightness() == 0
|
|
|
|
|
|
@pytest.mark.parametrize('target,key', [('ENGAGED', 'StandbyWakeEngage'), ('DISENGAGED', 'StandbyWakeDisengage')])
|
|
@pytest.mark.parametrize('selected', [False, True])
|
|
def test_dom_status_transition_out_of_override_obeys_corresponding_toggle(target, key, selected):
|
|
device, state, _ = make_device(**{**disabled(), key: selected})
|
|
state.status = type(state.status).OVERRIDE
|
|
device._update_wakefulness()
|
|
assert device._calculate_brightness() == 0
|
|
state.status = getattr(type(state.status), target)
|
|
device._update_wakefulness()
|
|
assert (device._calculate_brightness() > 0) is selected
|
|
|
|
|
|
@pytest.mark.parametrize('status,key', [('userPrompt', 'StandbyWakeWarningAlert'), ('critical', 'StandbyWakeCriticalAlert')])
|
|
def test_dom_non_normal_alert_status_wakes_even_without_alert_size(status, key):
|
|
device, state, _ = make_device(**{**disabled(), key: True})
|
|
state.sm['selfdriveState'].alertStatus = status
|
|
device._update_wakefulness()
|
|
assert device._calculate_brightness() > 0
|
|
|
|
|
|
def test_dom_standby_powers_display_down_until_touch():
|
|
device, _, app = make_device(**disabled())
|
|
device._update_wakefulness()
|
|
assert not device.awake
|
|
app.mouse_events = [SimpleNamespace(left_down=True)]
|
|
device._update_wakefulness()
|
|
assert device.awake
|
|
|
|
|
|
@pytest.mark.parametrize('event', ['engage', 'alert'])
|
|
def test_dom_manual_zero_suppresses_automatic_status_and_alert_wakes(event):
|
|
device, state, app = make_device(ScreenBrightnessOnroad=0)
|
|
if event == 'engage':
|
|
state.status = type(state.status).ENGAGED
|
|
state.sm['selfdriveState'].enabled = True
|
|
else:
|
|
state.sm['selfdriveState'].alertStatus = 'critical'
|
|
state.sm['selfdriveState'].alertSize = 'full'
|
|
device._update_wakefulness()
|
|
assert device._calculate_brightness() == 0
|
|
app.mouse_events = [SimpleNamespace(left_down=True)]
|
|
device._update_wakefulness()
|
|
assert device._calculate_brightness() == 5
|
|
|
|
|
|
@pytest.mark.parametrize('ignition', [False, True])
|
|
@pytest.mark.parametrize('started', [False, True])
|
|
@pytest.mark.parametrize('brightness', [0, 101])
|
|
def test_dom_ignition_transition_remains_unconditional(ignition, started, brightness):
|
|
device, state, _ = make_device(**disabled(), StandbyWakeDriveState=False,
|
|
ScreenBrightness=brightness, ScreenBrightnessOnroad=brightness)
|
|
state.started = started
|
|
state.ignition = device._ignition = not ignition
|
|
device._interaction_time = 90
|
|
state.ignition = ignition
|
|
device._update_wakefulness()
|
|
assert device.awake
|
|
assert device._calculate_brightness() > 0
|