|
|
|
@ -9,7 +9,7 @@ import requests |
|
|
|
|
from parameterized import parameterized |
|
|
|
|
from unittest import mock |
|
|
|
|
|
|
|
|
|
from openpilot.tools.lib.logreader import LogIterable, LogReader, parse_indirect, parse_slice, ReadMode |
|
|
|
|
from openpilot.tools.lib.logreader import LogIterable, LogReader, comma_api_source, parse_indirect, parse_slice, ReadMode |
|
|
|
|
from openpilot.tools.lib.route import SegmentRange |
|
|
|
|
|
|
|
|
|
NUM_SEGS = 17 # number of segments in the test route |
|
|
|
@ -137,6 +137,18 @@ class TestLogReader(unittest.TestCase): |
|
|
|
|
lr = LogReader(f"{TEST_ROUTE}/0:4") |
|
|
|
|
self.assertEqual(len(lr.run_across_segments(4, noop)), len(list(lr))) |
|
|
|
|
|
|
|
|
|
@pytest.mark.slow |
|
|
|
|
def test_auto_mode(self): |
|
|
|
|
lr = LogReader(f"{TEST_ROUTE}/0/q") |
|
|
|
|
qlog_len = len(list(lr)) |
|
|
|
|
with mock.patch("openpilot.tools.lib.route.Route.log_paths") as log_paths_mock: |
|
|
|
|
log_paths_mock.return_value = [None] * NUM_SEGS |
|
|
|
|
# Should fall back to qlogs since rlogs are not available |
|
|
|
|
lr = LogReader(f"{TEST_ROUTE}/0/a", default_source=comma_api_source) |
|
|
|
|
log_len = len(list(lr)) |
|
|
|
|
|
|
|
|
|
self.assertEqual(qlog_len, log_len) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
|
unittest.main() |
|
|
|
|