unshackle/tests/remote/unit/test_remote_resolve.py

"""Unit tests for resolve_server / resolve_proxy_arg in unshackle.core.remote_service."""

from __future__ import annotations

import click
import pytest

from unshackle.core.remote_service import resolve_proxy_arg, resolve_server

pytestmark = pytest.mark.unit


@pytest.fixture
def empty_remote_services(monkeypatch: pytest.MonkeyPatch) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(rs.config, "remote_services", {})


@pytest.fixture
def single_remote_service(monkeypatch: pytest.MonkeyPatch) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(
        rs.config,
        "remote_services",
        {
            "primary": {
                "url": "https://primary:8080",
                "api_key": "key-abc",
                "services": {"EXAMPLE": True, "DEMO": True},
                "server_cdm": True,
            }
        },
    )


@pytest.fixture
def multi_remote_services(monkeypatch: pytest.MonkeyPatch) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(
        rs.config,
        "remote_services",
        {
            "a": {"url": "https://a:8080", "api_key": "ka"},
            "b": {"url": "https://b:8080", "api_key": "kb"},
        },
    )


def test_resolve_server_default_auth_headers(single_remote_service) -> None:
    _, _, services = resolve_server("primary")
    assert services["_auth_headers"] == ["X-Secret-Key", "X-Api-Key"]


def test_resolve_server_custom_auth_headers_extend_defaults(monkeypatch: pytest.MonkeyPatch) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(
        rs.config, "remote_services", {"p": {"url": "https://p:8080", "auth_headers": ["Authorization"]}}
    )
    _, _, services = resolve_server("p")
    assert services["_auth_headers"] == ["Authorization", "X-Secret-Key", "X-Api-Key"]


def test_resolve_server_auth_headers_dedup_is_case_insensitive(monkeypatch: pytest.MonkeyPatch) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(
        rs.config,
        "remote_services",
        {"p": {"url": "https://p:8080", "auth_headers": ["Authorization", "x-secret-key"]}},
    )
    _, _, services = resolve_server("p")
    assert services["_auth_headers"] == ["Authorization", "x-secret-key", "X-Api-Key"]


@pytest.mark.parametrize("bad", ["X-Api-Key", [], [""], ["X-Api-Key", 3]])
def test_resolve_server_bad_auth_headers_raises(monkeypatch: pytest.MonkeyPatch, bad: object) -> None:
    from unshackle.core import remote_service as rs

    monkeypatch.setattr(rs.config, "remote_services", {"p": {"url": "https://p:8080", "auth_headers": bad}})
    with pytest.raises(click.ClickException) as exc:
        resolve_server("p")
    assert "auth_headers" in str(exc.value.message)


def test_resolve_server_no_config_raises_click(empty_remote_services) -> None:
    with pytest.raises(click.ClickException) as exc:
        resolve_server(None)
    assert "remote_services" in str(exc.value.message)


def test_resolve_server_single_picks_only_entry(single_remote_service) -> None:
    url, key, services = resolve_server(None)
    assert url == "https://primary:8080"
    assert key == "key-abc"
    assert services["_server_cdm"] is True
    assert services.get("EXAMPLE") is True


def test_resolve_server_explicit_name(single_remote_service) -> None:
    url, key, services = resolve_server("primary")
    assert url == "https://primary:8080"
    assert services["_server_cdm"] is True


def test_resolve_server_unknown_name_raises(single_remote_service) -> None:
    with pytest.raises(click.ClickException) as exc:
        resolve_server("bogus")
    assert "bogus" in str(exc.value.message)


def test_resolve_server_multi_requires_explicit(multi_remote_services) -> None:
    with pytest.raises(click.ClickException) as exc:
        resolve_server(None)
    assert "--server" in str(exc.value.message)


def test_resolve_server_multi_with_name(multi_remote_services) -> None:
    url, key, services = resolve_server("b")
    assert url == "https://b:8080"
    assert key == "kb"
    assert services["_server_cdm"] is None


def test_resolve_proxy_none_returns_none() -> None:
    assert resolve_proxy_arg(None) is None
    assert resolve_proxy_arg("") is None


def test_resolve_proxy_passes_through_value(monkeypatch: pytest.MonkeyPatch) -> None:
    import unshackle.core.proxies.resolve as resolve_mod

    monkeypatch.setattr(resolve_mod, "initialize_proxy_providers", lambda: [])
    monkeypatch.setattr(resolve_mod, "resolve_proxy", lambda arg, providers: f"http://proxy/{arg}")

    assert resolve_proxy_arg("us") == "http://proxy/us"


def test_resolve_proxy_value_error_becomes_click(monkeypatch: pytest.MonkeyPatch) -> None:
    import unshackle.core.proxies.resolve as resolve_mod

    monkeypatch.setattr(resolve_mod, "initialize_proxy_providers", lambda: [])

    def boom(*_):
        raise ValueError("no such country")

    monkeypatch.setattr(resolve_mod, "resolve_proxy", boom)

    with pytest.raises(click.ClickException) as exc:
        resolve_proxy_arg("xx")
    assert "no such country" in str(exc.value.message)