Convert test strings to uuid

Author: holtskinner

tests/client/test_base_client.py

@@ -65,8 +65,8 @@ async def test_send_message_streaming(
 ):
     async def create_stream(*args, **kwargs):
         yield Task(
-            id='task-123',
-            context_id='ctx-456',
+            id='536ab032-6915-47d1-9909-4172dbee4aa0',
+            context_id='9f18b6e9-63c4-4d44-a8b8-f4648003b6b8',
             status=TaskStatus(state=TaskState.completed),
         )
 
@@ -77,7 +77,7 @@ async def create_stream(*args, **kwargs):
     mock_transport.send_message_streaming.assert_called_once()
     assert not mock_transport.send_message.called
     assert len(events) == 1
-    assert events[0][0].id == 'task-123'
+    assert str(events[0][0].id) == '536ab032-6915-47d1-9909-4172dbee4aa0'
 
 
 @pytest.mark.asyncio
@@ -86,8 +86,8 @@ async def test_send_message_non_streaming(
 ):
     base_client._config.streaming = False
     mock_transport.send_message.return_value = Task(
-        id='task-456',
-        context_id='ctx-789',
+        id='9368e3b5-c796-46cf-9318-6c73e1a37e58',
+        context_id='0a934875-fa22-4af0-8b40-79b13d46e4a6',
         status=TaskStatus(state=TaskState.completed),
     )
 
