Files
lingniu-vehicle-ingest/tools/test_jt808_e2e_smoke.py
lingniu 8804c01d99
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
fix: prevent tdengine raw frame overwrite
2026-06-29 18:41:58 +08:00

81 lines
2.9 KiB
Python

import pathlib
import types
import unittest
from tools import jt808_e2e_smoke as smoke
class Jt808E2ESmokeTest(unittest.TestCase):
def test_build_frame_adds_boundaries_and_valid_checksum(self):
body = smoke.build_location_body(smoke.parse_event_time("2024-01-02T03:04:05"))
frame = smoke.build_frame(0x0200, "13079961234", 1, body)
self.assertEqual(frame[0], 0x7E)
self.assertEqual(frame[-1], 0x7E)
unescaped = smoke.unescape_payload(frame[1:-1])
self.assertEqual(smoke.bcc(unescaped), 0)
def test_next_page_params_include_cursor(self):
params = smoke.next_page_params(
{"phone": "13079961234", "limit": 1},
{"nextCursor": {"ts": "2024-01-01T19:04:05Z", "id": "frame-1"}},
)
self.assertEqual(params["cursorTs"], "2024-01-01T19:04:05Z")
self.assertEqual(params["cursorId"], "frame-1")
self.assertEqual(params["phone"], "13079961234")
def test_archive_path_maps_archive_uri_to_root(self):
path = smoke.archive_path(
pathlib.Path("/tmp/archive"),
"archive://2026/06/29/JT808/unknown/frame.bin",
)
self.assertEqual(path, pathlib.Path("/tmp/archive/2026/06/29/JT808/unknown/frame.bin"))
def test_raw_frame_count_sql_filters_by_vehicle_key_phone_and_raw_uri(self):
sql = smoke.raw_frame_count_sql("13079961234", [
{
"vehicleKey": "jt808:13079961234",
"rawUri": "archive://2026/06/29/JT808/13079961234/frame.bin",
},
])
self.assertEqual(
sql,
"SELECT COUNT(*) FROM raw_frames WHERE protocol = 'JT808' "
"AND vehicle_key = 'jt808:13079961234' "
"AND phone = '13079961234' "
"AND raw_uri = 'archive://2026/06/29/JT808/13079961234/frame.bin'",
)
def test_raw_frame_count_sqls_include_each_unique_visible_raw_uri(self):
sqls = smoke.raw_frame_count_sqls("13079961234", [
{
"vehicleKey": "jt808:13079961234",
"rawUri": "archive://2026/06/29/JT808/13079961234/frame-1.bin",
},
{
"vehicleKey": "jt808:13079961234",
"rawUri": "archive://2026/06/29/JT808/13079961234/frame-2.bin",
},
{
"vehicleKey": "jt808:13079961234",
"rawUri": "archive://2026/06/29/JT808/13079961234/frame-1.bin",
},
])
self.assertEqual(len(sqls), 2)
self.assertTrue(sqls[0].endswith("frame-1.bin'"))
self.assertTrue(sqls[1].endswith("frame-2.bin'"))
def test_expected_history_count_waits_for_pagination_sample(self):
args = types.SimpleNamespace(frames=3, expect_history_count=1, verify_pagination=True)
self.assertEqual(smoke.expected_history_count(args), 2)
if __name__ == "__main__":
unittest.main()