From 0377c3349063986b1da170687105402b30384166 Mon Sep 17 00:00:00 2001 From: Ville Brofeldt Date: Thu, 23 Apr 2020 12:43:30 +0300 Subject: [PATCH] Add typing to test func --- tests/base_tests.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/base_tests.py b/tests/base_tests.py index 37751fbb9d7b..ebe1d9aa99f0 100644 --- a/tests/base_tests.py +++ b/tests/base_tests.py @@ -29,6 +29,7 @@ from tests.test_app import app # isort:skip from superset import db, security_manager +from superset.connectors.base.models import BaseDatasource from superset.connectors.druid.models import DruidCluster, DruidDatasource from superset.connectors.sqla.models import SqlaTable from superset.models import core as models @@ -147,7 +148,7 @@ def get_druid_ds_by_name(name: str) -> DruidDatasource: return db.session.query(DruidDatasource).filter_by(datasource_name=name).first() @staticmethod - def get_datasource_mock(): + def get_datasource_mock() -> BaseDatasource: datasource = Mock() results = Mock() results.query = Mock()