Fixing broken tests

apache · Jun 7, 2022 · 40b2bf9 · 40b2bf9
1 parent 8305fc9
commit 40b2bf9
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 13 deletions.
diff --git a/tests/integration_tests/migrations/ad07e4fdbaba_rm_time_range_endpoints_from_qc_3__test.py b/tests/integration_tests/migrations/ad07e4fdbaba_rm_time_range_endpoints_from_qc_3__test.py
@@ -15,11 +15,14 @@
 # specific language governing permissions and limitations
 # under the License.
 import json
+from importlib import import_module
 
-from superset.migrations.versions.ad07e4fdbaba_rm_time_range_endpoints_from_qc_3 import (
-    Slice,
-    upgrade_slice,
+rm_time_range_endpoints_from_qc_3 = import_module(
+    "superset.migrations.versions."
+    "2022-04-18_11-20_ad07e4fdbaba_rm_time_range_endpoints_from_qc_3",
 )
+Slice = rm_time_range_endpoints_from_qc_3.Slice
+upgrade_slice = rm_time_range_endpoints_from_qc_3.upgrade_slice
 
 sample_query_context = {
     "datasource": {"id": 27, "type": "table"},

diff --git a/.../integration_tests/migrations/f1410ed7ec95_migrate_native_filters_to_new_schema__tests.py b/.../integration_tests/migrations/f1410ed7ec95_migrate_native_filters_to_new_schema__tests.py
@@ -15,11 +15,14 @@
 # specific language governing permissions and limitations
 # under the License.
 from copy import deepcopy
+from importlib import import_module
 
-from superset.migrations.versions.f1410ed7ec95_migrate_native_filters_to_new_schema import (
-    downgrade_dashboard,
-    upgrade_dashboard,
+migrate_native_filters_to_new_schema = import_module(
+    "superset.migrations.versions."
+    "2021-04-29_15-32_f1410ed7ec95_migrate_native_filters_to_new_schema",
 )
+downgrade_dashboard = migrate_native_filters_to_new_schema.downgrade_dashboard
+upgrade_dashboard = migrate_native_filters_to_new_schema.upgrade_dashboard
 
 dashboard_v1 = {
     "native_filter_configuration": [

diff --git a/tests/integration_tests/migrations/fb13d49b72f9_better_filters__tests.py b/tests/integration_tests/migrations/fb13d49b72f9_better_filters__tests.py
@@ -15,11 +15,13 @@
 # specific language governing permissions and limitations
 # under the License.
 import json
+from importlib import import_module
 
-from superset.migrations.versions.fb13d49b72f9_better_filters import (
-    Slice,
-    upgrade_slice,
+better_filters = import_module(
+    "superset.migrations.versions." "2018-12-11_22-03_fb13d49b72f9_better_filters",
 )
+Slice = better_filters.Slice
+upgrade_slice = better_filters.upgrade_slice
 
 
 def test_upgrade_slice():

diff --git a/tests/integration_tests/migrations/fc3a3a8ff221_migrate_filter_sets_to_new_format__tests.py b/tests/integration_tests/migrations/fc3a3a8ff221_migrate_filter_sets_to_new_format__tests.py
@@ -15,12 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 from copy import deepcopy
+from importlib import import_module
 
-from superset.migrations.versions.fc3a3a8ff221_migrate_filter_sets_to_new_format import (
-    downgrade_filter_set,
-    upgrade_filter_set,
-    upgrade_select_filters,
+migrate_filter_sets_to_new_format = import_module(
+    "superset.migrations.versions."
+    "2021-04-12_12-38_fc3a3a8ff221_migrate_filter_sets_to_new_format",
 )
+downgrade_filter_set = migrate_filter_sets_to_new_format.downgrade_filter_set
+upgrade_filter_set = migrate_filter_sets_to_new_format.upgrade_filter_set
+upgrade_select_filters = migrate_filter_sets_to_new_format.upgrade_select_filters
 
 native_filters_v1 = [
     {