25 lines
772 B
Python
25 lines
772 B
Python
from fastapi.testclient import TestClient
|
|
from app.main import app
|
|
|
|
c = TestClient(app)
|
|
|
|
# login
|
|
r = c.post("/login", data={"username": "admin", "password": "admin123"}, follow_redirects=False)
|
|
print("login:", r.status_code)
|
|
session = r.cookies.get("pc_session")
|
|
if session:
|
|
c.cookies.set("pc_session", session)
|
|
r = c.get("/dashboard")
|
|
print("dashboard:", r.status_code)
|
|
assert r.status_code == 200
|
|
r = c.get("/profiles")
|
|
print("profiles:", r.status_code)
|
|
assert r.status_code == 200
|
|
print("All checks passed")
|
|
else:
|
|
print("login failed, no session")
|
|
r2 = c.post("/login", data={"username": "admin", "password": "admin123"}, follow_redirects=True)
|
|
print("follow:", r2.status_code, "pc_session:", r2.cookies.get("pc_session"))
|
|
|
|
|