use FastAPI test client for testing (#322)

Author: pmeier

scripts/add_chats.py

@@ -20,7 +20,7 @@ def main():
             data={
                 "username": username,
                 "password": os.environ.get(
-                    "AI_PROXY_DEMO_AUTHENTICATION_PASSWORD", username
+                    "RAGNA_DEMO_AUTHENTICATION_PASSWORD", username
                 ),
             },
         )

tests/deploy/api/test_e2e.py

@@ -1,47 +1,40 @@
 import json
 import os
 
-import httpx
 import httpx_sse
 import pytest
+from fastapi.testclient import TestClient
 
-from ragna._utils import timeout_after
 from ragna.core._utils import default_user
 from ragna.deploy import Config
-from tests.utils import ragna_api
+from ragna.deploy._api import app
 
 
-@pytest.mark.parametrize("database", ["memory", "sqlite"])
 @pytest.mark.parametrize("stream_answer", [True, False])
-def test_e2e(tmp_local_root, database, stream_answer):
-    if database == "memory":
-        database_url = "memory"
-    elif database == "sqlite":
-        database_url = f"sqlite:///{tmp_local_root / 'ragna.db'}"
-
+def test_e2e(tmp_local_root, stream_answer):
     config = Config(
-        local_cache_root=tmp_local_root, api=dict(database_url=database_url)
+        local_cache_root=tmp_local_root,
+        api=dict(database_url=f"sqlite:///{tmp_local_root / 'ragna.db'}"),
     )
     check_api(config, stream_answer=stream_answer)
 
 
-@timeout_after()
 def check_api(config, *, stream_answer):
     document_root = config.local_cache_root / "documents"
     document_root.mkdir()
     document_path = document_root / "test.txt"
     with open(document_path, "w") as file:
         file.write("!\n")
 
-    with ragna_api(config), httpx.Client(base_url=config.api.url) as client:
+    with TestClient(app(config)) as client:
         username = default_user()
         token = (
             client.post(
                 "/token",
                 data={
                     "username": username,
                     "password": os.environ.get(
-                        "AI_PROXY_DEMO_AUTHENTICATION_PASSWORD", username
+                        "RAGNA_DEMO_AUTHENTICATION_PASSWORD", username
                     ),
                 },
             )