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

chore: upgrade black (#19410)

Browse Source
This commit is contained in:
Ville Brofeldt
2022-03-29 20:03:09 +03:00
committed by GitHub
parent 816a2c3e1e
commit a619cb4ea9
204 changed files with 2125 additions and 608 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/integration_tests/model_tests.py
View File

@@ -355,13 +355,15 @@ class TestDatabaseModel(SupersetTestCase):
if main_db.backend == "mysql":
main_db.get_df("USE superset; SELECT 1", username=test_username)
mocked_get_sqla_engine.assert_called_with(
schema=None, user_name="test_username_param",
schema=None,
user_name="test_username_param",
)
@mock.patch("superset.models.core.create_engine")
def test_get_sqla_engine(self, mocked_create_engine):
model = Database(
database_name="test_database", sqlalchemy_uri="mysql://root@localhost",
database_name="test_database",
sqlalchemy_uri="mysql://root@localhost",
)
model.db_engine_spec.get_dbapi_exception_mapping = mock.Mock(
return_value={Exception: SupersetException}
@@ -568,7 +570,9 @@ class TestSqlaTableModel(SupersetTestCase):
slc = (
metadata_db.session.query(Slice)
.filter_by(
datasource_id=tbl.id, datasource_type=tbl.type, slice_name="Genders",
datasource_id=tbl.id,
datasource_type=tbl.type,
slice_name="Genders",
)
.first()
)