mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-10-28 00:53:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Align updater test mocks with offroad status keyword

Browse Source
This commit is contained in:
Dom
2026-10-21 09:39:51 -05:00
parent 31d888ba79
commit 210e6504fe
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openpilot/system/updated/tests/test_fast_request.py
+4 -3
View File
@@ -204,7 +204,7 @@ class TestFastAdapter(unittest.TestCase):
source = Mock()
source.allowed.return_value = True
source.effective.return_value = False
source.status.side_effect = lambda source=source: {'allowed': source.allowed(), 'reason': None}
source.status.side_effect = lambda *, updating_offroad=False, source=source: {'allowed': source.allowed(), 'reason': None}
source.update_status.side_effect = lambda manager_offroad, source=source: PhysicalSource.update_status(source, manager_offroad)
source.update_ready.side_effect = lambda manager_offroad, source=source: PhysicalSource.update_ready(source, manager_offroad)
physical = ModuleType('openpilot.starpilot.drive_state.evidence')
@@ -225,6 +225,7 @@ class TestFastAdapter(unittest.TestCase):
self.assertEqual(updater.fast_update('SecretGoodStarPilot', 'b' * 40), 'reboot-requested')
source.close.assert_called_once()
source.update_ready.assert_called_once_with(True)
source.status.assert_called_once_with(updating_offroad=True)
self.assertEqual(updater.params.get_bool.call_args_list, [mock.call('IsOffroad')])
updater.params.put_bool.assert_called_once_with('UpdateAvailable', False, block=True)
@@ -243,7 +244,7 @@ class TestFastAdapter(unittest.TestCase):
source = Mock()
source.allowed.return_value = allowed
source.effective.return_value = effective
source.status.side_effect = lambda source=source: {'allowed': source.allowed(), 'reason': None}
source.status.side_effect = lambda *, updating_offroad=False, source=source: {'allowed': source.allowed(), 'reason': None}
source.update_status.side_effect = lambda manager_offroad, source=source: PhysicalSource.update_status(source, manager_offroad)
source.update_ready.side_effect = lambda manager_offroad, source=source: PhysicalSource.update_ready(source, manager_offroad)
physical = ModuleType('openpilot.starpilot.drive_state.evidence')
@@ -261,7 +262,7 @@ class TestFastAdapter(unittest.TestCase):
if not offroad:
source.status.assert_not_called()
else:
source.status.assert_called_once()
source.status.assert_called_once_with(updating_offroad=True)
if __name__ == '__main__':