Enhanced Dash MCP Server

import asyncio
import importlib.util
from pathlib import Path
from typing import Any

import pytest

MODULE_PATH = Path(__file__).resolve().parents[1] / "enhanced_dash_server.py"


class DummyServer:
    def __init__(self, *_args: Any, **_kwargs: Any) -> None:
        pass

    def list_tools(self) -> Any:
        def decorator(func: Any) -> Any:
            return func
        return decorator

    def call_tool(self) -> Any:
        def decorator(func: Any) -> Any:
            async def wrapper(*args: Any, **kwargs: Any) -> Any:
                return await func(*args, **kwargs)
            return wrapper
        return decorator

    async def run(self, *_args: Any, **_kwargs: Any) -> None:  # pragma: no cover
        pass


@pytest.mark.asyncio
async def test_symlink_resolved(monkeypatch, tmp_path, stub_modules) -> None:
    stub_modules["mcp.server"].Server = DummyServer  # type: ignore[attr-defined]

    # create actual docsets path
    actual = tmp_path / "Dropbox" / "DocSets"
    resources = actual / "Sample.docset" / "Contents" / "Resources"
    resources.mkdir(parents=True)
    (resources / "docSet.dsidx").write_text("")

    # create default path with symlink
    library = tmp_path / "Library" / "Application Support"
    library.mkdir(parents=True)
    (library / "Dash").symlink_to(actual.parent)

    monkeypatch.setattr(Path, "home", lambda: tmp_path)

    spec = importlib.util.spec_from_file_location("enhanced_dash_server", MODULE_PATH)
    assert spec and spec.loader
    server_mod = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(server_mod)

    dash_server = server_mod.DashMCPServer()
    docsets = await dash_server.get_available_docsets()
    names = {d["name"] for d in docsets}
    assert "Sample" in names


@pytest.mark.asyncio
async def test_env_points_to_dash(monkeypatch, tmp_path, stub_modules) -> None:
    """DASH_DOCSETS_PATH can refer to the Dash directory and still work."""
    stub_modules["mcp.server"].Server = DummyServer  # type: ignore[attr-defined]

    dash_root = tmp_path / "Dropbox"
    resources = dash_root / "DocSets" / "Sample.docset" / "Contents" / "Resources"
    resources.mkdir(parents=True)
    (resources / "docSet.dsidx").write_text("")

    monkeypatch.setenv("DASH_DOCSETS_PATH", str(dash_root))

    spec = importlib.util.spec_from_file_location("enhanced_dash_server", MODULE_PATH)
    assert spec and spec.loader
    server_mod = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(server_mod)

    dash_server = server_mod.DashMCPServer()
    docsets = await dash_server.get_available_docsets()
    names = {d["name"] for d in docsets}
    assert "Sample" in names


@pytest.mark.asyncio
async def test_env_symlink_to_docsets(monkeypatch, tmp_path, stub_modules) -> None:
    """DASH_DOCSETS_PATH can point to a symlink of the DocSets folder."""
    stub_modules["mcp.server"].Server = DummyServer  # type: ignore[attr-defined]

    actual = tmp_path / "Actual" / "DocSets"
    resources = actual / "Sample.docset" / "Contents" / "Resources"
    resources.mkdir(parents=True)
    (resources / "docSet.dsidx").write_text("")

    symlink = tmp_path / "LinkedDocSets"
    symlink.symlink_to(actual)

    monkeypatch.setenv("DASH_DOCSETS_PATH", str(symlink))

    spec = importlib.util.spec_from_file_location("enhanced_dash_server", MODULE_PATH)
    assert spec and spec.loader
    server_mod = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(server_mod)

    dash_server = server_mod.DashMCPServer()
    assert dash_server.docsets_path == actual.resolve()
    docsets = await dash_server.get_available_docsets()
    names = {d["name"] for d in docsets}
    assert "Sample" in names