from gpu_rent.config import parse_enable_swarmui
from gpu_rent.tunnel import tunnel_forwards


def test_parse_enable_swarmui_workload(monkeypatch):
    monkeypatch.delenv("WORKLOAD", raising=False)
    assert parse_enable_swarmui(None) is True
    assert parse_enable_swarmui("false") is False
    monkeypatch.setenv("WORKLOAD", "llm")
    assert parse_enable_swarmui("true") is False
    monkeypatch.setenv("WORKLOAD", "both")
    assert parse_enable_swarmui("false") is True


def test_tunnel_forwards_llm_only(monkeypatch):
    class Cfg:
        swarmui_local_port = 17801
        llm_runtime = "llamacpp"
        enable_swarmui = False
        ollama_local_port = 17811
        llamacpp_local_port = 17812

    monkeypatch.setattr("gpu_rent.tunnel.load_state", lambda: type("S", (), {"notes": {}})())
    assert tunnel_forwards(Cfg()) == [(17812, 8080)]


def test_access_links_llm_only(monkeypatch):
    from gpu_rent.access_card import collect_access_links, mcp_snippet_lines

    class Cfg:
        swarmui_local_port = 17801
        llm_runtime = "llamacpp"
        enable_swarmui = False
        ollama_local_port = 17811
        llamacpp_local_port = 17812

    monkeypatch.setattr(
        "gpu_rent.access_card.load_state",
        lambda: type("S", (), {"notes": {}})(),
    )
    labels = [x.label for x in collect_access_links(Cfg(), tunneled=True)]
    assert "SwarmUI UI" not in labels
    assert "llama.cpp" in labels
    assert mcp_snippet_lines(Cfg())[0].startswith("#")