fix: handle comments in has_table_query (#23882)

This commit is contained in:
Beto Dealmeida
2023-05-01 11:06:54 -07:00
committed by Elizabeth Thompson
parent 4af81bf70a
commit 60a1652be2
2 changed files with 11 additions and 1 deletions

View File

@@ -1195,6 +1195,14 @@ def test_sqlparse_issue_652():
("extract(HOUR from from_unixtime(hour_ts)", False),
("(SELECT * FROM table)", True),
("(SELECT COUNT(DISTINCT name) from birth_names)", True),
(
"(SELECT table_name FROM information_schema.tables WHERE table_name LIKE '%user%' LIMIT 1)",
True,
),
(
"(SELECT table_name FROM /**/ information_schema.tables WHERE table_name LIKE '%user%' LIMIT 1)",
True,
),
],
)
def test_has_table_query(sql: str, expected: bool) -> None: