From 5c850a56c1ef6ae6e94fc594b8e9e84273d10452 Mon Sep 17 00:00:00 2001 From: arkohut <39525455+arkohut@users.noreply.github.com> Date: Sat, 1 Jun 2024 13:46:30 +0800 Subject: [PATCH] test: add first test --- memos/server.py | 5 +++++ memos/test_server.py | 11 +++++++++++ 2 files changed, 16 insertions(+) create mode 100644 memos/test_server.py 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}