fix: fix select statement, remove unnecessary code in testing/__init__()

martynia · Nov 1, 2024 · 85da2cc · 85da2cc
1 parent 4787374
commit 85da2cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/diracx-routers/src/diracx/routers/pilot_logging/remote_logger.py b/diracx-routers/src/diracx/routers/pilot_logging/remote_logger.py
@@ -41,8 +41,8 @@ async def send_message(
 
     async with db.engine_context():
         async with db:
-            stmt = select(
-                (PilotAgents.PilotID).where(PilotAgents.PilotStamp == data.pilot_stamp)
+            stmt = select(PilotAgents.PilotID).where(
+                PilotAgents.PilotStamp == data.pilot_stamp
             )
             pilot_id = (await db.conn.execute(stmt)).scalar_one()
 

diff --git a/diracx-testing/src/diracx/testing/__init__.py b/diracx-testing/src/diracx/testing/__init__.py
@@ -211,8 +211,7 @@ def enrich_tokens(access_payload: dict, refresh_payload: dict):
                 test_dev_settings,
             ],
             database_urls=database_urls,
-            os_database_conn_kwargs=os_database_conn_kwargs
-            | {"PilotLogsDB": {"sqlalchemy_dsn": "sqlite+aiosqlite:///:memory:"}},
+            os_database_conn_kwargs=os_database_conn_kwargs,
             config_source=ConfigSource.create_from_url(
                 backend_url=f"git+file://{with_config_repo}"
             ),