Spaces:
Sleeping
Sleeping
File size: 2,403 Bytes
4950e3d 7d1fec5 4950e3d 10aaf26 4950e3d 7d1fec5 4950e3d 10aaf26 4950e3d 7d1fec5 10aaf26 7d1fec5 10aaf26 4950e3d 10aaf26 7d1fec5 10aaf26 7d1fec5 10aaf26 7d1fec5 4950e3d 7d1fec5 10aaf26 4950e3d 10aaf26 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | from fastapi.testclient import TestClient
from backend.app import app
from backend.api import notebooks as notebooks_api
from backend.services.storage import NotebookStore
client = TestClient(app)
AUTH_U1 = {"X-User-Id": "u1"}
AUTH_U2 = {"X-User-Id": "u2"}
def setup_function():
notebooks_api.store = NotebookStore(base_dir="tests_tmp_data")
def test_create_list_get_rename_delete_notebook(tmp_path):
notebooks_api.store = NotebookStore(base_dir=str(tmp_path))
payload = {"user_id": "u1", "name": "Notebook 1"}
resp = client.post("/api/notebooks/", json=payload, headers=AUTH_U1)
assert resp.status_code == 200
data = resp.json()
assert data["user_id"] == "u1"
assert data["name"] == "Notebook 1"
assert "notebook_id" in data
notebook_id = data["notebook_id"]
resp_list = client.get("/api/notebooks/", params={"user_id": "u1"}, headers=AUTH_U1)
assert resp_list.status_code == 200
listed = resp_list.json()
assert len(listed) == 1
assert listed[0]["notebook_id"] == notebook_id
resp_get = client.get(f"/api/notebooks/{notebook_id}", params={"user_id": "u1"}, headers=AUTH_U1)
assert resp_get.status_code == 200
data_get = resp_get.json()
assert data_get["notebook_id"] == notebook_id
resp_wrong_user = client.get(f"/api/notebooks/{notebook_id}", params={"user_id": "u2"}, headers=AUTH_U2)
assert resp_wrong_user.status_code == 404
resp_rename = client.patch(
f"/api/notebooks/{notebook_id}",
json={"user_id": "u1", "name": "Renamed"},
headers=AUTH_U1,
)
assert resp_rename.status_code == 200
assert resp_rename.json()["name"] == "Renamed"
resp_delete = client.delete(f"/api/notebooks/{notebook_id}", params={"user_id": "u1"}, headers=AUTH_U1)
assert resp_delete.status_code == 200
assert resp_delete.json() == {"deleted": True}
def test_get_missing_notebook(tmp_path):
notebooks_api.store = NotebookStore(base_dir=str(tmp_path))
resp = client.get("/api/notebooks/missing", params={"user_id": "u1"}, headers=AUTH_U1)
assert resp.status_code == 404
def test_rejects_mismatched_user_id_payload(tmp_path):
notebooks_api.store = NotebookStore(base_dir=str(tmp_path))
resp = client.post(
"/api/notebooks/",
json={"user_id": "u2", "name": "Notebook 1"},
headers=AUTH_U1,
)
assert resp.status_code == 403
|