added DB test

Author: sanderegg

scripts/maintenance/computational-clusters/autoscaled_monitor/cli.py

@@ -50,7 +50,7 @@ def _parse_environment(deploy_config: Path) -> dict[str, str | None]:
 def main(
     deploy_config: Annotated[
         Path, typer.Option(help="path to the deploy configuration")
-    ]
+    ],
 ):
     """Manages external clusters"""
 
@@ -157,5 +157,11 @@ def trigger_cluster_termination(
     asyncio.run(api.trigger_cluster_termination(state, user_id, wallet_id))
 
 
+@app.command()
+def test_database_connection() -> None:
+    """this will check the connection to simcore database is ready"""
+    asyncio.run(api.test_database_connection(state))
+
+
 if __name__ == "__main__":
     app()

scripts/maintenance/computational-clusters/autoscaled_monitor/core.py

@@ -638,3 +638,7 @@ async def trigger_cluster_termination(
         )
     else:
         rich.print("not deleting anything")
+
+
+async def test_database_connection(state: AppState) -> None:
+    await db.test_db_connection(state)

scripts/maintenance/computational-clusters/autoscaled_monitor/db.py

@@ -57,6 +57,23 @@ async def abort_job_in_db(
         rich.print(f"set comp_tasks for {project_id=}/{node_id=} set to ABORTED")
 
 
+async def test_db_connection(state: AppState) -> bool:
+    try:
+        async with contextlib.AsyncExitStack() as stack:
+            engine = await stack.enter_async_context(db_engine(state))
+            db_connection = await stack.enter_async_context(engine.connect())
+            # Perform a simple query to test the connection
+            result = await db_connection.execute(sa.text("SELECT 1"))
+            result.one()
+            rich.print(
+                "[green]Database connection test completed successfully![/green]"
+            )
+            return True
+    except Exception as e:  # pylint: disable=broad-exception-caught
+        rich.print(f"[red]Database connection test failed: {e}[/red]")
+    return False
+
+
 async def list_computational_tasks_from_db(
     state: AppState, user_id: int
 ) -> list[ComputationalTask]: