@@ -69,7 +69,7 @@ async def test_polling_task_failed_status(self, mcp_server, mocker):
69
69
)
70
70
71
71
# Mock polling to return FAILED status
72
- mock_get = mocker .patch (
72
+ _ = mocker .patch (
73
73
"modelscope_mcp_server.client.ModelScopeClient.get" ,
74
74
new_callable = mocker .AsyncMock ,
75
75
return_value = {
@@ -244,7 +244,7 @@ async def test_missing_output_images_in_success(self, mcp_server, mocker):
244
244
return_value = {"task_id" : "no-images-task" },
245
245
)
246
246
247
- mock_get = mocker .patch (
247
+ _ = mocker .patch (
248
248
"modelscope_mcp_server.client.ModelScopeClient.get" ,
249
249
new_callable = mocker .AsyncMock ,
250
250
return_value = {
@@ -273,7 +273,7 @@ async def test_empty_output_images_array(self, mcp_server, mocker):
273
273
return_value = {"task_id" : "empty-images-task" },
274
274
)
275
275
276
- mock_get = mocker .patch (
276
+ _ = mocker .patch (
277
277
"modelscope_mcp_server.client.ModelScopeClient.get" ,
278
278
new_callable = mocker .AsyncMock ,
279
279
return_value = {
@@ -341,7 +341,7 @@ async def mock_get_side_effect(url, *args, **kwargs):
341
341
return responses .pop (0 )
342
342
return {"task_status" : "PENDING" }
343
343
344
- mock_get = mocker .patch (
344
+ _ = mocker .patch (
345
345
"modelscope_mcp_server.client.ModelScopeClient.get" ,
346
346
new_callable = mocker .AsyncMock ,
347
347
side_effect = mock_get_side_effect ,
@@ -363,11 +363,15 @@ async def mock_get_side_effect(url, *args, **kwargs):
363
363
results = await asyncio .gather (* tasks , return_exceptions = True )
364
364
365
365
# Verify outcomes
366
+ from typing import Any , cast
367
+
366
368
assert not isinstance (results [0 ], Exception ) # First succeeded
367
- assert results [0 ].data .image_url == "https://example.com/image1.jpg"
369
+ result0 = cast (Any , results [0 ])
370
+ assert result0 .data .image_url == "https://example.com/image1.jpg"
368
371
369
372
assert isinstance (results [1 ], Exception ) # Second failed
370
373
assert "Task 2 failed" in str (results [1 ])
371
374
372
375
assert not isinstance (results [2 ], Exception ) # Third succeeded
373
- assert results [2 ].data .image_url == "https://example.com/image3.jpg"
376
+ result2 = cast (Any , results [2 ])
377
+ assert result2 .data .image_url == "https://example.com/image3.jpg"
0 commit comments