Add simple hello world test for api

wri · Feb 11, 2025 · bfabf23 · bfabf23
1 parent 919ceba
commit bfabf23
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -0,0 +1,14 @@
+from fastapi.testclient import TestClient
+
+from api import app
+
+client = TestClient(app)
+
+
+def test_read_main():
+    response = client.post(
+        "/stream/dist_alert",
+        json=dict(query="Hi Zeno", thread_id="1", query_type="human_input"),
+    )
+    assert response.status_code == 200
+    assert "Hello" in response.json()["content"]