fix: Saved queries list break if one query can't be parsed (#34289)

This commit is contained in:
Michael S. Molina
2025-07-24 08:30:04 -03:00
committed by GitHub
parent 9b88527883
commit 1e5a4e9bdc
2 changed files with 46 additions and 3 deletions

View File

@@ -22,7 +22,7 @@ from jinja2.exceptions import TemplateError
from pytest_mock import MockerFixture
from superset.errors import ErrorLevel, SupersetError, SupersetErrorType
from superset.exceptions import SupersetSecurityException
from superset.exceptions import SupersetParseError, SupersetSecurityException
from superset.models.sql_lab import Query, SavedQuery
@@ -43,6 +43,10 @@ from superset.models.sql_lab import Query, SavedQuery
level=ErrorLevel.ERROR,
)
),
SupersetParseError(
sql="INVALID SQL",
message="Invalid SQL syntax",
),
TemplateError,
],
)
@@ -57,3 +61,42 @@ def test_sql_tables_mixin_sql_tables_exception(
)
assert klass(sql="SELECT 1", database=MagicMock()).sql_tables == []
@pytest.mark.parametrize(
"klass",
[
Query,
SavedQuery,
],
)
@pytest.mark.parametrize(
"invalid_sql",
[
"SELECT * FROM table WHERE invalid syntax",
"INVALID SQL STATEMENT",
"SELECT * FROM; DROP TABLE users;",
"",
None,
],
)
def test_sql_tables_mixin_invalid_sql_returns_empty_list(
klass: type[Model],
invalid_sql: str,
mocker: MockerFixture,
) -> None:
"""Test that SqlTablesMixin returns empty list when SQL parsing fails."""
mocker.patch(
"superset.models.sql_lab.extract_tables_from_jinja_sql",
side_effect=SupersetParseError(
sql=invalid_sql or "INVALID SQL",
message=f"Failed to parse SQL: {invalid_sql}",
),
)
instance = (
klass(sql=invalid_sql, database=MagicMock())
if invalid_sql is not None
else klass(database=MagicMock())
)
assert instance.sql_tables == []