Compare commits

..

2 Commits

Author SHA1 Message Date
Evan
985ebb796e test(databases): assert form inputs are enabled after SQLAlchemy switch
Strengthens the presence check to also confirm the form fields are
usable, per bito-code-review feedback.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-07-06 18:26:41 -07:00
Claude Code
b304f33a9b test(databases): stabilize flaky SQLAlchemy-form visibility assertion
The form mounts inside DatabaseModal's animated tab pane, and
rc-motion's animation state in jsdom is nondeterministic: toBeVisible
intermittently times out (observed on unrelated PRs' jest shards) even
though the form is rendered. Assert on document presence instead, which
is what the tab-switch regression check actually needs.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-06 12:17:11 -07:00
3 changed files with 22 additions and 106 deletions

View File

@@ -1804,8 +1804,17 @@ describe('DatabaseModal', () => {
userEvent.click(screen.getByTestId('sqla-connect-btn'));
expect(await screen.findByTestId('database-name-input')).toBeVisible();
expect(screen.getByTestId('sqlalchemy-uri-input')).toBeVisible();
// assert on presence rather than visibility: the SQLAlchemy form mounts
// inside an animated tab pane, and rc-motion's animation state in jsdom
// is nondeterministic, so toBeVisible flakes while the form is in fact
// rendered (see the animated={{ tabPane: true }} Tabs in DatabaseModal)
const nameInput = await screen.findByTestId('database-name-input');
const uriInput = screen.getByTestId('sqlalchemy-uri-input');
expect(nameInput).toBeInTheDocument();
expect(uriInput).toBeInTheDocument();
// also confirm the form is actually usable, not just present
expect(nameInput).toBeEnabled();
expect(uriInput).toBeEnabled();
});
test.each([

View File

@@ -250,25 +250,6 @@ def json_to_dict(json_str: str) -> dict[Any, Any]:
return {}
UUID_NATIVE_TYPE_RE = re.compile(r"\buuid\b", re.IGNORECASE)
def is_uuid_native_type(native_type: Optional[str]) -> bool:
"""
Return True if a native column type represents a UUID.
Engines such as PostgreSQL and ClickHouse expose native UUID column types
(e.g. ``UUID``, ``Nullable(UUID)``) that map to ``GenericDataType.STRING``
yet reject LIKE/ILIKE against the raw column, so these columns need an
explicit cast to string before pattern matching. The match is on the
whole word ``uuid`` so unrelated types that merely contain the substring
(e.g. a hypothetical ``uuidish`` type) aren't misclassified.
"""
return native_type is not None and bool(
UUID_NATIVE_TYPE_RE.search(native_type.strip())
)
def convert_uuids(obj: Any) -> Any:
"""
Convert UUID objects to str so we can use yaml.safe_dump
@@ -3764,23 +3745,22 @@ class ExploreMixin: # pylint: disable=too-many-public-methods
elif op in {
utils.FilterOperator.ILIKE,
utils.FilterOperator.LIKE,
utils.FilterOperator.NOT_LIKE,
utils.FilterOperator.NOT_ILIKE,
}:
# Native UUID columns report GenericDataType.STRING but
# reject LIKE/ILIKE without a cast (see issue #41795)
needs_string_cast_for_like = (
target_generic_type != GenericDataType.STRING
or is_uuid_native_type(col_type)
)
if needs_string_cast_for_like:
if target_generic_type != GenericDataType.STRING:
sqla_col = sa.cast(sqla_col, sa.String)
if op == utils.FilterOperator.LIKE:
target_clause_list.append(sqla_col.like(eq))
elif op == utils.FilterOperator.ILIKE:
else:
target_clause_list.append(sqla_col.ilike(eq))
elif op == utils.FilterOperator.NOT_LIKE:
elif op in {
utils.FilterOperator.NOT_LIKE,
utils.FilterOperator.NOT_ILIKE,
}:
if target_generic_type != GenericDataType.STRING:
sqla_col = sa.cast(sqla_col, sa.String)
if op == utils.FilterOperator.NOT_LIKE:
target_clause_list.append(sqla_col.not_like(eq))
else:
target_clause_list.append(sqla_col.not_ilike(eq))

View File

@@ -29,8 +29,7 @@ from pytest_mock import MockerFixture
from sqlalchemy import create_engine
from sqlalchemy.orm.session import Session
from sqlalchemy.pool import StaticPool
from sqlalchemy.sql.elements import Cast, ColumnElement
from sqlalchemy.sql.visitors import iterate
from sqlalchemy.sql.elements import ColumnElement
from superset.superset_typing import AdhocColumn, AdhocMetric, OrderBy
from superset.utils.core import GenericDataType
@@ -3411,75 +3410,3 @@ def test_get_sqla_query_dotted_struct_column_bigquery(
# ```forecasts.original`.`total_cost``` (the regression), so this negative
# assertion catches the actual failure mode, not just an exact-string match.
assert "`forecasts.original`" not in sql
@pytest.mark.parametrize(
"native_type",
["UUID", "uuid", "Nullable(UUID)"],
)
@pytest.mark.parametrize(
"op",
["LIKE", "ILIKE", "NOT LIKE", "NOT ILIKE"],
)
def test_like_filter_on_uuid_column_casts_to_string(
database: Database, native_type: str, op: str
) -> None:
"""
LIKE-family filters on native UUID columns must cast the column to string.
UUID columns map to ``GenericDataType.STRING``, so the generic-type guard
alone skips the string cast — but engines such as PostgreSQL and ClickHouse
reject LIKE/ILIKE against a raw UUID column (issue #41795: table chart
server-pagination search fails with e.g. "Illegal type UUID of argument of
function ilike"). The native column type must force the cast.
"""
from superset.connectors.sqla.models import SqlaTable, TableColumn
table = SqlaTable(
database=database,
schema=None,
table_name="t",
columns=[TableColumn(column_name="event_id", type=native_type)],
)
result = table.get_sqla_query(
columns=["event_id"],
metrics=[],
extras={},
filter=[{"col": "event_id", "op": op, "val": "abc%"}],
granularity=None,
is_timeseries=False,
orderby=[],
)
whereclause = result.sqla_query.whereclause
assert any(isinstance(node, Cast) for node in iterate(whereclause)), (
f"Expected a Cast node in the filter expression: {whereclause}"
)
def test_like_filter_on_string_column_does_not_cast(database: Database) -> None:
"""
LIKE-family filters on plain string columns must not add a redundant cast.
"""
from superset.connectors.sqla.models import SqlaTable, TableColumn
table = SqlaTable(
database=database,
schema=None,
table_name="t",
columns=[TableColumn(column_name="b", type="TEXT")],
)
result = table.get_sqla_query(
columns=["b"],
metrics=[],
extras={},
filter=[{"col": "b", "op": "ILIKE", "val": "abc%"}],
granularity=None,
is_timeseries=False,
orderby=[],
)
whereclause = result.sqla_query.whereclause
assert not any(isinstance(node, Cast) for node in iterate(whereclause)), (
f"Unexpected Cast node in the filter expression: {whereclause}"
)