@@ -99,9 +99,9 @@ async def never_completes():
9999
100100 # Set up task status progression
101101 mock_session .get_task .side_effect = [
102- GetTaskResult (taskId = "task-1" , status = "submitted" , pollFrequency = 100 ),
103- GetTaskResult (taskId = "task-1" , status = "working" , pollFrequency = 100 ),
104- GetTaskResult (taskId = "task-1" , status = "completed" , pollFrequency = 100 ),
102+ GetTaskResult (taskId = "task-1" , status = "submitted" , pollInterval = 100 ),
103+ GetTaskResult (taskId = "task-1" , status = "working" , pollInterval = 100 ),
104+ GetTaskResult (taskId = "task-1" , status = "completed" , pollInterval = 100 ),
105105 ]
106106 mock_session .get_task_result .return_value = sample_result
107107
@@ -142,8 +142,8 @@ async def never_completes():
142142
143143 # Set up task status progression
144144 task_statuses = [
145- GetTaskResult (taskId = "task-2" , status = "submitted" , pollFrequency = 50 ),
146- GetTaskResult (taskId = "task-2" , status = "completed" , pollFrequency = 50 ),
145+ GetTaskResult (taskId = "task-2" , status = "submitted" , pollInterval = 50 ),
146+ GetTaskResult (taskId = "task-2" , status = "completed" , pollInterval = 50 ),
147147 ]
148148 mock_session .get_task .side_effect = task_statuses
149149 mock_session .get_task_result .return_value = sample_result
@@ -176,7 +176,7 @@ async def notify_created():
176176 async def test_polling_interval_respects_poll_frequency (
177177 self , mock_session : MagicMock , sample_result : CallToolResult
178178 ):
179- """Test that polling interval respects pollFrequency from task."""
179+ """Test that polling interval respects pollInterval from task."""
180180 task_created_event = asyncio .Event ()
181181 task_created_event .set ()
182182
@@ -193,9 +193,9 @@ async def never_completes():
193193 async def mock_get_task (task_id : str ):
194194 poll_times .append (asyncio .get_event_loop ().time ())
195195 if len (poll_times ) == 1 :
196- return GetTaskResult (taskId = task_id , status = "submitted" , pollFrequency = 100 ) # 100ms
196+ return GetTaskResult (taskId = task_id , status = "submitted" , pollInterval = 100 ) # 100ms
197197 else :
198- return GetTaskResult (taskId = task_id , status = "completed" , pollFrequency = 100 )
198+ return GetTaskResult (taskId = task_id , status = "completed" , pollInterval = 100 )
199199
200200 mock_session .get_task .side_effect = mock_get_task
201201 mock_session .get_task_result .return_value = sample_result
@@ -220,7 +220,7 @@ async def mock_get_task(task_id: str):
220220 async def test_polling_uses_default_interval_when_not_specified (
221221 self , mock_session : MagicMock , sample_result : CallToolResult
222222 ):
223- """Test that default polling interval is used when pollFrequency is None."""
223+ """Test that default polling interval is used when pollInterval is None."""
224224 task_created_event = asyncio .Event ()
225225 task_created_event .set ()
226226
@@ -236,9 +236,9 @@ async def never_completes():
236236 async def mock_get_task (task_id : str ):
237237 poll_times .append (asyncio .get_event_loop ().time ())
238238 if len (poll_times ) == 1 :
239- return GetTaskResult (taskId = task_id , status = "submitted" , pollFrequency = None )
239+ return GetTaskResult (taskId = task_id , status = "submitted" , pollInterval = None )
240240 else :
241- return GetTaskResult (taskId = task_id , status = "completed" , pollFrequency = None )
241+ return GetTaskResult (taskId = task_id , status = "completed" , pollInterval = None )
242242
243243 mock_session .get_task .side_effect = mock_get_task
244244 mock_session .get_task_result .return_value = sample_result
@@ -279,7 +279,7 @@ async def get_result():
279279 # Set up task polling to be slow
280280 async def slow_get_task (task_id : str ):
281281 await asyncio .sleep (0.2 )
282- return GetTaskResult (taskId = task_id , status = "submitted" , pollFrequency = 100 )
282+ return GetTaskResult (taskId = task_id , status = "submitted" , pollInterval = 100 )
283283
284284 mock_session .get_task .side_effect = slow_get_task
285285
@@ -319,7 +319,7 @@ async def never_completes():
319319 return sample_result
320320
321321 # Set up task polling to complete quickly
322- mock_session .get_task .return_value = GetTaskResult (taskId = "task-6" , status = "completed" , pollFrequency = 100 )
322+ mock_session .get_task .return_value = GetTaskResult (taskId = "task-6" , status = "completed" , pollInterval = 100 )
323323 mock_session .get_task_result .return_value = sample_result
324324
325325 pending = PendingRequest (
@@ -420,7 +420,7 @@ async def slow_result_with_error():
420420 raise RuntimeError ("Direct result also failed" )
421421
422422 # Task polling succeeds but result retrieval fails
423- mock_session .get_task .return_value = GetTaskResult (taskId = "task-9" , status = "completed" , pollFrequency = 100 )
423+ mock_session .get_task .return_value = GetTaskResult (taskId = "task-9" , status = "completed" , pollInterval = 100 )
424424 mock_session .get_task_result .side_effect = RuntimeError ("Failed to retrieve result" )
425425
426426 pending = PendingRequest (
@@ -454,7 +454,7 @@ async def never_completes():
454454 # Set up task polling to never complete
455455 async def never_complete_get_task (task_id : str ):
456456 await asyncio .sleep (10 )
457- return GetTaskResult (taskId = task_id , status = "submitted" , pollFrequency = 100 )
457+ return GetTaskResult (taskId = task_id , status = "submitted" , pollInterval = 100 )
458458
459459 mock_session .get_task .side_effect = never_complete_get_task
460460
@@ -494,7 +494,7 @@ async def cancellable_get_task(task_id: str):
494494 nonlocal get_task_cancelled
495495 try :
496496 await asyncio .sleep (10 ) # Will be cancelled
497- return GetTaskResult (taskId = task_id , status = "submitted" , pollFrequency = 100 )
497+ return GetTaskResult (taskId = task_id , status = "submitted" , pollInterval = 100 )
498498 except asyncio .CancelledError :
499499 get_task_cancelled = True
500500 raise
@@ -542,7 +542,7 @@ async def never_completes():
542542 await asyncio .Future () # Never completes
543543 return sample_result
544544
545- mock_session .get_task .return_value = GetTaskResult (taskId = "task-12" , status = "completed" , pollFrequency = 100 )
545+ mock_session .get_task .return_value = GetTaskResult (taskId = "task-12" , status = "completed" , pollInterval = 100 )
546546 mock_session .get_task_result .return_value = sample_result
547547
548548 pending = PendingRequest (
@@ -572,8 +572,8 @@ async def never_completes():
572572 return sample_result
573573
574574 mock_session .get_task .side_effect = [
575- GetTaskResult (taskId = "task-13" , status = "submitted" , pollFrequency = 50 ),
576- GetTaskResult (taskId = "task-13" , status = "failed" , pollFrequency = 50 , error = "Something went wrong" ),
575+ GetTaskResult (taskId = "task-13" , status = "submitted" , pollInterval = 50 ),
576+ GetTaskResult (taskId = "task-13" , status = "failed" , pollInterval = 50 , error = "Something went wrong" ),
577577 ]
578578 mock_session .get_task_result .return_value = sample_result
579579
@@ -604,8 +604,8 @@ async def never_completes():
604604 return sample_result
605605
606606 mock_session .get_task .side_effect = [
607- GetTaskResult (taskId = "task-14" , status = "working" , pollFrequency = 50 ),
608- GetTaskResult (taskId = "task-14" , status = "cancelled" , pollFrequency = 50 , error = "User cancelled" ),
607+ GetTaskResult (taskId = "task-14" , status = "working" , pollInterval = 50 ),
608+ GetTaskResult (taskId = "task-14" , status = "cancelled" , pollInterval = 50 , error = "User cancelled" ),
609609 ]
610610 mock_session .get_task_result .return_value = sample_result
611611
0 commit comments