File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -145,7 +145,7 @@ async def test_on_message_send_throws_error_for_unsupported_version(
145145 # 0.3 is currently not supported for direct message sending on RestDispatcher
146146 req = make_mock_request (method = 'POST' , headers = {'a2a-version' : '0.3.0' })
147147 response = await rest_dispatcher_instance .on_message_send (req )
148-
148+
149149 # VersionNotSupportedError maps to 400 Bad Request
150150 assert response .status_code == 400
151151
@@ -302,26 +302,24 @@ async def test_on_message_send_stream_success(
302302 ):
303303 req = make_mock_request (method = 'POST' )
304304 response = await rest_dispatcher_instance .on_message_send_stream (req )
305-
305+
306306 assert response .status_code == 200
307-
307+
308308 chunks = []
309309 async for chunk in response .body_iterator :
310310 chunks .append (chunk )
311-
311+
312312 assert len (chunks ) == 2
313313 # sse-starlette yields strings or bytes formatted as Server-Sent Events
314314 assert 'chunk1' in str (chunks [0 ])
315315 assert 'chunk2' in str (chunks [1 ])
316316
317- async def test_on_subscribe_to_task_success (
318- self , rest_dispatcher_instance
319- ):
317+ async def test_on_subscribe_to_task_success (self , rest_dispatcher_instance ):
320318 req = make_mock_request (method = 'GET' , path_params = {'id' : 'test_task' })
321319 response = await rest_dispatcher_instance .on_subscribe_to_task (req )
322-
320+
323321 assert response .status_code == 200
324-
322+
325323 chunks = []
326324 async for chunk in response .body_iterator :
327325 chunks .append (chunk )
You can’t perform that action at this time.
0 commit comments