Skip to content

Commit 2dabcc8

Browse files
committed
Skip coverage for wrapper methods
send_ping and the list_changed methods also follow this
1 parent db2d295 commit 2dabcc8

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/mcp/server/session.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -552,14 +552,16 @@ async def send_prompt_list_changed(self) -> None: # pragma: no cover
552552
"""Send a prompt list changed notification."""
553553
await self.send_notification(types.ServerNotification(types.PromptListChangedNotification()))
554554

555-
async def get_task(self, task_id: str) -> types.GetTaskResult:
555+
async def get_task(self, task_id: str) -> types.GetTaskResult: # pragma: no cover
556556
"""Get the current status of a task."""
557557
return await self.send_request(
558558
types.ServerRequest(types.GetTaskRequest(method="tasks/get", params=types.GetTaskParams(taskId=task_id))),
559559
types.GetTaskResult,
560560
)
561561

562-
async def get_task_result(self, task_id: str, result_type: type[ServerResultT]) -> ServerResultT:
562+
async def get_task_result(
563+
self, task_id: str, result_type: type[ServerResultT]
564+
) -> ServerResultT: # pragma: no cover
563565
"""Retrieve the result of a completed task."""
564566
return await self.send_request(
565567
types.ServerRequest(
@@ -568,7 +570,7 @@ async def get_task_result(self, task_id: str, result_type: type[ServerResultT])
568570
result_type,
569571
)
570572

571-
async def list_tasks(self, cursor: str | None = None) -> types.ListTasksResult:
573+
async def list_tasks(self, cursor: str | None = None) -> types.ListTasksResult: # pragma: no cover
572574
"""List tasks, optionally starting from a pagination cursor."""
573575
return await self.send_request(
574576
types.ServerRequest(
@@ -579,7 +581,7 @@ async def list_tasks(self, cursor: str | None = None) -> types.ListTasksResult:
579581
types.ListTasksResult,
580582
)
581583

582-
async def delete_task(self, task_id: str) -> types.EmptyResult:
584+
async def delete_task(self, task_id: str) -> types.EmptyResult: # pragma: no cover
583585
"""Delete a specific task."""
584586
return await self.send_request(
585587
types.ServerRequest(

0 commit comments

Comments
 (0)