QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-12 19:35: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 Ville Brofeldt
parent 1edd5f1343
commit a70f4dc52f
204 changed files with 2161 additions and 604 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}
@@ -541,7 +543,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()
)