29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
import unittest
|
|
from unittest.mock import Mock
|
|
|
|
from app.common.bootstrap import SORT_KEY, identifier, verify_table
|
|
|
|
|
|
class TestBootstrap(unittest.TestCase):
|
|
def test_identifiers_cannot_inject_sql(self):
|
|
self.assertEqual(identifier("cdr_records"), "cdr_records")
|
|
for value in ("a; DROP TABLE b", "a.b", "", "1table"):
|
|
with self.subTest(value=value), self.assertRaises(ValueError):
|
|
identifier(value)
|
|
|
|
def test_sorting_key_includes_deterministic_record_identity(self):
|
|
self.assertEqual(
|
|
SORT_KEY,
|
|
"service_type, radio_access_type, cell_id, session_start_time, msisdn, cdr_id",
|
|
)
|
|
|
|
def test_sorting_key_without_record_identity_fails_without_mutating_data(self):
|
|
client = Mock()
|
|
client.query.return_value.result_rows = [(
|
|
"ReplacingMergeTree",
|
|
"service_type, radio_access_type, cell_id, session_start_time, msisdn",
|
|
)]
|
|
with self.assertRaisesRegex(RuntimeError, "reset"):
|
|
verify_table(client, "telecom", "cdr_records")
|
|
client.command.assert_not_called()
|