diff --git a/memos/server.py b/memos/server.py index 878f6ce..f3c0881 100644 --- a/memos/server.py +++ b/memos/server.py @@ -33,6 +33,11 @@ def get_db(): db.close() +@app.get("/") +def root(): + return {"healthy": True} + + @app.post("/libraries", response_model=Library) def new_library(library_param: NewLibraryParam, db: Session = Depends(get_db)): library = create_library(library_param, db) diff --git a/memos/test_server.py b/memos/test_server.py new file mode 100644 index 0000000..ca7c4e1 --- /dev/null +++ b/memos/test_server.py @@ -0,0 +1,11 @@ +from fastapi.testclient import TestClient + +from .server import app + +client = TestClient(app) + + +def test_read_main(): + response = client.get("/") + assert response.status_code == 200 + assert response.json() == {"healthy": True}