mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-10 06:23:43 +08:00
firestar5683
39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
from contextlib import ExitStack
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from openpilot.starpilot.aol import dm as owner
|
|
from openpilot.starpilot.aol.tests.stock_lateral_test_helpers import callers
|
|
|
|
|
|
@pytest.mark.parametrize('family', ['gm', 'toyota'])
|
|
@pytest.mark.parametrize('restriction', ['healthy', 'red', 'lockout', 'always_lockout', 'invalid', 'native_denied'])
|
|
def test_actual_dm_proof_is_lazy_only_without_an_exception(family, restriction):
|
|
with ExitStack() as cleanup:
|
|
sd, _, drive, _, _ = callers(cleanup, family, True)
|
|
original = sd.update_events
|
|
|
|
def events(state):
|
|
value = sd.sm['driverMonitoringState'].as_builder()
|
|
value.alertLevel = 'three' if restriction in ('red', 'native_denied') else 'none'
|
|
value.lockout = restriction == 'lockout'
|
|
value.alwaysOnLockout = restriction == 'always_lockout'
|
|
sd.sm.data['driverMonitoringState'] = value.as_reader()
|
|
if restriction == 'invalid':
|
|
sd.sm.valid['driverMonitoringState'] = False
|
|
return original(state)
|
|
|
|
cleanup.enter_context(mock.patch.object(sd, 'update_events', side_effect=events))
|
|
with mock.patch.object(owner, 'armed_lateral_current', wraps=owner.armed_lateral_current) as proof:
|
|
axis, _, command = drive(2, native_request_long=False, allowed=restriction != 'native_denied')
|
|
if restriction in ('healthy', 'invalid'):
|
|
proof.assert_not_called()
|
|
else:
|
|
assert proof.call_count > 0
|
|
if restriction in ('invalid', 'native_denied'):
|
|
assert not axis.lateralActive and not command.latActive
|
|
else:
|
|
assert axis.lateralActive and command.latActive
|
|
assert not axis.longitudinalActive and not command.longActive
|