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

chore: Add more database-related tests (follow up to #31948) (#32054)

Browse Source
This commit is contained in:
Vitor Avila
2025-01-31 08:36:09 -03:00
committed by GitHub
parent f73d61a597
commit 5fc11fb706
3 changed files with 76 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/unit_tests/commands/databases/tables_test.py
View File

@@ -34,13 +34,13 @@ def database_with_catalog(mocker: MockerFixture) -> MagicMock:
database = mocker.MagicMock()
database.database_name = "test_database"
database.get_all_table_names_in_schema.return_value = [
DatasourceName("table1", "schema1", "catalog1"),
DatasourceName("table2", "schema1", "catalog1"),
]
database.get_all_view_names_in_schema.return_value = [
DatasourceName("view1", "schema1", "catalog1"),
]
database.get_all_table_names_in_schema.return_value = {
("table1", "schema1", "catalog1"),
("table2", "schema1", "catalog1"),
}
database.get_all_view_names_in_schema.return_value = {
("view1", "schema1", "catalog1"),
}
DatabaseDAO = mocker.patch("superset.commands.database.tables.DatabaseDAO") # noqa: N806
DatabaseDAO.find_by_id.return_value = database
@@ -57,13 +57,13 @@ def database_without_catalog(mocker: MockerFixture) -> MagicMock:
database = mocker.MagicMock()
database.database_name = "test_database"
database.get_all_table_names_in_schema.return_value = [
DatasourceName("table1", "schema1"),
DatasourceName("table2", "schema1"),
]
database.get_all_view_names_in_schema.return_value = [
DatasourceName("view1", "schema1"),
]
database.get_all_table_names_in_schema.return_value = {
("table1", "schema1", None),
("table2", "schema1", None),
}
database.get_all_view_names_in_schema.return_value = {
("view1", "schema1", None),
}
DatabaseDAO = mocker.patch("superset.commands.database.tables.DatabaseDAO") # noqa: N806
DatabaseDAO.find_by_id.return_value = database