diff --git a/diracx-testing/src/diracx/testing/__init__.py b/diracx-testing/src/diracx/testing/__init__.py index 9779cafa..90b0498b 100644 --- a/diracx-testing/src/diracx/testing/__init__.py +++ b/diracx-testing/src/diracx/testing/__init__.py @@ -144,8 +144,8 @@ def __init__( from diracx.core.config import ConfigSource from diracx.core.extensions import select_from_extension from diracx.core.settings import ServiceSettingsBase - from diracx.db.sql.utils import BaseSQLDB from diracx.db.os.utils import BaseOSDB + from diracx.db.sql.utils import BaseSQLDB from diracx.routers import create_app_inner from diracx.routers.access_policies import BaseAccessPolicy @@ -200,7 +200,13 @@ def enrich_tokens(access_payload: dict, refresh_payload: dict): for obj in self.all_dependency_overrides: assert issubclass( obj.__self__, - (ServiceSettingsBase, BaseSQLDB, BaseOSDB, ConfigSource, BaseAccessPolicy), + ( + ServiceSettingsBase, + BaseSQLDB, + BaseOSDB, + ConfigSource, + BaseAccessPolicy, + ), ), obj self.all_lifetime_functions = self.app.lifetime_functions[:]