ontology

ontology_platform/tests/e2e/test_phase0_full_pipeline.py

@@ -0,0 +1,93 @@
+"""Phase 0 end-to-end pipeline test.
+
+Runs the actual OntoCast workflow through the FastAPI ``/process`` endpoint
+with a tiny JSON input. Verifies Acceptance Gate 0 evidence:
+
+- /health, /info, /process respond OK
+- ontology + facts Turtle are produced
+- BudgetTracker reports non-zero LLM call/triple counts
+- Filesystem TripleStoreManager writes artifacts under working_directory
+
+Skipped unless ``LLM_API_KEY`` is set (see ``conftest.py``).
+"""
+
+from __future__ import annotations
+
+import importlib
+import json
+import os
+import sys
+from pathlib import Path
+
+import pytest
+from fastapi.testclient import TestClient
+
+REPO_ROOT = Path(__file__).resolve().parents[2]
+if str(REPO_ROOT) not in sys.path:
+    sys.path.insert(0, str(REPO_ROOT))
+
+main_module = importlib.import_module("platform.api.main")
+deps_module = importlib.import_module("platform.api.deps")
+platform_config = importlib.import_module("platform.config")
+
+
+@pytest.mark.e2e
+@pytest.mark.asyncio
+async def test_full_pipeline_writes_ontology_and_facts(
+    tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    # Point the working directory at the test's tmp_path so artifacts don't
+    # leak between runs.
+    monkeypatch.setenv("ONTOCAST_WORKING_DIRECTORY", str(tmp_path / "work"))
+    # Honor whatever LLM provider the operator configured.
+    assert os.environ.get("LLM_API_KEY"), "LLM_API_KEY must be set for e2e"
+
+    # Force a fresh AppContext so the new working_directory wins.
+    deps_module.reset_app_context_for_testing()
+    settings = platform_config.load_settings()
+    await deps_module.initialize_app_context(settings, head_chunks=1)
+
+    app = main_module.create_app()
+
+    # Tiny but ontology-rich payload.
+    payload = {
+        "text": (
+            "Alice works at Acme Corporation in Berlin. "
+            "Acme Corporation manufactures bicycles."
+        ),
+        "ontology_user_instruction": "Focus on person-organization-location relations.",
+        "facts_user_instruction": "Extract employment and manufacturing facts.",
+    }
+
+    with TestClient(app) as client:
+        # /health & /info first as smoke
+        assert client.get("/health").status_code == 200
+        assert client.get("/info").status_code == 200
+
+        response = client.post(
+            "/process",
+            content=json.dumps(payload),
+            headers={"Content-Type": "application/json"},
+        )
+
+    assert response.status_code == 200, response.text
+    body = response.json()
+    assert body["status"] == "success"
+    assert body["data"]["ontology"], "ontology TTL must not be empty"
+    assert body["data"]["facts"], "facts TTL must not be empty"
+
+    budget = body["metadata"]["budget"]
+    assert budget["calls_count"] > 0, "BudgetTracker must record at least one LLM call"
+    assert (
+        budget["ontology_triples_generated"] > 0
+        or budget["facts_triples_generated"] > 0
+    ), "BudgetTracker must record triple generation"
+
+    # The filesystem manager should have created artifacts somewhere under
+    # working_directory. We don't assert exact filenames (those depend on
+    # the document hash) — just that the directory is non-empty.
+    work_dir = tmp_path / "work"
+    written = list(work_dir.rglob("*"))
+    assert any(p.suffix in {".ttl", ".rdf"} for p in written if p.is_file()), (
+        f"No RDF artifacts produced under {work_dir}"
+    )