update

apache · Nov 23, 2021 · 107376f · 107376f
1 parent a97905d
commit 107376f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/integration_tests/datasets/api_tests.py b/tests/integration_tests/datasets/api_tests.py
@@ -1384,7 +1384,7 @@ def test_export_dataset_gamma(self):
         """
         Dataset API: Test export dataset has gamma
         """
-        dataset = self.insert_default_dataset()
+        dataset = self.get_fixture_datasets()[0]
 
         argument = [dataset.id]
         uri = f"api/v1/dataset/export/?q={prison.dumps(argument)}"
@@ -1393,9 +1393,6 @@ def test_export_dataset_gamma(self):
         rv = self.client.get(uri)
         assert rv.status_code == 401
 
-        db.session.delete(dataset)
-        db.session.commit()
-
     @patch.dict(
         "superset.extensions.feature_flag_manager._feature_flags",
         {"VERSIONED_EXPORT": True},