Rename FastMCP to MCPServer (#1951)

tests/server/mcpserver/servers/test_file_server.py

@@ -0,0 +1,128 @@
+import json
+from pathlib import Path
+
+import pytest
+
+from mcp.server.mcpserver import MCPServer
+
+
+@pytest.fixture()
+def test_dir(tmp_path_factory: pytest.TempPathFactory) -> Path:
+    """Create a temporary directory with test files."""
+    tmp = tmp_path_factory.mktemp("test_files")
+
+    # Create test files
+    (tmp / "example.py").write_text("print('hello world')")
+    (tmp / "readme.md").write_text("# Test Directory\nThis is a test.")
+    (tmp / "config.json").write_text('{"test": true}')
+
+    return tmp
+
+
+@pytest.fixture
+def mcp() -> MCPServer:
+    mcp = MCPServer()
+
+    return mcp
+
+
+@pytest.fixture(autouse=True)
+def resources(mcp: MCPServer, test_dir: Path) -> MCPServer:
+    @mcp.resource("dir://test_dir")
+    def list_test_dir() -> list[str]:
+        """List the files in the test directory"""
+        return [str(f) for f in test_dir.iterdir()]
+
+    @mcp.resource("file://test_dir/example.py")
+    def read_example_py() -> str:
+        """Read the example.py file"""
+        try:
+            return (test_dir / "example.py").read_text()
+        except FileNotFoundError:
+            return "File not found"
+
+    @mcp.resource("file://test_dir/readme.md")
+    def read_readme_md() -> str:
+        """Read the readme.md file"""
+        try:  # pragma: no cover
+            return (test_dir / "readme.md").read_text()
+        except FileNotFoundError:  # pragma: no cover
+            return "File not found"
+
+    @mcp.resource("file://test_dir/config.json")
+    def read_config_json() -> str:
+        """Read the config.json file"""
+        try:  # pragma: no cover
+            return (test_dir / "config.json").read_text()
+        except FileNotFoundError:  # pragma: no cover
+            return "File not found"
+
+    return mcp
+
+
+@pytest.fixture(autouse=True)
+def tools(mcp: MCPServer, test_dir: Path) -> MCPServer:
+    @mcp.tool()
+    def delete_file(path: str) -> bool:
+        # ensure path is in test_dir
+        if Path(path).resolve().parent != test_dir:  # pragma: no cover
+            raise ValueError(f"Path must be in test_dir: {path}")
+        Path(path).unlink()
+        return True
+
+    return mcp
+
+
+@pytest.mark.anyio
+async def test_list_resources(mcp: MCPServer):
+    resources = await mcp.list_resources()
+    assert len(resources) == 4
+
+    assert [str(r.uri) for r in resources] == [
+        "dir://test_dir",
+        "file://test_dir/example.py",
+        "file://test_dir/readme.md",
+        "file://test_dir/config.json",
+    ]
+
+
+@pytest.mark.anyio
+async def test_read_resource_dir(mcp: MCPServer):
+    res_iter = await mcp.read_resource("dir://test_dir")
+    res_list = list(res_iter)
+    assert len(res_list) == 1
+    res = res_list[0]
+    assert res.mime_type == "text/plain"
+
+    files = json.loads(res.content)
+
+    assert sorted([Path(f).name for f in files]) == [
+        "config.json",
+        "example.py",
+        "readme.md",
+    ]
+
+
+@pytest.mark.anyio
+async def test_read_resource_file(mcp: MCPServer):
+    res_iter = await mcp.read_resource("file://test_dir/example.py")
+    res_list = list(res_iter)
+    assert len(res_list) == 1
+    res = res_list[0]
+    assert res.content == "print('hello world')"
+
+
+@pytest.mark.anyio
+async def test_delete_file(mcp: MCPServer, test_dir: Path):
+    await mcp.call_tool("delete_file", arguments={"path": str(test_dir / "example.py")})
+    assert not (test_dir / "example.py").exists()
+
+
+@pytest.mark.anyio
+async def test_delete_file_and_check_resources(mcp: MCPServer, test_dir: Path):
+    await mcp.call_tool("delete_file", arguments={"path": str(test_dir / "example.py")})
+    res_iter = await mcp.read_resource("file://test_dir/example.py")
+    res_list = list(res_iter)
+    assert len(res_list) == 1
+    res = res_list[0]
+    assert res.content == "File not found"