From c1b7a2d2eee307b76e6f87577527631282c95e5f Mon Sep 17 00:00:00 2001 From: Beto Dealmeida Date: Wed, 20 May 2026 12:11:19 -0400 Subject: [PATCH] Bump code coverage to 100% --- tests/unit_tests/sql/rls_splice_unit_tests.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/unit_tests/sql/rls_splice_unit_tests.py b/tests/unit_tests/sql/rls_splice_unit_tests.py index 5b1119adeae..ba9846cea1a 100644 --- a/tests/unit_tests/sql/rls_splice_unit_tests.py +++ b/tests/unit_tests/sql/rls_splice_unit_tests.py @@ -21,6 +21,7 @@ from sqlglot import Dialect, exp from superset.sql.parse import SQLStatement, Table from superset.sql.rls_splice import ( + _after_previous_token, _classify_source_predicate, _find_condition_end, _find_join_splice, @@ -119,6 +120,11 @@ def test_classify_source_predicate_returns_none_for_unsupported_parent() -> None assert result == ("none", None, None) +def test_after_previous_token_returns_zero_at_stream_start() -> None: + tokens = _tokenize("SELECT 1") + assert _after_previous_token(tokens, 0) == 0 + + def test_scan_until_scope_boundary_tracks_parenthesis_depth() -> None: sql = "SELECT * FROM t WHERE (a = 1)" tokens = _tokenize(sql)