fix: flaky test_explore_json_async test v2 (#26106)

(cherry picked from commit 91a8b69)
apache · Dec 4, 2023 · a3212cc · a3212cc
1 parent 00db4df
commit a3212cc
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/tests/integration_tests/core_tests.py b/tests/integration_tests/core_tests.py
@@ -840,10 +840,16 @@ def test_explore_json_async(self):
         keys = list(data.keys())
 
         # If chart is cached, it will return 200, otherwise 202
-        self.assertTrue(rv.status_code in {200, 202})
-        self.assertCountEqual(
-            keys, ["channel_id", "job_id", "user_id", "status", "errors", "result_url"]
-        )
+        assert rv.status_code in {200, 202}
+        if rv.status_code == 202:
+            assert keys == [
+                "channel_id",
+                "job_id",
+                "user_id",
+                "status",
+                "errors",
+                "result_url",
+            ]
 
     @pytest.mark.usefixtures("load_birth_names_dashboard_with_slices")
     @mock.patch.dict(