mirror of
https://github.com/apache/superset.git
synced 2024-09-12 08:39:45 -04:00
* add BUSINESS_TYPE_ADDONS to config with example callback * Removing uneeded whitespace * [Work in progress] Modifying cidr function to allow for single ip and adding port outline * Added test REST endpoint, added some more ports I've thrown in a test.py script as well that will try to connect to the business_type endpoint. * Moving code from config.py into the business api Very simple api is exposed that will allow someone to call a checkport endpoint and get back a response. * Removing commented out bits. * Adding fucntion dict back to the config * Moving business_type endpoint to charts * Adding schema for get endpoint * Removing imports, updating docstring, fixing typo Just some small changes as described in the title. I've updated the test.py as well so it functions with the endpoint changes. * Adding translation dict * Fixing ops * Adding check for list * Modifying changes to add quotes where needed Also changed BusinessTypeResponse to resp. * Adding in some code to call the filter config If a column starts with "cidr_" it will call the code in config.py to try to translate the filter. Nothing is changed in the JSON being executed, just some information is dumped to console. * Porting Ryan's changes * Adding migration script (as per Ryan's PR) * Fixing typo * Prettier fixes * [CLDN-1043] Adding rough version of filter changes for business types * fix down migration * Fixing bugs after merge * adding functionality to appy filters in back end * Fixing linting issues * fix down revision * Changing conversion callback to handle multiple values at once * Adding string representation of values * Code cleanup plus fixing debouce to only be called once for each entry * Removing non needed logginh * Changing operator list to use sting values * Using text value operators * Removing clear operator call * Moving business type endpoints * fix down revision * Adding port functions * update migration * fix bad rebase and add ff * implement validator * dont add invalid values to response * [CLDN-1205] Added a new exception type for a business type translation error. Added the error message in the display_value field within the business type response. Modified the IP and Port business types to populate the error message field in the response if an error occurs * [CLDN-1205] Added meaningful error message for port translation errors * Removing status field from businesstype Response and adding in error message * [CLDN-1205] Added check to make sure the port business type is within the valid range of ports, if it is not, it will populate the error message * [CLDN-1205] Fixed the if statement that checks to see if the string_value is in the valid range of port numbers. It did not corrently verify this before now. * [CLDN-1205] Fixed an error where it was trying to use string_value in <= statements. I just casted string_value to an integer if it is numeric, which allows <= operators to be used on it * [CLDN-1207] Added unit tests for the cidr_func and port_translation_func functions which are located in /superset/config.py * [CLDN-1207] removed the assertRaises line as it does not work with the cidr_func and port_translation_func functions * [CLDN-1207] Added the skeleton of the test_cidr_translate_filter_func unit test, still need to update what the expected response from the function will be. * [CLDN-1207] Added the remainder of the back-end unit tests for the business types * [CLDN-1207] Fixed the syntax error which caused the test_cidr_translate_filter_func_NOT_IN_double unit test to fail * [CLDN-1207] Removed the logging that was added for debugging purposes * [CLDN-1207] Formatted the commands_tests.py file to make it nicer to look at/read through * [CLDN-1207] Fixed the code so that it conformed to the pylint requirements (i.e., pylint no longer complains about the code in commands_tests.py) * [CLDN-1207] Modified some of the docstrings so they made better use of the 100 character per line, line limit * [CLDN-1207] Added the beginnings of the unit tests for the business types API * [CLDN-1207] Added a comment to the top of the commands_tests.py file explaining how to run the unit tests. This prevents the next person who tries to run them from having to waste time trying the different forms of testing that Superset supports (e.g., pytest, tox, etc.) * [CLDN-1207] Added a grammar fix to the comments describing how to run the unit tests * [CLDN-1207] Modified the description of the business_type API endpoints as they did not represent what the API was actually doing * [CLDN-1207] Added further instructions on how to run the unit tests that are within the business_type/api_tests.py file * add request validation * disable request if business type missing * [CLDN-1207] Unit tests for the business type API are now working, however, they need to be modified to make use of @mock as we don't want to have to run the server to be able to run the unit tests * Removing businesss types deffinitons from config * Adding select to only show valid business types * Fixed Enzyme tests * Added scalfolding for selecting filter dropdown * Adding intigration tests * fix revision * fix typos and unnecessary requests * break out useBusinessTypes * Added front-end RTL unit tests for the business type API endpoint * Fixed error from unit tests * Added a unit test to ensure the operator list is updated after a business type API response is received * Removing elect compoenet for business types * Adding feature flag and allowing saving when no business type present * fixing useEffect hooks * Adding feature flag to model * Changing behavior such that an empty string returns a default response * add form validation * Modified comments in unit test as command to run test has changed * Modified comments in unit test as filename to run test has changed * Modified the api_tests.py file to conform to the linting requirements * Changed the name of one of the tests to reflect what the test is actually testing * Added cypress back to the package.json * Added informative comments * Updated comments in files as well as removed imports which were not being used * Changes made by npm run prettier * Fixed spelling mistakes * Updated models.py to remove placeholder comments used in development * Added feature flag mocking in unit test * Fixing open api failure * Fixing business types to pass unit tests * Reverting unsafe connections back to false * Removing print statement * Adding business tpye to export test * setting default feature flag to false for business type * Reverting pre commit * Reverting pre commit and running pre commit * Reverting pre commit and running pre commit * Fixing formatting * Adding license * Fixing Linting * Protecting api enpoints * updating model * Fixing code path when business type exists * Linting * Linting * Fixing linting * Fixing spelling * Fixing schemas * Fixing app import * fixing item render * Added RTL test to make sure business type operator list is updated after API response * Fixing linting * fix migration * Changing unit tests * Fixing import and DB migration after rebase * Renaming to advanced types * Fixing Linting * More renaming * Removing uneeded change * Fixing linting and test errors * Removing unused imports * linting * Adding more detailed name for migration * Moving files to plugins * more renaming * Fixing schema name * Disabling feature flag that should not be enabled by default * Adding extra cehck * NameChange * formatting * Fixing equals check * Moveing all advanced type classes and types to one file, and converting tests to functional * Adding advanced type to test and fix linitng Co-authored-by: Ville Brofeldt <ville.v.brofeldt@gmail.com> Co-authored-by: Dan Parent <daniel.parent@cse-cst.gc.ca> Co-authored-by: GITHUB_USERNAME <EMAIL> Co-authored-by: cccs-Dustin <96579982+cccs-Dustin@users.noreply.github.com>
522 lines
19 KiB
Python
522 lines
19 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
from operator import itemgetter
|
|
from typing import Any, List
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from superset import db, security_manager
|
|
from superset.commands.exceptions import CommandInvalidError
|
|
from superset.commands.importers.exceptions import IncorrectVersionError
|
|
from superset.connectors.sqla.models import SqlaTable
|
|
from superset.databases.commands.importers.v1 import ImportDatabasesCommand
|
|
from superset.datasets.commands.exceptions import DatasetNotFoundError
|
|
from superset.datasets.commands.export import ExportDatasetsCommand
|
|
from superset.datasets.commands.importers import v0, v1
|
|
from superset.models.core import Database
|
|
from superset.utils.core import get_example_default_schema
|
|
from superset.utils.database import get_example_database
|
|
from tests.integration_tests.base_tests import SupersetTestCase
|
|
from tests.integration_tests.fixtures.energy_dashboard import (
|
|
load_energy_table_data,
|
|
load_energy_table_with_slice,
|
|
)
|
|
from tests.integration_tests.fixtures.importexport import (
|
|
database_config,
|
|
database_metadata_config,
|
|
dataset_cli_export,
|
|
dataset_config,
|
|
dataset_metadata_config,
|
|
dataset_ui_export,
|
|
)
|
|
from tests.integration_tests.fixtures.world_bank_dashboard import (
|
|
load_world_bank_dashboard_with_slices,
|
|
load_world_bank_data,
|
|
)
|
|
|
|
|
|
class TestExportDatasetsCommand(SupersetTestCase):
|
|
@patch("superset.security.manager.g")
|
|
@pytest.mark.usefixtures("load_energy_table_with_slice")
|
|
def test_export_dataset_command(self, mock_g):
|
|
mock_g.user = security_manager.find_user("admin")
|
|
|
|
example_db = get_example_database()
|
|
example_dataset = _get_table_from_list_by_name(
|
|
"energy_usage", example_db.tables
|
|
)
|
|
command = ExportDatasetsCommand([example_dataset.id])
|
|
contents = dict(command.run())
|
|
|
|
assert list(contents.keys()) == [
|
|
"metadata.yaml",
|
|
"datasets/examples/energy_usage.yaml",
|
|
"databases/examples.yaml",
|
|
]
|
|
|
|
metadata = yaml.safe_load(contents["datasets/examples/energy_usage.yaml"])
|
|
|
|
# sort columns for deterministc comparison
|
|
metadata["columns"] = sorted(metadata["columns"], key=itemgetter("column_name"))
|
|
metadata["metrics"] = sorted(metadata["metrics"], key=itemgetter("metric_name"))
|
|
|
|
# types are different depending on the backend
|
|
type_map = {
|
|
column.column_name: str(column.type) for column in example_dataset.columns
|
|
}
|
|
|
|
assert metadata == {
|
|
"cache_timeout": None,
|
|
"columns": [
|
|
{
|
|
"column_name": "source",
|
|
"description": None,
|
|
"expression": "",
|
|
"filterable": True,
|
|
"groupby": True,
|
|
"is_active": True,
|
|
"is_dttm": False,
|
|
"python_date_format": None,
|
|
"type": type_map["source"],
|
|
"advanced_data_type": None,
|
|
"verbose_name": None,
|
|
"extra": None,
|
|
},
|
|
{
|
|
"column_name": "target",
|
|
"description": None,
|
|
"expression": "",
|
|
"filterable": True,
|
|
"groupby": True,
|
|
"is_active": True,
|
|
"is_dttm": False,
|
|
"python_date_format": None,
|
|
"type": type_map["target"],
|
|
"advanced_data_type": None,
|
|
"verbose_name": None,
|
|
"extra": None,
|
|
},
|
|
{
|
|
"column_name": "value",
|
|
"description": None,
|
|
"expression": "",
|
|
"filterable": True,
|
|
"groupby": True,
|
|
"is_active": True,
|
|
"is_dttm": False,
|
|
"python_date_format": None,
|
|
"type": type_map["value"],
|
|
"advanced_data_type": None,
|
|
"verbose_name": None,
|
|
"extra": None,
|
|
},
|
|
],
|
|
"database_uuid": str(example_db.uuid),
|
|
"default_endpoint": None,
|
|
"description": "Energy consumption",
|
|
"extra": None,
|
|
"fetch_values_predicate": None,
|
|
"filter_select_enabled": False,
|
|
"main_dttm_col": None,
|
|
"metrics": [
|
|
{
|
|
"d3format": None,
|
|
"description": None,
|
|
"expression": "COUNT(*)",
|
|
"extra": None,
|
|
"metric_name": "count",
|
|
"metric_type": "count",
|
|
"verbose_name": "COUNT(*)",
|
|
"warning_text": None,
|
|
},
|
|
{
|
|
"d3format": None,
|
|
"description": None,
|
|
"expression": "SUM(value)",
|
|
"extra": None,
|
|
"metric_name": "sum__value",
|
|
"metric_type": None,
|
|
"verbose_name": None,
|
|
"warning_text": None,
|
|
},
|
|
],
|
|
"offset": 0,
|
|
"params": None,
|
|
"schema": get_example_default_schema(),
|
|
"sql": None,
|
|
"table_name": "energy_usage",
|
|
"template_params": None,
|
|
"uuid": str(example_dataset.uuid),
|
|
"version": "1.0.0",
|
|
}
|
|
|
|
@patch("superset.security.manager.g")
|
|
def test_export_dataset_command_no_access(self, mock_g):
|
|
"""Test that users can't export datasets they don't have access to"""
|
|
mock_g.user = security_manager.find_user("gamma")
|
|
|
|
example_db = get_example_database()
|
|
example_dataset = example_db.tables[0]
|
|
command = ExportDatasetsCommand([example_dataset.id])
|
|
contents = command.run()
|
|
with self.assertRaises(DatasetNotFoundError):
|
|
next(contents)
|
|
|
|
@patch("superset.security.manager.g")
|
|
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([-1])
|
|
contents = command.run()
|
|
with self.assertRaises(DatasetNotFoundError):
|
|
next(contents)
|
|
|
|
@patch("superset.security.manager.g")
|
|
@pytest.mark.usefixtures("load_energy_table_with_slice")
|
|
def test_export_dataset_command_key_order(self, mock_g):
|
|
"""Test that they keys in the YAML have the same order as export_fields"""
|
|
mock_g.user = security_manager.find_user("admin")
|
|
|
|
example_db = get_example_database()
|
|
example_dataset = _get_table_from_list_by_name(
|
|
"energy_usage", example_db.tables
|
|
)
|
|
command = ExportDatasetsCommand([example_dataset.id])
|
|
contents = dict(command.run())
|
|
|
|
metadata = yaml.safe_load(contents["datasets/examples/energy_usage.yaml"])
|
|
assert list(metadata.keys()) == [
|
|
"table_name",
|
|
"main_dttm_col",
|
|
"description",
|
|
"default_endpoint",
|
|
"offset",
|
|
"cache_timeout",
|
|
"schema",
|
|
"sql",
|
|
"params",
|
|
"template_params",
|
|
"filter_select_enabled",
|
|
"fetch_values_predicate",
|
|
"extra",
|
|
"uuid",
|
|
"metrics",
|
|
"columns",
|
|
"version",
|
|
"database_uuid",
|
|
]
|
|
|
|
@patch("superset.security.manager.g")
|
|
@pytest.mark.usefixtures("load_energy_table_with_slice")
|
|
def test_export_dataset_command_no_related(self, mock_g):
|
|
"""
|
|
Test that only datasets are exported when export_related=False.
|
|
"""
|
|
mock_g.user = security_manager.find_user("admin")
|
|
|
|
example_db = get_example_database()
|
|
example_dataset = _get_table_from_list_by_name(
|
|
"energy_usage", example_db.tables
|
|
)
|
|
command = ExportDatasetsCommand([example_dataset.id], export_related=False)
|
|
contents = dict(command.run())
|
|
|
|
assert list(contents.keys()) == [
|
|
"metadata.yaml",
|
|
"datasets/examples/energy_usage.yaml",
|
|
]
|
|
|
|
|
|
class TestImportDatasetsCommand(SupersetTestCase):
|
|
@pytest.mark.usefixtures("load_world_bank_dashboard_with_slices")
|
|
def test_import_v0_dataset_cli_export(self):
|
|
num_datasets = db.session.query(SqlaTable).count()
|
|
|
|
contents = {
|
|
"20201119_181105.yaml": yaml.safe_dump(dataset_cli_export),
|
|
}
|
|
command = v0.ImportDatasetsCommand(contents)
|
|
command.run()
|
|
|
|
new_num_datasets = db.session.query(SqlaTable).count()
|
|
assert new_num_datasets == num_datasets + 1
|
|
|
|
dataset = (
|
|
db.session.query(SqlaTable).filter_by(table_name="birth_names_2").one()
|
|
)
|
|
assert (
|
|
dataset.params
|
|
== '{"remote_id": 3, "database_name": "examples", "import_time": 1604342885}'
|
|
)
|
|
assert len(dataset.metrics) == 2
|
|
assert dataset.main_dttm_col == "ds"
|
|
assert dataset.filter_select_enabled
|
|
dataset.columns.sort(key=lambda obj: obj.column_name)
|
|
expected_columns = [
|
|
"num_california",
|
|
"ds",
|
|
"state",
|
|
"gender",
|
|
"name",
|
|
"num_boys",
|
|
"num_girls",
|
|
"num",
|
|
]
|
|
expected_columns.sort()
|
|
assert [col.column_name for col in dataset.columns] == expected_columns
|
|
|
|
db.session.delete(dataset)
|
|
db.session.commit()
|
|
|
|
@pytest.mark.usefixtures("load_world_bank_dashboard_with_slices")
|
|
def test_import_v0_dataset_ui_export(self):
|
|
num_datasets = db.session.query(SqlaTable).count()
|
|
|
|
contents = {
|
|
"20201119_181105.yaml": yaml.safe_dump(dataset_ui_export),
|
|
}
|
|
command = v0.ImportDatasetsCommand(contents)
|
|
command.run()
|
|
|
|
new_num_datasets = db.session.query(SqlaTable).count()
|
|
assert new_num_datasets == num_datasets + 1
|
|
|
|
dataset = (
|
|
db.session.query(SqlaTable).filter_by(table_name="birth_names_2").one()
|
|
)
|
|
assert (
|
|
dataset.params
|
|
== '{"remote_id": 3, "database_name": "examples", "import_time": 1604342885}'
|
|
)
|
|
assert len(dataset.metrics) == 2
|
|
assert dataset.main_dttm_col == "ds"
|
|
assert dataset.filter_select_enabled
|
|
assert set(col.column_name for col in dataset.columns) == {
|
|
"num_california",
|
|
"ds",
|
|
"state",
|
|
"gender",
|
|
"name",
|
|
"num_boys",
|
|
"num_girls",
|
|
"num",
|
|
}
|
|
|
|
db.session.delete(dataset)
|
|
db.session.commit()
|
|
|
|
@patch("superset.datasets.commands.importers.v1.utils.g")
|
|
@pytest.mark.usefixtures("load_energy_table_with_slice")
|
|
def test_import_v1_dataset(self, mock_g):
|
|
"""Test that we can import a dataset"""
|
|
mock_g.user = security_manager.find_user("admin")
|
|
contents = {
|
|
"metadata.yaml": yaml.safe_dump(dataset_metadata_config),
|
|
"databases/imported_database.yaml": yaml.safe_dump(database_config),
|
|
"datasets/imported_dataset.yaml": yaml.safe_dump(dataset_config),
|
|
}
|
|
command = v1.ImportDatasetsCommand(contents)
|
|
command.run()
|
|
|
|
dataset = (
|
|
db.session.query(SqlaTable).filter_by(uuid=dataset_config["uuid"]).one()
|
|
)
|
|
assert dataset.table_name == "imported_dataset"
|
|
assert dataset.main_dttm_col is None
|
|
assert dataset.description == "This is a dataset that was exported"
|
|
assert dataset.default_endpoint == ""
|
|
assert dataset.offset == 66
|
|
assert dataset.cache_timeout == 55
|
|
assert dataset.schema == ""
|
|
assert dataset.sql == ""
|
|
assert dataset.params is None
|
|
assert dataset.template_params == "{}"
|
|
assert dataset.filter_select_enabled
|
|
assert dataset.fetch_values_predicate is None
|
|
assert (
|
|
dataset.extra
|
|
== '{"certification": {"certified_by": "Data Platform Team", "details": "This table is the source of truth."}, "warning_markdown": "This is a warning."}'
|
|
)
|
|
|
|
# user should be included as one of the owners
|
|
assert dataset.owners == [mock_g.user]
|
|
|
|
# database is also imported
|
|
assert str(dataset.database.uuid) == "b8a1ccd3-779d-4ab7-8ad8-9ab119d7fe89"
|
|
|
|
assert len(dataset.metrics) == 1
|
|
metric = dataset.metrics[0]
|
|
assert metric.metric_name == "count"
|
|
assert metric.verbose_name == ""
|
|
assert metric.metric_type is None
|
|
assert metric.expression == "count(1)"
|
|
assert metric.description is None
|
|
assert metric.d3format is None
|
|
assert metric.extra == "{}"
|
|
assert metric.warning_text is None
|
|
|
|
assert len(dataset.columns) == 1
|
|
column = dataset.columns[0]
|
|
assert column.column_name == "cnt"
|
|
assert column.verbose_name == "Count of something"
|
|
assert not column.is_dttm
|
|
assert column.is_active # imported columns are set to active
|
|
assert column.type == "NUMBER"
|
|
assert not column.groupby
|
|
assert column.filterable
|
|
assert column.expression == ""
|
|
assert column.description is None
|
|
assert column.python_date_format is None
|
|
|
|
dataset.owners = []
|
|
dataset.database.owners = []
|
|
db.session.delete(dataset)
|
|
db.session.delete(dataset.database)
|
|
db.session.commit()
|
|
|
|
def test_import_v1_dataset_multiple(self):
|
|
"""Test that a dataset can be imported multiple times"""
|
|
contents = {
|
|
"metadata.yaml": yaml.safe_dump(dataset_metadata_config),
|
|
"databases/imported_database.yaml": yaml.safe_dump(database_config),
|
|
"datasets/imported_dataset.yaml": yaml.safe_dump(dataset_config),
|
|
}
|
|
command = v1.ImportDatasetsCommand(contents, overwrite=True)
|
|
command.run()
|
|
command.run()
|
|
dataset = (
|
|
db.session.query(SqlaTable).filter_by(uuid=dataset_config["uuid"]).one()
|
|
)
|
|
assert dataset.table_name == "imported_dataset"
|
|
|
|
# test that columns and metrics sync, ie, old ones not the import
|
|
# are removed
|
|
new_config = dataset_config.copy()
|
|
new_config["metrics"][0]["metric_name"] = "count2"
|
|
new_config["columns"][0]["column_name"] = "cnt2"
|
|
contents = {
|
|
"metadata.yaml": yaml.safe_dump(dataset_metadata_config),
|
|
"databases/imported_database.yaml": yaml.safe_dump(database_config),
|
|
"datasets/imported_dataset.yaml": yaml.safe_dump(new_config),
|
|
}
|
|
command = v1.ImportDatasetsCommand(contents, overwrite=True)
|
|
command.run()
|
|
dataset = (
|
|
db.session.query(SqlaTable).filter_by(uuid=dataset_config["uuid"]).one()
|
|
)
|
|
assert len(dataset.metrics) == 1
|
|
assert dataset.metrics[0].metric_name == "count2"
|
|
assert len(dataset.columns) == 1
|
|
assert dataset.columns[0].column_name == "cnt2"
|
|
|
|
db.session.delete(dataset)
|
|
db.session.delete(dataset.database)
|
|
db.session.commit()
|
|
|
|
def test_import_v1_dataset_validation(self):
|
|
"""Test different validations applied when importing a dataset"""
|
|
# metadata.yaml must be present
|
|
contents = {
|
|
"datasets/imported_dataset.yaml": yaml.safe_dump(dataset_config),
|
|
}
|
|
command = v1.ImportDatasetsCommand(contents)
|
|
with pytest.raises(IncorrectVersionError) as excinfo:
|
|
command.run()
|
|
assert str(excinfo.value) == "Missing metadata.yaml"
|
|
|
|
# version should be 1.0.0
|
|
contents["metadata.yaml"] = yaml.safe_dump(
|
|
{
|
|
"version": "2.0.0",
|
|
"type": "SqlaTable",
|
|
"timestamp": "2020-11-04T21:27:44.423819+00:00",
|
|
}
|
|
)
|
|
command = v1.ImportDatasetsCommand(contents)
|
|
with pytest.raises(IncorrectVersionError) as excinfo:
|
|
command.run()
|
|
assert str(excinfo.value) == "Must be equal to 1.0.0."
|
|
|
|
# type should be SqlaTable
|
|
contents["metadata.yaml"] = yaml.safe_dump(database_metadata_config)
|
|
command = v1.ImportDatasetsCommand(contents)
|
|
with pytest.raises(CommandInvalidError) as excinfo:
|
|
command.run()
|
|
assert str(excinfo.value) == "Error importing dataset"
|
|
assert excinfo.value.normalized_messages() == {
|
|
"metadata.yaml": {"type": ["Must be equal to SqlaTable."]}
|
|
}
|
|
|
|
# must also validate databases
|
|
broken_config = database_config.copy()
|
|
del broken_config["database_name"]
|
|
contents["metadata.yaml"] = yaml.safe_dump(dataset_metadata_config)
|
|
contents["databases/imported_database.yaml"] = yaml.safe_dump(broken_config)
|
|
command = v1.ImportDatasetsCommand(contents)
|
|
with pytest.raises(CommandInvalidError) as excinfo:
|
|
command.run()
|
|
assert str(excinfo.value) == "Error importing dataset"
|
|
assert excinfo.value.normalized_messages() == {
|
|
"databases/imported_database.yaml": {
|
|
"database_name": ["Missing data for required field."],
|
|
}
|
|
}
|
|
|
|
def test_import_v1_dataset_existing_database(self):
|
|
"""Test that a dataset can be imported when the database already exists"""
|
|
# first import database...
|
|
contents = {
|
|
"metadata.yaml": yaml.safe_dump(database_metadata_config),
|
|
"databases/imported_database.yaml": yaml.safe_dump(database_config),
|
|
}
|
|
command = ImportDatabasesCommand(contents)
|
|
command.run()
|
|
|
|
database = (
|
|
db.session.query(Database).filter_by(uuid=database_config["uuid"]).one()
|
|
)
|
|
assert len(database.tables) == 0
|
|
|
|
# ...then dataset
|
|
contents = {
|
|
"metadata.yaml": yaml.safe_dump(dataset_metadata_config),
|
|
"datasets/imported_dataset.yaml": yaml.safe_dump(dataset_config),
|
|
"databases/imported_database.yaml": yaml.safe_dump(database_config),
|
|
}
|
|
command = v1.ImportDatasetsCommand(contents, overwrite=True)
|
|
command.run()
|
|
|
|
database = (
|
|
db.session.query(Database).filter_by(uuid=database_config["uuid"]).one()
|
|
)
|
|
assert len(database.tables) == 1
|
|
|
|
database.tables[0].owners = []
|
|
database.owners = []
|
|
db.session.delete(database.tables[0])
|
|
db.session.delete(database)
|
|
db.session.commit()
|
|
|
|
|
|
def _get_table_from_list_by_name(name: str, tables: List[Any]):
|
|
for table in tables:
|
|
if table.table_name == name:
|
|
return table
|
|
raise ValueError(f"Table {name} does not exists in database")
|