QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 08:04:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: create base class for export commands (#11463)

Browse Source 
* Add UUID to saved_query

* Reuse function from previous migration

* Point to new head

* feat: add backend to export saved queries using new format

* Rename ImportMixin to ImportExportMixin

* Create base class for exports

* Add saved queries as well

* Add constant, small fixes

* Fix wrong import

* Fix lint
This commit is contained in:
Beto Dealmeida
2020-10-30 11:52:11 -07:00
committed by GitHub
parent ca40877640
commit fbcfaacda3
12 changed files with 199 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/datasets/commands_tests.py
View File

@@ -35,10 +35,11 @@ class TestExportDatasetsCommand(SupersetTestCase):
example_db = get_example_database()
example_dataset = example_db.tables[0]
command = ExportDatasetsCommand(dataset_ids=[example_dataset.id])
command = ExportDatasetsCommand([example_dataset.id])
contents = dict(command.run())
assert list(contents.keys()) == [
"metadata.yaml",
"datasets/examples/energy_usage.yaml",
"databases/examples.yaml",
]
@@ -140,7 +141,7 @@ class TestExportDatasetsCommand(SupersetTestCase):
example_db = get_example_database()
example_dataset = example_db.tables[0]
command = ExportDatasetsCommand(dataset_ids=[example_dataset.id])
command = ExportDatasetsCommand([example_dataset.id])
contents = command.run()
with self.assertRaises(DatasetNotFoundError):
next(contents)
@@ -149,7 +150,7 @@ class TestExportDatasetsCommand(SupersetTestCase):
def test_export_dataset_command_invalid_dataset(self, mock_g):
"""Test that an error is raised when exporting an invalid dataset"""
mock_g.user = security_manager.find_user("admin")
command = ExportDatasetsCommand(dataset_ids=[-1])
command = ExportDatasetsCommand([-1])
contents = command.run()
with self.assertRaises(DatasetNotFoundError):
next(contents)
@@ -161,7 +162,7 @@ class TestExportDatasetsCommand(SupersetTestCase):
example_db = get_example_database()
example_dataset = example_db.tables[0]
command = ExportDatasetsCommand(dataset_ids=[example_dataset.id])
command = ExportDatasetsCommand([example_dataset.id])
contents = dict(command.run())
metadata = yaml.safe_load(contents["datasets/examples/energy_usage.yaml"])