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"))