Skip to content

Commit 4907865

Browse files
authored
style: normalize indentation to 4 spaces for ruff-format compliance
1 parent 0b1d01a commit 4907865

1 file changed

Lines changed: 67 additions & 67 deletions

File tree

tests/issues/test_2001_progress_related_request_id.py

Lines changed: 67 additions & 67 deletions
Original file line numberDiff line numberDiff line change
@@ -13,75 +13,75 @@
1313
pytestmark = pytest.mark.anyio
1414

1515
async def test_report_progress_passes_related_request_id() -> None:
16-
"""report_progress must forward request_id as related_request_id."""
17-
mock_session = AsyncMock()
18-
mock_session.send_progress_notification = AsyncMock()
19-
request_context = ServerRequestContext(
20-
request_id="req-2001",
21-
session=mock_session,
22-
meta={"progress_token": "tok-progress"},
23-
lifespan_context=None,
24-
experimental=Experimental(),
25-
)
26-
ctx = Context(request_context=request_context, mcp_server=MagicMock())
27-
await ctx.report_progress(25, 100, message="quarter done")
28-
await ctx.report_progress(50, 100)
29-
await ctx.report_progress(100, 100, message="complete")
30-
assert mock_session.send_progress_notification.call_count == 3
31-
mock_session.send_progress_notification.assert_any_call(
32-
progress_token="tok-progress",
33-
progress=25.0,
34-
total=100.0,
35-
message="quarter done",
36-
related_request_id="req-2001",
37-
)
38-
mock_session.send_progress_notification.assert_any_call(
39-
progress_token="tok-progress",
40-
progress=50.0,
41-
total=100.0,
42-
message=None,
43-
related_request_id="req-2001",
44-
)
45-
mock_session.send_progress_notification.assert_any_call(
46-
progress_token="tok-progress",
47-
progress=100.0,
48-
total=100.0,
49-
message="complete",
50-
related_request_id="req-2001",
51-
)
16+
"""report_progress must forward request_id as related_request_id."""
17+
mock_session = AsyncMock()
18+
mock_session.send_progress_notification = AsyncMock()
19+
request_context = ServerRequestContext(
20+
request_id="req-2001",
21+
session=mock_session,
22+
meta={"progress_token": "tok-progress"},
23+
lifespan_context=None,
24+
experimental=Experimental(),
25+
)
26+
ctx = Context(request_context=request_context, mcp_server=MagicMock())
27+
await ctx.report_progress(25, 100, message="quarter done")
28+
await ctx.report_progress(50, 100)
29+
await ctx.report_progress(100, 100, message="complete")
30+
assert mock_session.send_progress_notification.call_count == 3
31+
mock_session.send_progress_notification.assert_any_call(
32+
progress_token="tok-progress",
33+
progress=25.0,
34+
total=100.0,
35+
message="quarter done",
36+
related_request_id="req-2001",
37+
)
38+
mock_session.send_progress_notification.assert_any_call(
39+
progress_token="tok-progress",
40+
progress=50.0,
41+
total=100.0,
42+
message=None,
43+
related_request_id="req-2001",
44+
)
45+
mock_session.send_progress_notification.assert_any_call(
46+
progress_token="tok-progress",
47+
progress=100.0,
48+
total=100.0,
49+
message="complete",
50+
related_request_id="req-2001",
51+
)
5252

5353
async def test_report_progress_no_token_skips_notification() -> None:
54-
"""report_progress is a no-op when no progress_token is present."""
55-
mock_session = AsyncMock()
56-
mock_session.send_progress_notification = AsyncMock()
57-
request_context = ServerRequestContext(
58-
request_id="req-no-token",
59-
session=mock_session,
60-
meta={},
61-
lifespan_context=None,
62-
experimental=Experimental(),
63-
)
64-
ctx = Context(request_context=request_context, mcp_server=MagicMock())
65-
await ctx.report_progress(50, 100)
54+
"""report_progress is a no-op when no progress_token is present."""
55+
mock_session = AsyncMock()
56+
mock_session.send_progress_notification = AsyncMock()
57+
request_context = ServerRequestContext(
58+
request_id="req-no-token",
59+
session=mock_session,
60+
meta={},
61+
lifespan_context=None,
62+
experimental=Experimental(),
63+
)
64+
ctx = Context(request_context=request_context, mcp_server=MagicMock())
65+
await ctx.report_progress(50, 100)
6666
mock_session.send_progress_notification.assert_not_called()
6767

6868
async def test_report_progress_integer_token() -> None:
69-
"""report_progress works when progress_token is an integer (e.g. 0)."""
70-
mock_session = AsyncMock()
71-
mock_session.send_progress_notification = AsyncMock()
72-
request_context = ServerRequestContext(
73-
request_id="req-int-token",
74-
session=mock_session,
75-
meta={"progress_token": 0},
76-
lifespan_context=None,
77-
experimental=Experimental(),
78-
)
79-
ctx = Context(request_context=request_context, mcp_server=MagicMock())
80-
await ctx.report_progress(1, 10)
81-
mock_session.send_progress_notification.assert_awaited_once_with(
82-
progress_token=0,
83-
progress=1.0,
84-
total=10.0,
85-
message=None,
86-
related_request_id="req-int-token",
87-
)
69+
"""report_progress works when progress_token is an integer (e.g. 0)."""
70+
mock_session = AsyncMock()
71+
mock_session.send_progress_notification = AsyncMock()
72+
request_context = ServerRequestContext(
73+
request_id="req-int-token",
74+
session=mock_session,
75+
meta={"progress_token": 0},
76+
lifespan_context=None,
77+
experimental=Experimental(),
78+
)
79+
ctx = Context(request_context=request_context, mcp_server=MagicMock())
80+
await ctx.report_progress(1, 10)
81+
mock_session.send_progress_notification.assert_awaited_once_with(
82+
progress_token=0,
83+
progress=1.0,
84+
total=10.0,
85+
message=None,
86+
related_request_id="req-int-token",
87+
)

0 commit comments

Comments
 (0)