36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import unittest
|
|
|
|
from tools import tdengine_smoke
|
|
|
|
|
|
class TdengineSmokeTest(unittest.TestCase):
|
|
def test_sql_literal_escapes_single_quotes(self):
|
|
self.assertEqual(tdengine_smoke.sql_literal("VIN'001"), "'VIN''001'")
|
|
|
|
def test_raw_frame_count_sql_filters_by_protocol_vehicle_and_raw_uri(self):
|
|
sql = tdengine_smoke.raw_frame_count_sql(
|
|
protocol="JT808",
|
|
vehicle_key="jt808:13079962000",
|
|
raw_uri="archive://2026/06/29/JT808/frame.bin",
|
|
)
|
|
|
|
self.assertEqual(
|
|
sql,
|
|
"SELECT COUNT(*) FROM raw_frames WHERE protocol = 'JT808' "
|
|
"AND vehicle_key = 'jt808:13079962000' "
|
|
"AND raw_uri = 'archive://2026/06/29/JT808/frame.bin'",
|
|
)
|
|
|
|
def test_raw_frame_count_sql_requires_at_least_one_identity_or_raw_uri_filter(self):
|
|
with self.assertRaises(ValueError):
|
|
tdengine_smoke.raw_frame_count_sql(protocol="GB32960")
|
|
|
|
def test_count_from_response_reads_first_cell(self):
|
|
response = {"code": 0, "data": [[3]], "rows": 1}
|
|
|
|
self.assertEqual(tdengine_smoke.count_from_response(response), 3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|