retoors/tests/test_site.py

async def test_index_get(client):
    resp = await client.get("/")
    assert resp.status == 200
    text = await resp.text()
    assert "Elevate Your Business Data" in text
    assert "Start Free Trial" in text


async def test_dashboard_get_unauthorized(client):
    resp = await client.get("/dashboard", allow_redirects=False)
    assert resp.status == 302
    assert resp.headers["Location"] == "/login"


async def test_dashboard_get_authorized(client):
    await client.post(
        "/register",
        data={
            "full_name": "Test User",
            "email": "test@example.com",
            "password": "password",
            "confirm_password": "password",
        },
    )
    await client.post(
        "/login", data={"email": "test@example.com", "password": "password"}
    )
    resp = await client.get("/dashboard")
    assert resp.status == 200
    text = await resp.text()
    assert "Your Data, Organized and Accessible" in text
    assert "My Files" in text


async def test_order_post_unauthorized(client):
    resp = await client.post(
        "/order", data={"storage_amount": "10.5"}, allow_redirects=False
    )
    assert resp.status == 302
    assert resp.headers["Location"] == "/login"


async def test_order_post_authorized(client):
    await client.post(
        "/register",
        data={
            "full_name": "Test User",
            "email": "test@example.com",
            "password": "password",
            "confirm_password": "password",
        },
    )
    await client.post(
        "/login", data={"email": "test@example.com", "password": "password"}
    )
    resp = await client.post("/order", data={"storage_amount": "10.5"}, allow_redirects=False)
    assert resp.status == 302
    assert resp.headers["Location"] == "/dashboard"

    # Verify that the user's quota was updated
    user_service = client.app["user_service"]
    user = user_service.get_user_by_email("test@example.com")
    assert user["storage_quota_gb"] == 10.5


async def test_order_post_invalid_amount(client):
    await client.post(
        "/register",
        data={
            "full_name": "Test User",
            "email": "test@example.com",
            "password": "password",
            "confirm_password": "password",
        },
    )
    await client.post(
        "/login", data={"email": "test@example.com", "password": "password"}
    )
    resp = await client.post("/order", data={"storage_amount": "0"})
    assert resp.status == 200
    text = await resp.text()
    assert "ensure this value is greater than 0" in text

    resp = await client.post("/order", data={"storage_amount": "1001"})
    assert resp.status == 200
    text = await resp.text()
    assert "ensure this value is less than or equal to 1000" in text