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

feat(SIP-95): new endpoint for table metadata (#28122)

Browse Source
This commit is contained in:
Beto Dealmeida
2024-04-25 12:23:49 -04:00
committed by GitHub
parent 52f8734662
commit 6cf681df68
71 changed files with 1048 additions and 513 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/integration_tests/sql_validator_tests.py
View File

@@ -56,7 +56,7 @@ class TestPrestoValidator(SupersetTestCase):
sql = "SELECT 1 FROM default.notarealtable"
schema = "default"
errors = self.validator.validate(sql, schema, self.database)
errors = self.validator.validate(sql, None, schema, self.database)
self.assertEqual([], errors)
@@ -70,7 +70,7 @@ class TestPrestoValidator(SupersetTestCase):
fetch_fn.side_effect = DatabaseError("dummy db error")
with self.assertRaises(PrestoSQLValidationError):
self.validator.validate(sql, schema, self.database)
self.validator.validate(sql, None, schema, self.database)
@patch("superset.utils.core.g")
def test_validator_unexpected_error(self, flask_g):
@@ -82,7 +82,7 @@ class TestPrestoValidator(SupersetTestCase):
fetch_fn.side_effect = Exception("a mysterious failure")
with self.assertRaises(Exception):
self.validator.validate(sql, schema, self.database)
self.validator.validate(sql, None, schema, self.database)
@patch("superset.utils.core.g")
def test_validator_query_error(self, flask_g):
@@ -93,7 +93,7 @@ class TestPrestoValidator(SupersetTestCase):
fetch_fn = self.database.db_engine_spec.fetch_data
fetch_fn.side_effect = DatabaseError(self.PRESTO_ERROR_TEMPLATE)
errors = self.validator.validate(sql, schema, self.database)
errors = self.validator.validate(sql, None, schema, self.database)
self.assertEqual(1, len(errors))
@@ -105,7 +105,10 @@ class TestPostgreSQLValidator(SupersetTestCase):
mock_database = MagicMock()
annotations = PostgreSQLValidator.validate(
sql='SELECT 1, "col" FROM "table"', schema="", database=mock_database
sql='SELECT 1, "col" FROM "table"',
catalog=None,
schema="",
database=mock_database,
)
assert annotations == []
@@ -115,7 +118,10 @@ class TestPostgreSQLValidator(SupersetTestCase):
mock_database = MagicMock()
annotations = PostgreSQLValidator.validate(
sql='SELECT 1, "col"\nFROOM "table"', schema="", database=mock_database
sql='SELECT 1, "col"\nFROOM "table"',
catalog=None,
schema="",
database=mock_database,
)
assert len(annotations) == 1