From ae65a01afd1e99f849fe20d2f1c89ebaeca08ce6 Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Wed, 14 Feb 2024 05:29:08 -0600 Subject: [PATCH] SegmentRange: test API call (#31456) * test * better * better old-commit-hash: c4f7991bb63ba7953f37eb08f14d227fec8a5524 --- tools/lib/tests/test_logreader.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tools/lib/tests/test_logreader.py b/tools/lib/tests/test_logreader.py index c21c94342c..5dd7f7fa0d 100644 --- a/tools/lib/tests/test_logreader.py +++ b/tools/lib/tests/test_logreader.py @@ -90,6 +90,19 @@ class TestLogReader(unittest.TestCase): sr = SegmentRange(segment_range) parse_slice(sr) + @parameterized.expand([ + (f"{TEST_ROUTE}/0", False), + (f"{TEST_ROUTE}/:2", False), + (f"{TEST_ROUTE}/0:", True), + (f"{TEST_ROUTE}/-1", True), + (f"{TEST_ROUTE}", True), + ]) + def test_slicing_api_call(self, segment_range, api_call): + with mock.patch("openpilot.tools.lib.route.get_max_seg_number_cached") as max_seg_mock: + max_seg_mock.return_value = NUM_SEGS + parse_slice(SegmentRange(segment_range)) + self.assertEqual(api_call, max_seg_mock.called) + @pytest.mark.slow def test_modes(self): qlog_len = len(list(LogReader(f"{TEST_ROUTE}/0", ReadMode.QLOG)))