@@ -96,7 +96,7 @@ async def test_send_message_non_streaming(
     mock_transport.send_message.assert_called_once()
     assert not mock_transport.send_message_streaming.called
     assert len(events) == 1
-    assert events[0][0].id == 'task-456'
+    assert str(events[0][0].id) == '9368e3b5-c796-46cf-9318-6c73e1a37e58'
 
 
 @pytest.mark.asyncio
@@ -105,7 +105,7 @@ async def test_send_message_non_streaming_agent_capability_false(
 ):
     base_client._card.capabilities.streaming = False
     mock_transport.send_message.return_value = Task(
-        id='task-789',
+        id='d7541723-0796-4231-8849-f6f137ea3bf8',
         context_id='dab80cd1-224d-47cd-abd8-cc53101fb273',
         status=TaskStatus(state=TaskState.completed),
     )
@@ -115,4 +115,4 @@ async def test_send_message_non_streaming_agent_capability_false(
     mock_transport.send_message.assert_called_once()
     assert not mock_transport.send_message_streaming.called
     assert len(events) == 1
-    assert events[0][0].id == 'task-789'
+    assert str(events[0][0].id) == 'd7541723-0796-4231-8849-f6f137ea3bf8'

tests/client/test_client_task_manager.py

@@ -99,7 +99,8 @@ async def test_save_task_event_with_artifact_update(
 ):
     await task_manager.save_task_event(sample_task)
     artifact = Artifact(
-        artifact_id='art1', parts=[Part(root=TextPart(text='artifact content'))]
+        artifact_id='c9aa91c0-7f03-44dd-8ffe-6d168b48d3d5',
+        parts=[Part(root=TextPart(text='artifact content'))],
     )
     artifact_update = TaskArtifactUpdateEvent(
         task_id=sample_task.id,
@@ -119,14 +120,14 @@ async def test_save_task_event_creates_task_if_not_exists(
     task_manager: ClientTaskManager,
 ):
     status_update = TaskStatusUpdateEvent(
-        task_id='new_task',
-        context_id='new_context',
+        task_id='b3cd0887-5074-4534-8f02-172a36f304a0',
+        context_id='7f1ee76a-85ee-406d-856d-b9128ff7bb98',
         status=TaskStatus(state=TaskState.working),
         final=False,
     )
     updated_task = await task_manager.save_task_event(status_update)
     assert updated_task is not None
-    assert updated_task.id == 'new_task'
+    assert str(updated_task.id) == 'b3cd0887-5074-4534-8f02-172a36f304a0'
     assert updated_task.status.state == TaskState.working
 
 
@@ -162,7 +163,7 @@ def test_update_with_message_moves_status_message(
     task_manager: ClientTaskManager, sample_task: Task, sample_message: Message
 ):
     status_message = Message(
-        message_id='status_msg',
+        message_id='f82da050-9487-40eb-ae60-a9eda5dc20fc',
         role=Role.agent,
         parts=[Part(root=TextPart(text='Status'))],
     )

tests/client/test_grpc_client.py

@@ -77,7 +77,7 @@ def sample_message_send_params() -> MessageSendParams:
 def sample_task() -> Task:
     """Provides a sample Task object."""
     return Task(
-        id='task-1',
+        id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
         context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
         status=TaskStatus(state=TaskState.completed),
     )

tests/client/test_legacy_client.py

@@ -74,8 +74,8 @@ async def test_a2a_client_send_message(
 
     # Mock the underlying transport's send_message method
     mock_response_task = Task(
-        id='task-123',
-        context_id='ctx-456',
+        id='536ab032-6915-47d1-9909-4172dbee4aa0',
+        context_id='9f18b6e9-63c4-4d44-a8b8-f4648003b6b8',
         status=TaskStatus(state=TaskState.completed),
     )
 
@@ -91,7 +91,7 @@ async def test_a2a_client_send_message(
     )
     response = await client.send_message(request)
 
-    assert response.root.result.id == 'task-123'
+    assert str(response.root.result.id) == '536ab032-6915-47d1-9909-4172dbee4aa0'
 
 
 @pytest.mark.asyncio
@@ -101,15 +101,15 @@ async def test_a2a_grpc_client_get_task(
     client = A2AGrpcClient(grpc_stub=mock_grpc_stub, agent_card=grpc_agent_card)
 
     mock_response_task = Task(
-        id='task-456',
-        context_id='ctx-789',
+        id='9368e3b5-c796-46cf-9318-6c73e1a37e58',
+        context_id='0a934875-fa22-4af0-8b40-79b13d46e4a6',
         status=TaskStatus(state=TaskState.working),
     )
 
     client.get_task = AsyncMock(return_value=mock_response_task)
 
-    params = TaskQueryParams(id='task-456')
+    params = TaskQueryParams(id='9368e3b5-c796-46cf-9318-6c73e1a37e58')
     response = await client.get_task(params)
 
-    assert response.id == 'task-456'
+    assert str(response.id) == '9368e3b5-c796-46cf-9318-6c73e1a37e58'
     client.get_task.assert_awaited_once_with(params)

tests/integration/test_client_server_integration.py

@@ -232,7 +232,7 @@ async def test_http_transport_sends_message_streaming(
 
     message_to_send = Message(
         role=Role.user,
-        message_id='msg-integration-test',
+        message_id='11aec17b-882d-4c14-a1c3-629e04041bb5',
         parts=[Part(root=TextPart(text='Hello, integration test!'))],
     )
     params = MessageSendParams(message=message_to_send)
@@ -276,7 +276,7 @@ def channel_factory(address: str) -> Channel:
 
     message_to_send = Message(
         role=Role.user,
-        message_id='msg-grpc-integration-test',
+        message_id='11aec17b-882d-4c14-a1c3-629e04041bb5',
         parts=[Part(root=TextPart(text='Hello, gRPC integration test!'))],
     )
     params = MessageSendParams(message=message_to_send)
@@ -322,7 +322,7 @@ async def test_http_transport_sends_message_blocking(
 
     message_to_send = Message(
         role=Role.user,
-        message_id='msg-integration-test-blocking',
+        message_id='11aec17b-882d-4c14-a1c3-629e04041bb5',
         parts=[Part(root=TextPart(text='Hello, blocking test!'))],
     )
     params = MessageSendParams(message=message_to_send)
@@ -365,7 +365,7 @@ def channel_factory(address: str) -> Channel:
 
     message_to_send = Message(
         role=Role.user,
-        message_id='msg-grpc-integration-test-blocking',
+        message_id='11aec17b-882d-4c14-a1c3-629e04041bb5',
         parts=[Part(root=TextPart(text='Hello, gRPC blocking test!'))],
     )
     params = MessageSendParams(message=message_to_send)

tests/server/agent_execution/test_context.py

@@ -30,7 +30,11 @@ def mock_params(self, mock_message):
     @pytest.fixture
     def mock_task(self):
         """Fixture for a mock Task."""
-        return Mock(spec=Task, id='task-123', context_id='context-456')
+        return Mock(
+            spec=Task,
+            id='536ab032-6915-47d1-9909-4172dbee4aa0',
+            context_id='context-456',
+        )
 
     def test_init_without_params(self):
         """Test initialization without parameters."""
@@ -66,7 +70,7 @@ def test_init_with_params_no_ids(self, mock_params):
 
     def test_init_with_task_id(self, mock_params):
         """Test initialization with task ID provided."""
-        task_id = 'task-123'
+        task_id = '536ab032-6915-47d1-9909-4172dbee4aa0'
         context = RequestContext(request=mock_params, task_id=task_id)
 
         assert context.task_id == task_id
@@ -82,7 +86,7 @@ def test_init_with_context_id(self, mock_params):
 
     def test_init_with_both_ids(self, mock_params):
         """Test initialization with both task and context IDs provided."""
-        task_id = 'task-123'
+        task_id = '536ab032-6915-47d1-9909-4172dbee4aa0'
         context_id = 'context-456'
         context = RequestContext(
             request=mock_params, task_id=task_id, context_id=context_id

tests/server/request_handlers/test_grpc_handler.py

@@ -69,7 +69,7 @@ async def test_send_message_success(
         )
     )
     response_model = types.Task(
-        id='task-1',
+        id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
         context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
         status=types.TaskStatus(state=types.TaskState.completed),
     )
@@ -80,7 +80,7 @@ async def test_send_message_success(
     mock_request_handler.on_message_send.assert_awaited_once()
     assert isinstance(response, a2a_pb2.SendMessageResponse)
     assert response.HasField('task')
-    assert response.task.id == 'task-1'
+    assert response.task.id == '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
 
 
 @pytest.mark.asyncio
@@ -110,7 +110,7 @@ async def test_get_task_success(
     """Test successful GetTask call."""
     request_proto = a2a_pb2.GetTaskRequest(name='tasks/task-1')
     response_model = types.Task(
-        id='task-1',
+        id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
         context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
         status=types.TaskStatus(state=types.TaskState.working),
     )
@@ -120,7 +120,7 @@ async def test_get_task_success(
 
     mock_request_handler.on_get_task.assert_awaited_once()
     assert isinstance(response, a2a_pb2.Task)
-    assert response.id == 'task-1'
+    assert response.id == '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
 
 
 @pytest.mark.asyncio
@@ -169,7 +169,7 @@ async def test_send_streaming_message(
 
     async def mock_stream():
         yield types.Task(
-            id='task-1',
+            id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
             context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
             status=types.TaskStatus(state=types.TaskState.working),
         )
@@ -186,7 +186,7 @@ async def mock_stream():
 
     assert len(results) == 1
     assert results[0].HasField('task')
-    assert results[0].task.id == 'task-1'
+    assert results[0].task.id == '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
 
 
 @pytest.mark.asyncio
@@ -332,7 +332,7 @@ def side_effect(request, context: ServerCallContext):
             context.activated_extensions.add('foo')
             context.activated_extensions.add('baz')
             return types.Task(
-                id='task-1',
+                id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
                 context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
                 status=types.TaskStatus(state=types.TaskState.completed),
             )
@@ -397,7 +397,7 @@ async def side_effect(request, context: ServerCallContext):
             context.activated_extensions.add('foo')
             context.activated_extensions.add('baz')
             yield types.Task(
-                id='task-1',
+                id='1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7',
                 context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
                 status=types.TaskStatus(state=types.TaskState.working),
             )

tests/server/tasks/test_database_push_notification_config_store.py

@@ -170,7 +170,7 @@ async def test_set_and_get_info_single_config(
     db_store_parameterized: DatabasePushNotificationConfigStore,
 ):
     """Test setting and retrieving a single configuration."""
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config = PushNotificationConfig(id='config-1', url='http://example.com')
 
     await db_store_parameterized.set_info(task_id, config)
@@ -186,7 +186,7 @@ async def test_set_and_get_info_multiple_configs(
 ):
     """Test setting and retrieving multiple configurations for a single task."""
 
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config1 = PushNotificationConfig(id='config-1', url='http://example.com/1')
     config2 = PushNotificationConfig(id='config-2', url='http://example.com/2')
 
@@ -204,7 +204,7 @@ async def test_set_info_updates_existing_config(
     db_store_parameterized: DatabasePushNotificationConfigStore,
 ):
     """Test that setting an existing config ID updates the record."""
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config_id = 'config-1'
     initial_config = PushNotificationConfig(
         id=config_id, url='http://initial.url'
@@ -226,7 +226,7 @@ async def test_set_info_defaults_config_id_to_task_id(
     db_store_parameterized: DatabasePushNotificationConfigStore,
 ):
     """Test that config.id defaults to task_id if not provided."""
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config = PushNotificationConfig(url='http://example.com')  # id is None
 
     await db_store_parameterized.set_info(task_id, config)
@@ -252,7 +252,7 @@ async def test_delete_info_specific_config(
     db_store_parameterized: DatabasePushNotificationConfigStore,
 ):
     """Test deleting a single, specific configuration."""
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config1 = PushNotificationConfig(id='config-1', url='http://a.com')
     config2 = PushNotificationConfig(id='config-2', url='http://b.com')
 
@@ -272,7 +272,7 @@ async def test_delete_info_all_for_task(
 ):
     """Test deleting all configurations for a task when config_id is None."""
 
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config1 = PushNotificationConfig(id='config-1', url='http://a.com')
     config2 = PushNotificationConfig(id='config-2', url='http://b.com')
 
@@ -291,7 +291,9 @@ async def test_delete_info_not_found(
 ):
     """Test that deleting a non-existent config does not raise an error."""
     # Should not raise
-    await db_store_parameterized.delete_info('task-1', 'non-existent-config')
+    await db_store_parameterized.delete_info(
+        '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7', 'non-existent-config'
+    )
 
 
 @pytest.mark.asyncio
@@ -453,7 +455,7 @@ async def test_set_and_get_info_multiple_configs_no_key(
     )
     await store.initialize()
 
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config1 = PushNotificationConfig(id='config-1', url='http://example.com/1')
     config2 = PushNotificationConfig(id='config-2', url='http://example.com/2')
 
@@ -479,7 +481,7 @@ async def test_data_is_not_encrypted_in_db_if_no_key_is_set(
     )
     await store.initialize()
 
-    task_id = 'task-1'
+    task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
     config = PushNotificationConfig(id='config-1', url='http://example.com/1')
     plain_json = config.model_dump_json()