fix: stabilize ingest wiki references

This commit is contained in:
Andrey Avtomonov 2026-05-13 13:18:50 +02:00
parent 53c9a1eefa
commit 7e7795859b
29 changed files with 502 additions and 111 deletions

View file

@ -687,6 +687,12 @@ class SqlGenerator:
if isinstance(node, exp.AggFunc):
if isinstance(node, exp.Count):
count_arg = node.this
if isinstance(count_arg, exp.Star):
node.set(
"this",
_make_case(exp.Literal.number(1)),
)
return node
if (
isinstance(count_arg, exp.Distinct)
and count_arg.expressions

View file

@ -243,6 +243,37 @@ def test_filtered_count_distinct_keeps_distinct_inside_count():
assert_valid_sql(result.sql)
def test_filtered_count_star_uses_case_one_not_case_star():
engine = make_engine(
{
"accounts": {
"name": "accounts",
"table": "public.accounts",
"grain": ["id"],
"columns": [
{"name": "id", "type": "number"},
{"name": "risk_level", "type": "string"},
],
"measures": [
{
"name": "high_risk_account_count",
"expr": "count(*)",
"filter": "risk_level = 'high'",
}
],
}
}
)
result = engine.query(
{"measures": ["accounts.high_risk_account_count"], "dimensions": []}
)
assert "THEN *" not in result.sql
assert "COUNT(CASE WHEN accounts.risk_level = 'high' THEN 1 END)" in result.sql
assert_valid_sql(result.sql)
def test_predefined_measure_via_alias_uses_real_table_and_alias_qualification():
engine = make_engine(_alias_measure_sources())
result = engine.query(