fixed integration tests

Author: lucaseduoli

tests/integration/core/test_startup_ingest.py

@@ -37,15 +37,15 @@ def count_files_in_documents() -> int:
 
 @pytest.mark.parametrize("disable_langflow_ingest", [True, False])
 @pytest.mark.asyncio
-async def test_startup_ingest_creates_task(disable_langflow_ingest: bool):
+async def test_startup_ingest_creates_task(disable_langflow_ingest: bool, monkeypatch):
     # Ensure startup ingest runs and choose pipeline per param
-    os.environ["DISABLE_STARTUP_INGEST"] = "false"
-    os.environ["DISABLE_INGEST_WITH_LANGFLOW"] = (
-        "true" if disable_langflow_ingest else "false"
+    monkeypatch.setenv("DISABLE_STARTUP_INGEST", "false")
+    monkeypatch.setenv(
+        "DISABLE_INGEST_WITH_LANGFLOW", "true" if disable_langflow_ingest else "false"
     )
     # Force no-auth mode for simpler endpoint access
-    os.environ["GOOGLE_OAUTH_CLIENT_ID"] = ""
-    os.environ["GOOGLE_OAUTH_CLIENT_SECRET"] = ""
+    monkeypatch.setenv("GOOGLE_OAUTH_CLIENT_ID", "")
+    monkeypatch.setenv("GOOGLE_OAUTH_CLIENT_SECRET", "")
 
     # Reload settings to pick up env for this test run
     import sys
@@ -70,8 +70,9 @@ async def test_startup_ingest_creates_task(disable_langflow_ingest: bool):
         pass
 
     app = await create_app()
-    # Trigger startup tasks explicitly
-    await startup_tasks(app.state.services)
+    lifespan_ctx = app.router.lifespan_context(app)
+    await lifespan_ctx.__aenter__()
+    app.state.lifespan_ctx = lifespan_ctx
 
     # Ensure index exists for tests (startup_tasks only creates it if DISABLE_INGEST_WITH_LANGFLOW=True)
     from main import _ensure_opensearch_index
@@ -113,6 +114,7 @@ async def _wait_for_task(timeout_s: float = 60.0):
             assert "task_id" in newest
             assert newest.get("total_files") == expected_files
     finally:
+        await lifespan_ctx.__aexit__(None, None, None)
         # Explicitly close global clients to avoid aiohttp warnings
         from config.settings import clients
         try: