"""Extract Fragment cookies from browser stores and validate required keys."""

from unittest.mock import MagicMock, patch

import pytest

from pyfragment import CookieError, get_cookies_from_browser
from pyfragment.core.constants import REQUIRED_COOKIE_KEYS

FAKE_JAR = [
    {"name": "stel_ssid", "value": "abc123", "domain": "fragment.com", "expires": "2027-04-03T20:52:16.375Z"},
    {"name": "stel_dt", "value": "-120", "domain": "fragment.com"},
    {"name": "stel_token", "value": "tok_xyz", "domain": "fragment.com"},
    {"name": "stel_ton_token", "value": "ton_xyz", "domain": "fragment.com"},
    {"name": "unrelated", "value": "noise", "domain": "fragment.com"},
]


def _mock_rookiepy(jar: list[dict[str, str]] | None = None) -> MagicMock:
    mock = MagicMock()
    mock.chrome.return_value = jar if jar is not None else FAKE_JAR
    return mock


PATCH = "pyfragment.core.cookies.rookiepy"


# unsupported browser tests


def test_unsupported_browser_raises() -> None:
    with pytest.raises(CookieError, match="Unsupported browser"):
        get_cookies_from_browser("internet_explorer")


# successful extraction tests


def test_returns_required_keys_only() -> None:
    with patch(PATCH, _mock_rookiepy()):
        result = get_cookies_from_browser("chrome")

    assert set(result.cookies.keys()) == set(REQUIRED_COOKIE_KEYS)
    assert result.cookies["stel_ssid"] == "abc123"
    assert result.cookies["stel_dt"] == "-120"
    assert result.cookies["stel_token"] == "tok_xyz"
    assert result.cookies["stel_ton_token"] == "ton_xyz"
    assert "unrelated" not in result.cookies


def test_returns_expiry() -> None:
    with patch(PATCH, _mock_rookiepy()):
        result = get_cookies_from_browser("chrome")

    assert result.expires == "2027-04-03T20:52:16.375000+00:00"


def test_returns_none_expiry_when_missing() -> None:
    jar = [{"name": k, "value": "v"} for k in REQUIRED_COOKIE_KEYS]
    with patch(PATCH, _mock_rookiepy(jar)):
        result = get_cookies_from_browser("chrome")

    assert result.expires is None


def test_default_browser_is_chrome() -> None:
    mock_rp = _mock_rookiepy()
    with patch(PATCH, mock_rp):
        get_cookies_from_browser()

    mock_rp.chrome.assert_called_once_with(["fragment.com"])


def test_browser_name_is_case_insensitive() -> None:
    mock_rp = _mock_rookiepy()
    with patch(PATCH, mock_rp):
        result = get_cookies_from_browser("Chrome")

    assert result.cookies["stel_ssid"] == "abc123"


# missing cookies tests


def test_missing_cookies_raises() -> None:
    partial_jar = [
        {"name": "stel_ssid", "value": "abc123"},
        {"name": "stel_dt", "value": "-120"},
        # stel_token and stel_ton_token missing
    ]
    with patch(PATCH, _mock_rookiepy(partial_jar)):
        with pytest.raises(CookieError, match="Fragment cookies not found in chrome"):
            get_cookies_from_browser("chrome")


def test_empty_cookie_value_treated_as_missing() -> None:
    jar_with_empty = [
        {"name": "stel_ssid", "value": "abc123"},
        {"name": "stel_dt", "value": ""},  # empty — should be treated as missing
        {"name": "stel_token", "value": "tok_xyz"},
        {"name": "stel_ton_token", "value": "ton_xyz"},
    ]
    with patch(PATCH, _mock_rookiepy(jar_with_empty)):
        with pytest.raises(CookieError, match="Fragment cookies not found in chrome"):
            get_cookies_from_browser("chrome")


# expired cookie tests


def test_expired_cookie_raises() -> None:
    expired_jar = [
        {"name": "stel_ssid", "value": "abc123", "expires": "2020-01-01T00:00:00.000Z"},
        {"name": "stel_dt", "value": "-120"},
        {"name": "stel_token", "value": "tok_xyz"},
        {"name": "stel_ton_token", "value": "ton_xyz"},
    ]
    with patch(PATCH, _mock_rookiepy(expired_jar)):
        with pytest.raises(CookieError, match="expired"):
            get_cookies_from_browser("chrome")


# read failure tests


def test_browser_read_error_raises() -> None:
    mock_rp = MagicMock()
    mock_rp.chrome.side_effect = PermissionError("locked")
    with patch(PATCH, mock_rp):
        with pytest.raises(CookieError, match="Failed to read chrome cookies"):
            get_cookies_from_browser("chrome")