"""虎牙网页 Cookie 回写与 raw_cookie 重建（不发起登录或支付）。"""

import pytest

from core.huya.web_cookie_fields import cookie_fields, missing_web_cookie_fields
from tests.support.db import get_test_engine, make_test_session
from web.backend.models import HuyaAccount, User
from web.backend.routers.huya.cookies import rebuild_web_cookie
from web.backend.services.huya_service import (
    persist_huya_session_cookie,
    rebuild_web_cookie_from_raw,
)

RAW_COOKIE = (
    "udb_cred=cred; udb_uid=100; yyuid=100; udb_biztoken=biz; udb_appid=5008; sdid=sdid"
)


class TestHuyaCookieRebuild:
    def setup_method(self):
        self.engine = get_test_engine()
        self.session = make_test_session(self.engine)
        self.user = User(username="operator", password_hash="hash", role="super_admin")
        self.account = HuyaAccount(
            uid="100",
            yyuid="100",
            username="",
            cookie=RAW_COOKIE,
            raw_cookie=RAW_COOKIE,
        )
        self.session.add_all([self.user, self.account])
        self.session.commit()

    def teardown_method(self):
        self.session.close()

    def test_persist_writes_filled_session_cookie(self):
        filled = RAW_COOKIE + "; guid=0123456789abcdef0123456789abcdef"
        assert persist_huya_session_cookie(
            self.session, self.account.id, RAW_COOKIE, filled
        )
        self.session.refresh(self.account)
        assert "guid=" in self.account.cookie
        assert (
            persist_huya_session_cookie(self.session, self.account.id, filled, filled)
            is False
        )

    def test_rebuild_from_raw_keeps_app_creds_and_fills_web_fields(self):
        rebuilt = rebuild_web_cookie_from_raw(self.session, self.account)
        fields = cookie_fields(rebuilt)
        assert fields["udb_cred"] == "cred"
        assert fields["udb_appid"] == "5002"
        assert not missing_web_cookie_fields(rebuilt)
        self.session.refresh(self.account)
        assert self.account.cookie == rebuilt
        assert self.account.raw_cookie == RAW_COOKIE

    def test_rebuild_without_raw_cookie_raises(self):
        self.account.raw_cookie = ""
        self.session.commit()
        with pytest.raises(ValueError, match="原始凭据"):
            rebuild_web_cookie_from_raw(self.session, self.account)

    def test_rebuild_route_returns_cookie_without_login(self):
        payload = rebuild_web_cookie(
            account_id=self.account.id,
            db=self.session,
            current=self.user,
        )
        assert payload["id"] == self.account.id
        assert payload["has_raw_cookie"] is True
        assert payload["cookie"]
        assert "udb_cred=cred" in payload["cookie"]