fix: add single entrypoint verification

Author: radugheo

.gitignore

@@ -186,3 +186,5 @@ cython_debug/
 **/samples/**/CLAUDE.md
 **/samples/**/entry-points.json
 **/samples/**/uv.lock
+
+.claude/

pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "uipath-mcp"
-version = "0.2.0"
+version = "0.2.1"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
     "mcp==1.26.0",
     "pysignalr==1.3.0",
-    "uipath>=2.10.40, <2.11.0",
+    "uipath>=2.10.42, <2.11.0",
     "uipath-runtime>=0.10.0, <0.11.0",
 ]
 classifiers = [
@@ -42,14 +42,13 @@ dev = [
     "mypy>=1.14.1",
     "ruff>=0.9.4",
     "pytest>=7.4.0",
-    "pytest-asyncio>=0.23.0",
+    "pytest-asyncio>=1.3.0",
     "pytest-cov>=4.1.0",
     "pytest-mock>=3.11.1",
     "pre-commit>=4.5.1",
     "filelock>=3.20.3",
     "virtualenv>=20.36.1",
     "numpy>=1.24.0",
-    "pytest-asyncio>=1.3.0",
 ]
 
 [tool.ruff]

src/uipath_mcp/_cli/_utils/_config.py

@@ -106,14 +106,7 @@ def get_servers(self) -> list[McpServer]:
     def get_server(self, name: str) -> McpServer | None:
         """
         Get a server model by name.
-        If there's only one server available, return that one regardless of name.
-        Otherwise, look up the server by the provided name.
         """
-        # If there's only one server, return it
-        if len(self._servers) == 1:
-            return next(iter(self._servers.values()))
-
-        # Otherwise, fall back to looking up by name
         return self._servers.get(name)
 
     def get_server_names(self) -> list[str]:

tests/cli/test_config.py

@@ -0,0 +1,30 @@
+from uipath_mcp._cli._utils._config import McpConfig
+
+
+def test_get_server_exact_match_single_server(tmp_path):
+    """Single-server config should NOT return the server for a mismatched name."""
+    config_file = tmp_path / "mcp.json"
+    config_file.write_text('{"servers": {"my-server": {"command": "python"}}}')
+
+    config = McpConfig(config_path=str(config_file))
+
+    assert config.get_server("my-server") is not None
+    assert config.get_server("wrong-name") is None
+
+
+def test_get_server_exact_match_multiple_servers(tmp_path):
+    """Multi-server config returns correct server by name."""
+    config_file = tmp_path / "mcp.json"
+    config_file.write_text(
+        '{"servers": {"server-a": {"command": "a"}, "server-b": {"command": "b"}}}'
+    )
+
+    config = McpConfig(config_path=str(config_file))
+
+    server_a = config.get_server("server-a")
+    server_b = config.get_server("server-b")
+    assert server_a is not None
+    assert server_b is not None
+    assert server_a.name == "server-a"
+    assert server_b.name == "server-b"
+    assert config.get_server("server-c") is None