from gpu_rent.access_card import collect_access_links, mcp_snippet_lines


class _Cfg:
    swarmui_local_port = 17801
    llm_runtime = "ollama"
    ollama_local_port = 17811
    llamacpp_local_port = 17812


def test_collect_links_swarm_and_ollama(monkeypatch):
    monkeypatch.setattr(
        "gpu_rent.access_card.load_state",
        lambda: type("S", (), {"notes": {"llm_runtime": "ollama"}})(),
    )
    links = collect_access_links(_Cfg(), tunneled=True)
    labels = [x.label for x in links]
    assert "SwarmUI UI" in labels
    assert "SwarmUI MCP" in labels
    assert "Ollama API" in labels
    assert any("17811" in x.url for x in links)


def test_collect_links_no_tunnel():
    links = collect_access_links(_Cfg(), tunneled=False)
    assert links[0].url.startswith("gpu-rent tunnel")


def test_mcp_snippet_json():
    lines = mcp_snippet_lines(_Cfg())
    assert any("17801/mcp" in line for line in lines)