mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
from unittest.mock import Mock, patch
|
|
from openpilot.selfdrive.ui.onroad import cameraview as camera
|
|
|
|
|
|
class ViewWithoutResources(camera.CameraView):
|
|
def close(self):
|
|
pass
|
|
|
|
|
|
def test_constructor_caches_once_across_repeated_switches():
|
|
with patch.object(camera, 'COMMA_HARDWARE', False), \
|
|
patch.object(camera, 'VisionIpcClient', side_effect=lambda *a, **k: Mock()) as factory, \
|
|
patch.object(camera.rl, 'load_shader_from_memory', return_value=Mock(id=0)), \
|
|
patch.object(camera.rl, 'get_shader_location', return_value=1), \
|
|
patch.object(camera.ui_state, 'add_offroad_transition_callback'):
|
|
view = ViewWithoutResources('camera-cache-test', camera.VisionStreamType.VISION_STREAM_NARROW_ROAD)
|
|
expected = len(list(camera.VisionStreamType))
|
|
assert factory.call_count == expected
|
|
identities = dict(view._clients)
|
|
with patch.object(view, '_initialize_textures'):
|
|
for _ in range(50):
|
|
for stream in camera.VisionStreamType:
|
|
view.switch_stream(stream)
|
|
if view._switching:
|
|
view._complete_switch()
|
|
assert view.client is identities[stream]
|
|
assert factory.call_count == expected
|
|
|
|
|
|
def test_new_drive_drains_cached_clients_and_cancels_pending_switch():
|
|
view = ViewWithoutResources.__new__(ViewWithoutResources)
|
|
view._clients = {stream: Mock() for stream in camera.VisionStreamType}
|
|
view.client = view._clients[camera.VisionStreamType.VISION_STREAM_NARROW_ROAD]
|
|
view.frame = Mock()
|
|
view._target_client = Mock()
|
|
view._target_stream_type = Mock()
|
|
view._switching = True
|
|
with patch.object(camera.ui_state, 'is_onroad', return_value=True):
|
|
view._offroad_transition()
|
|
assert view.frame is None and view._target_client is None
|
|
assert view._target_stream_type is None and not view._switching
|
|
for client in view._clients.values():
|
|
client.recv.assert_called_once_with(timeout_ms=0)
|