fix test

Author: Kludex

src/mcp/client/client.py

@@ -128,7 +128,7 @@ async def __aenter__(self) -> Client:
 
     async def __aexit__(self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: Any) -> None:
         """Exit the async context manager."""
-        if self._exit_stack:
+        if self._exit_stack:  # pragma: no branch
             await self._exit_stack.__aexit__(exc_type, exc_val, exc_tb)
         self._session = None
 

tests/client/test_client.py

@@ -77,11 +77,6 @@ def greet(name: str) -> str:
         """Greet someone by name."""
         return f"Hello, {name}!"
 
-    @server.tool()
-    def add(a: int, b: int) -> int:
-        """Add two numbers."""
-        return a + b
-
     @server.resource("test://resource")
     def test_resource() -> str:
         """A test resource."""
@@ -146,27 +141,7 @@ async def test_client_list_tools(app: FastMCP):
                             "title": "greetOutput",
                             "type": "object",
                         },
-                    ),
-                    Tool(
-                        name="add",
-                        description="Add two numbers.",
-                        input_schema={
-                            "properties": {
-                                "a": {"title": "A", "type": "integer"},
-                                "b": {"title": "B", "type": "integer"},
-                            },
-                            "required": ["a", "b"],
-                            "title": "addArguments",
-                            "type": "object",
-                        },
-                        output_schema={
-                            "properties": {"result": {"title": "Result", "type": "integer"}},
-                            "required": ["result"],
-                            "title": "addOutput",
-                            "type": "object",
-                        },
-                    ),
-                ]
+                    )]
             )
         )