Skip to content

Commit 0972fbb

Browse files
committed
coverage: run ruff on long lines
1 parent d524e49 commit 0972fbb

File tree

13 files changed

+69
-32
lines changed

13 files changed

+69
-32
lines changed

src/mcp/server/auth/provider.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,7 @@ async def authorize(self, client: OAuthClientInformationFull, params: Authorizat
173173
Raises:
174174
AuthorizeError: If the authorization request is invalid.
175175
"""
176+
... # pragma: no cover
176177

177178
async def load_authorization_code(
178179
self, client: OAuthClientInformationFull, authorization_code: str
@@ -187,6 +188,7 @@ async def load_authorization_code(
187188
Returns:
188189
The AuthorizationCode, or None if not found
189190
"""
191+
... # pragma: no cover
190192

191193
async def exchange_authorization_code(
192194
self, client: OAuthClientInformationFull, authorization_code: AuthorizationCodeT
@@ -204,6 +206,7 @@ async def exchange_authorization_code(
204206
Raises:
205207
TokenError: If the request is invalid
206208
"""
209+
... # pragma: no cover
207210

208211
async def load_refresh_token(self, client: OAuthClientInformationFull, refresh_token: str) -> RefreshTokenT | None:
209212
"""
@@ -216,6 +219,7 @@ async def load_refresh_token(self, client: OAuthClientInformationFull, refresh_t
216219
Returns:
217220
The RefreshToken object if found, or None if not found.
218221
"""
222+
... # pragma: no cover
219223

220224
async def exchange_refresh_token(
221225
self,
@@ -239,6 +243,7 @@ async def exchange_refresh_token(
239243
Raises:
240244
TokenError: If the request is invalid
241245
"""
246+
...
242247

243248
async def load_access_token(self, token: str) -> AccessTokenT | None:
244249
"""

src/mcp/server/fastmcp/resources/types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ class DirectoryResource(Resource):
171171

172172
@pydantic.field_validator("path")
173173
@classmethod
174-
def validate_absolute_path(cls, path: Path) -> Path: # pragma: no cover
174+
def validate_absolute_path(cls, path: Path) -> Path: # pragma: no cover
175175
"""Ensure path is absolute."""
176176
if not path.is_absolute():
177177
raise ValueError("Path must be absolute")

src/mcp/server/fastmcp/server.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1225,7 +1225,9 @@ async def log(
12251225
@property
12261226
def client_id(self) -> str | None:
12271227
"""Get the client ID if available."""
1228-
return getattr(self.request_context.meta, "client_id", None) if self.request_context.meta else None # pragma: no cover
1228+
return (
1229+
getattr(self.request_context.meta, "client_id", None) if self.request_context.meta else None
1230+
) # pragma: no cover
12291231

12301232
@property
12311233
def request_id(self) -> str:

src/mcp/server/fastmcp/utilities/func_metadata.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -477,7 +477,9 @@ class DictModel(RootModel[dict_annotation]):
477477

478478

479479
def _get_typed_annotation(annotation: Any, globalns: dict[str, Any]) -> Any:
480-
def try_eval_type(value: Any, globalns: dict[str, Any], localns: dict[str, Any]) -> tuple[Any, bool]: # pragma: no cover
480+
def try_eval_type(
481+
value: Any, globalns: dict[str, Any], localns: dict[str, Any]
482+
) -> tuple[Any, bool]: # pragma: no cover
481483
try:
482484
return eval_type_backport(value, globalns, localns), True
483485
except NameError:

src/mcp/server/lowlevel/server.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -679,7 +679,9 @@ async def _handle_request(
679679
try:
680680
# Extract request context from message metadata
681681
request_data = None
682-
if message.message_metadata is not None and isinstance(message.message_metadata, ServerMessageMetadata): # pragma: no cover
682+
if message.message_metadata is not None and isinstance(
683+
message.message_metadata, ServerMessageMetadata
684+
): # pragma: no cover
683685
request_data = message.message_metadata.request_context
684686

685687
# Set our global state that can be retrieved via

src/mcp/server/streamable_http.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -369,9 +369,11 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
369369
return
370370

371371
# Check if this is an initialization request
372-
is_initialization_request = isinstance(message.root, JSONRPCRequest) and message.root.method == "initialize" # pragma: no cover
372+
is_initialization_request = (
373+
isinstance(message.root, JSONRPCRequest) and message.root.method == "initialize"
374+
) # pragma: no cover
373375

374-
if is_initialization_request: # pragma: no cover
376+
if is_initialization_request: # pragma: no cover
375377
# Check if the server already has an established session
376378
if self.mcp_session_id:
377379
# Check if request has a session ID
@@ -385,11 +387,11 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
385387
)
386388
await response(scope, receive, send)
387389
return
388-
elif not await self._validate_request_headers(request, send): # pragma: no cover
390+
elif not await self._validate_request_headers(request, send): # pragma: no cover
389391
return
390392

391393
# For notifications and responses only, return 202 Accepted
392-
if not isinstance(message.root, JSONRPCRequest): # pragma: no cover
394+
if not isinstance(message.root, JSONRPCRequest): # pragma: no cover
393395
# Create response object and send it
394396
response = self._create_json_response(
395397
None,
@@ -405,10 +407,10 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
405407
return
406408

407409
# Extract the request ID outside the try block for proper scope
408-
request_id = str(message.root.id) # pragma: no cover
410+
request_id = str(message.root.id) # pragma: no cover
409411
# Register this stream for the request ID
410-
self._request_streams[request_id] = anyio.create_memory_object_stream[EventMessage](0) # pragma: no cover
411-
request_stream_reader = self._request_streams[request_id][1] # pragma: no cover
412+
self._request_streams[request_id] = anyio.create_memory_object_stream[EventMessage](0) # pragma: no cover
413+
request_stream_reader = self._request_streams[request_id][1] # pragma: no cover
412414

413415
if self.is_json_response_enabled: # pragma: no cover
414416
# Process the message
@@ -453,7 +455,7 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
453455
await response(scope, receive, send)
454456
finally:
455457
await self._clean_up_memory_streams(request_id)
456-
else: # pragma: no cover
458+
else: # pragma: no cover
457459
# Create SSE stream
458460
sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream[dict[str, str]](0)
459461

@@ -509,7 +511,7 @@ async def sse_writer():
509511
await sse_stream_reader.aclose()
510512
await self._clean_up_memory_streams(request_id)
511513

512-
except Exception as err: # pragma: no cover
514+
except Exception as err: # pragma: no cover
513515
logger.exception("Error handling POST request")
514516
response = self._create_error_response(
515517
f"Error handling POST request: {err}",

src/mcp/shared/session.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,9 @@
4646
class ProgressFnT(Protocol):
4747
"""Protocol for progress notification callbacks."""
4848

49-
async def __call__(self, progress: float, total: float | None, message: str | None) -> None: ... # pragma: no branch
49+
async def __call__(
50+
self, progress: float, total: float | None, message: str | None
51+
) -> None: ... # pragma: no branch
5052

5153

5254
class RequestResponder(Generic[ReceiveRequestT, SendResultT]):

tests/client/test_resource_cleanup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@ async def test_send_request_stream_cleanup():
1919

2020
# Create a mock session with the minimal required functionality
2121
class TestSession(BaseSession[ClientRequest, ClientNotification, ClientResult, Any, Any]):
22-
async def _send_response(self, request_id: RequestId, response: SendResultT | ErrorData) -> None: # pragma: no cover
22+
async def _send_response(
23+
self, request_id: RequestId, response: SendResultT | ErrorData
24+
) -> None: # pragma: no cover
2325
pass
2426

2527
# Create streams

tests/server/fastmcp/test_elicitation.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,9 @@ async def test_stdio_elicitation():
7171
create_ask_user_tool(mcp)
7272

7373
# Create a custom handler for elicitation requests
74-
async def elicitation_callback(context: RequestContext[ClientSession, None], params: ElicitRequestParams): # pragma: no cover
74+
async def elicitation_callback(
75+
context: RequestContext[ClientSession, None], params: ElicitRequestParams
76+
): # pragma: no cover
7577
if params.message == "Tool wants to ask: What is your name?":
7678
return ElicitResult(action="accept", content={"answer": "Test User"})
7779
else:
@@ -126,7 +128,9 @@ class InvalidNestedSchema(BaseModel):
126128
create_validation_tool("nested_model", InvalidNestedSchema)
127129

128130
# Dummy callback (won't be called due to validation failure)
129-
async def elicitation_callback(context: RequestContext[ClientSession, None], params: ElicitRequestParams): # pragma: no cover
131+
async def elicitation_callback(
132+
context: RequestContext[ClientSession, None], params: ElicitRequestParams
133+
): # pragma: no cover
130134
return ElicitResult(action="accept", content={})
131135

132136
async with create_connected_server_and_client_session(
@@ -203,7 +207,9 @@ async def invalid_optional_tool(ctx: Context[ServerSession, None]) -> str: # pr
203207
except TypeError as e:
204208
return f"Validation failed: {str(e)}"
205209

206-
async def elicitation_callback(context: RequestContext[ClientSession, None], params: ElicitRequestParams): # pragma: no cover
210+
async def elicitation_callback(
211+
context: RequestContext[ClientSession, None], params: ElicitRequestParams
212+
): # pragma: no cover
207213
return ElicitResult(action="accept", content={})
208214

209215
await call_tool_and_assert(

tests/server/fastmcp/test_tool_manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,9 @@ def tool_without_context(x: int) -> str: # pragma: no cover
347347
tool = manager.add_tool(tool_without_context)
348348
assert tool.context_kwarg is None
349349

350-
def tool_with_parametrized_context(x: int, ctx: Context[ServerSessionT, LifespanContextT, RequestT]) -> str: # pragma: no cover
350+
def tool_with_parametrized_context(
351+
x: int, ctx: Context[ServerSessionT, LifespanContextT, RequestT]
352+
) -> str: # pragma: no cover
351353
return str(x)
352354

353355
tool = manager.add_tool(tool_with_parametrized_context)

0 commit comments

Comments
 (0)