fix: use lax no cover and no branch pragmas for src/ files

Replace incorrect pragma types with the appropriate ones:
- Use 'lax no cover' for covered lines with uncovered bodies (excluded
  from coverage but not checked by strict-no-cover)
- Use 'no branch' for branches where target lines are covered but the
  branch path itself is not taken
- Keep 'no cover' only for genuinely uncovered code paths

Author: maxisbey

src/mcp/cli/claude.py

@@ -72,7 +72,7 @@ def update_claude_config(
         )
 
     config_file = config_dir / "claude_desktop_config.json"
-    if not config_file.exists():
+    if not config_file.exists():  # pragma: lax no cover
         try:
             config_file.write_text("{}")
         except Exception:

src/mcp/cli/cli.py

@@ -77,7 +77,7 @@ def _build_uv_command(
 
     if with_packages:
         for pkg in with_packages:
-            if pkg:
+            if pkg:  # pragma: no branch
                 cmd.extend(["--with", pkg])
 
     # Add mcp run command

src/mcp/client/auth/oauth2.py

@@ -534,7 +534,7 @@ async def async_auth_flow(self, request: httpx.Request) -> AsyncGenerator[httpx.
                     )
 
                     # Step 2: Discover OAuth Authorization Server Metadata (OASM) (with fallback for legacy servers)
-                    for url in asm_discovery_urls:
+                    for url in asm_discovery_urls:  # pragma: no branch
                         oauth_metadata_request = create_oauth_metadata_request(url)
                         oauth_metadata_response = yield oauth_metadata_request
 

src/mcp/client/session_group.py

@@ -223,22 +223,22 @@ async def disconnect_from_server(self, session: mcp.ClientSession) -> None:
                 )
             )
 
-        if session_known_for_components:
+        if session_known_for_components:  # pragma: no branch
             component_names = self._sessions.pop(session)  # Pop from _sessions tracking
 
             # Remove prompts associated with the session.
             for name in component_names.prompts:
-                if name in self._prompts:
+                if name in self._prompts:  # pragma: no branch
                     del self._prompts[name]
             # Remove resources associated with the session.
             for name in component_names.resources:
-                if name in self._resources:
+                if name in self._resources:  # pragma: no branch
                     del self._resources[name]
             # Remove tools associated with the session.
             for name in component_names.tools:
-                if name in self._tools:
+                if name in self._tools:  # pragma: no branch
                     del self._tools[name]
-                if name in self._tool_to_session:
+                if name in self._tool_to_session:  # pragma: no branch
                     del self._tool_to_session[name]
 
         # Clean up the session's resources via its dedicated exit stack

src/mcp/client/sse.py

@@ -119,10 +119,10 @@ async def sse_reader(task_status: TaskStatus[str] = anyio.TASK_STATUS_IGNORED):
                                         await read_stream_writer.send(session_message)
                                     case _:  # pragma: no cover
                                         logger.warning(f"Unknown SSE event: {sse.event}")  # pragma: no cover
-                        except SSEError as sse_exc:
+                        except SSEError as sse_exc:  # pragma: lax no cover
                             logger.exception("Encountered SSE exception")
                             raise sse_exc
-                        except Exception as exc:
+                        except Exception as exc:  # pragma: lax no cover
                             logger.exception("Error in sse_reader")
                             await read_stream_writer.send(exc)
                         finally:
@@ -143,7 +143,7 @@ async def post_writer(endpoint_url: str):
                                     )
                                     response.raise_for_status()
                                     logger.debug(f"Client message sent successfully: {response.status_code}")
-                        except Exception:
+                        except Exception:  # pragma: lax no cover
                             logger.exception("Error in post_writer")
                         finally:
                             await write_stream.aclose()

src/mcp/client/streamable_http.py

@@ -474,16 +474,16 @@ async def handle_request_async():
 
     async def terminate_session(self, client: httpx.AsyncClient) -> None:
         """Terminate the session by sending a DELETE request."""
-        if not self.session_id:  # pragma: no branch
+        if not self.session_id:  # pragma: lax no cover
             return
 
         try:
             headers = self._prepare_headers()
             response = await client.delete(self.url, headers=headers)
 
-            if response.status_code == 405:  # pragma: no branch
+            if response.status_code == 405:  # pragma: lax no cover
                 logger.debug("Server does not allow session termination")
-            elif response.status_code not in (200, 204):  # pragma: no branch
+            elif response.status_code not in (200, 204):  # pragma: lax no cover
                 logger.warning(f"Session termination failed: {response.status_code}")
         except Exception as exc:  # pragma: no cover
             logger.warning(f"Session termination failed: {exc}")

src/mcp/server/auth/handlers/token.py

@@ -178,7 +178,7 @@ async def handle(self, request: Request):
                 except TokenError as e:
                     return self.response(TokenErrorResponse(error=e.error, error_description=e.error_description))
 
-            case RefreshTokenRequest():
+            case RefreshTokenRequest():  # pragma: no branch
                 refresh_token = await self.provider.load_refresh_token(client_info, token_request.refresh_token)
                 if refresh_token is None or refresh_token.client_id != token_request.client_id:
                     # if token belongs to different client, pretend it doesn't exist

src/mcp/server/experimental/task_context.py

@@ -317,7 +317,7 @@ async def elicit_url(
             response_data = await resolver.wait()
             await self._store.update_task(self.task_id, status=TASK_STATUS_WORKING)
             return ElicitResult.model_validate(response_data)
-        except anyio.get_cancelled_exc_class():
+        except anyio.get_cancelled_exc_class():  # pragma: no cover
             await self._store.update_task(self.task_id, status=TASK_STATUS_WORKING)
             raise
 
@@ -486,7 +486,7 @@ async def elicit_as_task(
             await self._store.update_task(self.task_id, status=TASK_STATUS_WORKING)
             return result
 
-        except anyio.get_cancelled_exc_class():
+        except anyio.get_cancelled_exc_class():  # pragma: no cover
             await self._store.update_task(self.task_id, status=TASK_STATUS_WORKING)
             raise
 

src/mcp/server/fastmcp/server.py

@@ -876,7 +876,7 @@ async def handle_sse(scope: Scope, receive: Receive, send: Send):  # pragma: no
         else:
             # Auth is disabled, no need for RequireAuthMiddleware
             # Since handle_sse is an ASGI app, we need to create a compatible endpoint
-            async def sse_endpoint(request: Request) -> Response:
+            async def sse_endpoint(request: Request) -> Response:  # pragma: no cover
                 # Convert the Starlette request to ASGI parameters
                 return await handle_sse(request.scope, request.receive, request._send)  # type: ignore[reportPrivateUsage]
 

src/mcp/server/fastmcp/tools/base.py

@@ -118,7 +118,7 @@ async def run(
 
 
 def _is_async_callable(obj: Any) -> bool:
-    while isinstance(obj, functools.partial):
+    while isinstance(obj, functools.partial):  # pragma: lax no cover
         obj = obj.func
 
     return inspect.iscoroutinefunction(obj) or (