fix: Adjust viz migrations to also migrate the queries object (#33285)

Co-authored-by: Michael S. Molina <michael.s.molina@gmail.com>
Co-authored-by: Michael S. Molina <70410625+michael-s-molina@users.noreply.github.com>
(cherry picked from commit 57183da315)
This commit is contained in:
Luiz Otavio
2025-05-26 15:00:07 -03:00
committed by Michael Molina
parent 685b259f6f
commit cfba5cdc57
11 changed files with 1887 additions and 107 deletions

View File

@@ -20,6 +20,7 @@ from superset.migrations.shared.migrate_viz import MigrateViz
from superset.utils import json
TIMESERIES_SOURCE_FORM_DATA: dict[str, Any] = {
"datasource": "1__table",
"bottom_margin": 20,
"comparison_type": "absolute",
"contribution": True,
@@ -42,6 +43,7 @@ TIMESERIES_SOURCE_FORM_DATA: dict[str, Any] = {
}
TIMESERIES_TARGET_FORM_DATA: dict[str, Any] = {
"datasource": "1__table",
"comparison_type": "difference",
"contributionMode": "row",
"logAxis": True,
@@ -75,7 +77,7 @@ def migrate_and_assert(
viz_type=cls.source_viz_type,
datasource_type="table",
params=dumped_form_data,
query_context=f'{{"form_data": {dumped_form_data}}}',
query_context=f'{{"form_data": {dumped_form_data}, "queries": []}}',
)
# upgrade
@@ -83,6 +85,7 @@ def migrate_and_assert(
# verify form_data
new_form_data = json.loads(slc.params)
new_form_data.pop("queries_bak", None)
assert new_form_data == target
assert new_form_data["form_data_bak"] == source