|
13 | 13 | pytestmark = pytest.mark.anyio |
14 | 14 |
|
15 | 15 | 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 | + ) |
52 | 52 |
|
53 | 53 | 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) |
66 | 66 | mock_session.send_progress_notification.assert_not_called() |
67 | 67 |
|
68 | 68 | 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