Update test

apache · Jul 14, 2021 · 1814f11 · 1814f11
1 parent 5abcef1
commit 1814f11
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/tests/integration_tests/strategy_tests.py → ...ion_tests/cache_warm_up_strategy_tests.py b/tests/integration_tests/strategy_tests.py → ...ion_tests/cache_warm_up_strategy_tests.py
@@ -198,7 +198,12 @@ def test_top_n_dashboards_strategy(self):
 
         strategy = TopNDashboardsStrategy(1)
         result = sorted(strategy.get_urls())
-        expected = sorted([f"{URL_PREFIX}{slc.url}" for slc in dash.slices])
+        expected = sorted(
+            [
+                f"{URL_PREFIX}/superset/warm_up_cache/?slice_id={slc.id}"
+                for slc in dash.slices
+            ]
+        )
         self.assertEqual(result, expected)
 
     def reset_tag(self, tag):
@@ -224,7 +229,12 @@ def test_dashboard_tags(self):
         # tag dashboard 'births' with `tag1`
         tag1 = get_tag("tag1", db.session, TagTypes.custom)
         dash = self.get_dash_by_slug("births")
-        tag1_urls = sorted([f"{URL_PREFIX}{slc.url}" for slc in dash.slices])
+        tag1_urls = sorted(
+            [
+                f"{URL_PREFIX}/superset/warm_up_cache/?slice_id={slc.id}"
+                for slc in dash.slices
+            ]
+        )
         tagged_object = TaggedObject(
             tag_id=tag1.id, object_id=dash.id, object_type=ObjectTypes.dashboard
         )
@@ -244,7 +254,7 @@ def test_dashboard_tags(self):
         # tag first slice
         dash = self.get_dash_by_slug("unicode-test")
         slc = dash.slices[0]
-        tag2_urls = [f"{URL_PREFIX}{slc.url}"]
+        tag2_urls = [f"{URL_PREFIX}/superset/warm_up_cache/?slice_id={slc.id}"]
         object_id = slc.id
         tagged_object = TaggedObject(
             tag_id=tag2.id, object_id=object_id, object_type=ObjectTypes.chart