Skip to content

Commit b8be1dd

Browse files
committed
feat: add support for PilotLogsDB in tests
1 parent 02c24a9 commit b8be1dd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

diracx-testing/src/diracx/testing/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,7 @@ def __init__(
145145
from diracx.core.extensions import select_from_extension
146146
from diracx.core.settings import ServiceSettingsBase
147147
from diracx.db.sql.utils import BaseSQLDB
148+
from diracx.db.os.utils import BaseOSDB
148149
from diracx.routers import create_app_inner
149150
from diracx.routers.access_policies import BaseAccessPolicy
150151

@@ -186,6 +187,7 @@ def enrich_tokens(access_payload: dict, refresh_payload: dict):
186187
database_urls=database_urls,
187188
os_database_conn_kwargs={
188189
# TODO: JobParametersDB
190+
"PilotLogsDB": {}
189191
},
190192
config_source=ConfigSource.create_from_url(
191193
backend_url=f"git+file://{with_config_repo}"
@@ -198,14 +200,14 @@ def enrich_tokens(access_payload: dict, refresh_payload: dict):
198200
for obj in self.all_dependency_overrides:
199201
assert issubclass(
200202
obj.__self__,
201-
(ServiceSettingsBase, BaseSQLDB, ConfigSource, BaseAccessPolicy),
203+
(ServiceSettingsBase, BaseSQLDB, BaseOSDB, ConfigSource, BaseAccessPolicy),
202204
), obj
203205

204206
self.all_lifetime_functions = self.app.lifetime_functions[:]
205207
self.app.lifetime_functions = []
206208
for obj in self.all_lifetime_functions:
207209
assert isinstance(
208-
obj.__self__, (ServiceSettingsBase, BaseSQLDB, ConfigSource)
210+
obj.__self__, (ServiceSettingsBase, BaseSQLDB, BaseOSDB, ConfigSource)
209211
), obj
210212

211213
@contextlib.contextmanager

0 commit comments

Comments
 (0)