# 项目文件：shelf_project/tests/conftest.py
import pytest
from bs4 import BeautifulSoup
from werkzeug.security import generate_password_hash

from shelfweb import create_app
from shelfweb.db import get_db, init_db


@pytest.fixture(scope="session")
def hashes():
    # 只复用计算出的摘要，登录仍经过真正的校验。
    return {name: generate_password_hash(f"learn-{name}") for name in ["alice", "bob"]}


@pytest.fixture
def app(tmp_path, hashes):
    application = create_app(
        {
            "TESTING": True,
            "SECRET_KEY": "only-for-tests",
            "DATABASE": str(tmp_path / "books.sqlite3"),
        },
        instance_path=tmp_path / "instance",
    )
    with application.app_context():
        init_db()
        connection = get_db()
        with connection:
            connection.executemany(
                "INSERT INTO users(username,password_hash) VALUES (?,?)", hashes.items()
            )
    return application


@pytest.fixture
def client(app):
    return app.test_client()


def csrf(client, path="/auth/login"):
    response = client.get(path)
    assert response.status_code == 200
    return BeautifulSoup(response.data, "html.parser").select_one(
        'input[name="csrf_token"]'
    )["value"]


def login(client, name="alice"):
    token = csrf(client)
    response = client.post(
        "/auth/login",
        data={"username": name, "password": f"learn-{name}", "csrf_token": token},
    )
    assert response.status_code == 303
    return response


def create(client, title="Python Web", **values):
    data = {
        "title": title,
        "tags": "基础, 网络",
        "csrf_token": csrf(client, "/books/new"),
        **values,
    }
    return client.post("/books/new", data=data)