from __future__ import annotations

from types import SimpleNamespace

from opensquilla.gateway import boot, rpc_onboarding


def test_rpc_onboarding_sync_search_provider_passes_api_key_env(monkeypatch) -> None:
    calls: list[dict[str, object]] = []

    def fake_configure_search(**kwargs) -> None:
        calls.append(kwargs)

    monkeypatch.setattr("opensquilla.tools.builtin.web.configure_search", fake_configure_search)

    rpc_onboarding._sync_search_provider(
        SimpleNamespace(
            search_provider="exa",
            search_max_results=7,
            search_api_key="",
            search_api_key_env="CUSTOM_EXA_KEY",
            search_proxy="http://proxy.test",
            search_use_env_proxy=True,
            search_fallback_policy="network",
            search_diagnostics=True,
        )
    )

    assert calls == [
        {
            "provider_name": "exa",
            "max_results": 7,
            "api_key": "",
            "api_key_env": "CUSTOM_EXA_KEY",
            "proxy": "http://proxy.test",
            "use_env_proxy": True,
            "fallback_policy": "network",
            "diagnostics": True,
        }
    ]


def test_boot_search_provider_setup_no_longer_has_brave_only_promotion() -> None:
    source = boot.__loader__.get_source(boot.__name__)

    assert source is not None
    assert "Auto-select: use brave" not in source
    assert "os.environ.get(\"BRAVE_SEARCH_API_KEY\")" not in source