from datetime import UTC, datetime

from restreamer.resolver import parse_expiry_from_url


def test_parse_expiry_from_url_reads_expire_param() -> None:
    url = "https://r.googlevideo.com/videoplayback?id=abc&expire=1700000000&itag=96"

    result = parse_expiry_from_url(url)

    assert result == datetime.fromtimestamp(1700000000, tz=UTC)


def test_parse_expiry_from_url_returns_none_without_param() -> None:
    url = "https://example.com/stream.m3u8?token=abc"

    assert parse_expiry_from_url(url) is None


def test_parse_expiry_from_url_ignores_non_numeric_value() -> None:
    url = "https://example.com/stream.m3u8?expire=not-a-number"

    assert parse_expiry_from_url(url) is None