test: loop until result on A2A tests (#1706)

Author: Kludex

tests/test_a2a.py

@@ -73,8 +73,10 @@ async def test_a2a_simple():
             assert 'result' in response
             task_id = response['result']['id']
 
-            await anyio.sleep(0.1)
-            task = await a2a_client.get_task(task_id)
+            while task := await a2a_client.get_task(task_id):
+                if 'result' in task and task['result']['status']['state'] == 'completed':
+                    break
+                await anyio.sleep(0.1)
             assert task == snapshot(
                 {
                     'jsonrpc': '2.0',
@@ -137,8 +139,10 @@ async def test_a2a_file_message_with_file():
             assert 'result' in response
             task_id = response['result']['id']
 
-            await anyio.sleep(0.1)
-            task = await a2a_client.get_task(task_id)
+            while task := await a2a_client.get_task(task_id):
+                if 'result' in task and task['result']['status']['state'] == 'completed':
+                    break
+                await anyio.sleep(0.1)
             assert task == snapshot(
                 {
                     'jsonrpc': '2.0',
@@ -203,8 +207,10 @@ async def test_a2a_file_message_with_file_content():
             assert 'result' in response
             task_id = response['result']['id']
 
-            await anyio.sleep(0.1)
-            task = await a2a_client.get_task(task_id)
+            while task := await a2a_client.get_task(task_id):
+                if 'result' in task and task['result']['status']['state'] == 'completed':
+                    break
+                await anyio.sleep(0.1)
             assert task == snapshot(
                 {
                     'jsonrpc': '2.0',
@@ -257,8 +263,10 @@ async def test_a2a_file_message_with_data():
             assert 'result' in response
             task_id = response['result']['id']
 
-            await anyio.sleep(0.1)
-            task = await a2a_client.get_task(task_id)
+            while task := await a2a_client.get_task(task_id):
+                if 'result' in task and task['result']['status']['state'] == 'failed':
+                    break
+                await anyio.sleep(0.1)
             assert task == snapshot(
                 {
                     'jsonrpc': '2.0',