Skip to content

Commit 13bce87

Browse files
committed
Update more strings to uuid
1 parent 2485fa2 commit 13bce87

File tree

10 files changed

+37
-34
lines changed

10 files changed

+37
-34
lines changed

tests/client/test_base_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ async def test_send_message_non_streaming_agent_capability_false(
106106
base_client._card.capabilities.streaming = False
107107
mock_transport.send_message.return_value = Task(
108108
id='task-789',
109-
context_id='ctx-101',
109+
context_id='dab80cd1-224d-47cd-abd8-cc53101fb273',
110110
status=TaskStatus(state=TaskState.completed),
111111
)
112112

tests/client/test_grpc_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def sample_task() -> Task:
7878
"""Provides a sample Task object."""
7979
return Task(
8080
id='task-1',
81-
context_id='ctx-1',
81+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
8282
status=TaskStatus(state=TaskState.completed),
8383
)
8484

@@ -88,7 +88,7 @@ def sample_message() -> Message:
8888
"""Provides a sample Message object."""
8989
return Message(
9090
role=Role.agent,
91-
message_id='msg-response',
91+
message_id='a76b49b6-494b-47fb-b416-36e3f2d622da',
9292
parts=[Part(root=TextPart(text='Hi there'))],
9393
)
9494

tests/server/agent_execution/test_simple_request_context_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,8 @@ async def test_build_basic_context_no_populate(self):
8686
)
8787

8888
params = MessageSendParams(message=create_sample_message())
89-
task_id = 'test_task_id_1'
90-
context_id = 'test_context_id_1'
89+
task_id = 'ea57b599-5bff-4f72-8376-6e47b1e3bac0'
90+
context_id = '1a4bd17f-952b-4c19-9d5e-603ae0cab8cd_1'
9191
current_task = create_sample_task(
9292
task_id=task_id, context_id=context_id
9393
)

tests/server/apps/jsonrpc/test_jsonrpc_app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ def mock_handler(self):
193193
root=SendMessageSuccessResponse(
194194
id='1',
195195
result=Message(
196-
message_id='test',
196+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
197197
role=Role.agent,
198198
parts=[Part(TextPart(text='response message'))],
199199
),
@@ -324,7 +324,7 @@ def side_effect(request, context: ServerCallContext):
324324
root=SendMessageSuccessResponse(
325325
id='1',
326326
result=Message(
327-
message_id='test',
327+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
328328
role=Role.agent,
329329
parts=[Part(TextPart(text='response message'))],
330330
),

tests/server/apps/rest/test_rest_fastapi_app.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -146,15 +146,15 @@ async def test_send_message_success_message(
146146
) -> None:
147147
expected_response = a2a_pb2.SendMessageResponse(
148148
msg=a2a_pb2.Message(
149-
message_id='test',
149+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
150150
role=a2a_pb2.Role.ROLE_AGENT,
151151
content=[
152152
a2a_pb2.Part(text='response message'),
153153
],
154154
),
155155
)
156156
request_handler.on_message_send.return_value = Message(
157-
message_id='test',
157+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
158158
role=Role.agent,
159159
parts=[Part(TextPart(text='response message'))],
160160
)
@@ -181,12 +181,12 @@ async def test_send_message_success_task(
181181
) -> None:
182182
expected_response = a2a_pb2.SendMessageResponse(
183183
task=a2a_pb2.Task(
184-
id='test_task_id',
185-
context_id='test_context_id',
184+
id='2cc58515-03ae-4122-a9a5-560507c822aa',
185+
context_id='1a4bd17f-952b-4c19-9d5e-603ae0cab8cd',
186186
status=a2a_pb2.TaskStatus(
187187
state=a2a_pb2.TaskState.TASK_STATE_COMPLETED,
188188
update=a2a_pb2.Message(
189-
message_id='test',
189+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
190190
role=a2a_pb2.ROLE_AGENT,
191191
content=[
192192
a2a_pb2.Part(text='response task message'),
@@ -196,12 +196,12 @@ async def test_send_message_success_task(
196196
),
197197
)
198198
request_handler.on_message_send.return_value = Task(
199-
id='test_task_id',
200-
context_id='test_context_id',
199+
id='2cc58515-03ae-4122-a9a5-560507c822aa',
200+
context_id='1a4bd17f-952b-4c19-9d5e-603ae0cab8cd',
201201
status=TaskStatus(
202202
state=TaskState.completed,
203203
message=Message(
204-
message_id='test',
204+
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
205205
role=Role.agent,
206206
parts=[Part(TextPart(text='response task message'))],
207207
),

tests/server/events/test_inmemory_queue_manager.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,14 @@ async def test_init(self, queue_manager):
3535
@pytest.mark.asyncio
3636
async def test_add_new_queue(self, queue_manager, event_queue):
3737
"""Test adding a new queue to the manager."""
38-
task_id = 'test_task_id'
38+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
3939
await queue_manager.add(task_id, event_queue)
4040
assert queue_manager._task_queue[task_id] == event_queue
4141

4242
@pytest.mark.asyncio
4343
async def test_add_existing_queue(self, queue_manager, event_queue):
4444
"""Test adding a queue with an existing task_id raises TaskQueueExists."""
45-
task_id = 'test_task_id'
45+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
4646
await queue_manager.add(task_id, event_queue)
4747

4848
with pytest.raises(TaskQueueExists):
@@ -51,7 +51,7 @@ async def test_add_existing_queue(self, queue_manager, event_queue):
5151
@pytest.mark.asyncio
5252
async def test_get_existing_queue(self, queue_manager, event_queue):
5353
"""Test getting an existing queue returns the queue."""
54-
task_id = 'test_task_id'
54+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
5555
await queue_manager.add(task_id, event_queue)
5656

5757
result = await queue_manager.get(task_id)
@@ -66,7 +66,7 @@ async def test_get_nonexistent_queue(self, queue_manager):
6666
@pytest.mark.asyncio
6767
async def test_tap_existing_queue(self, queue_manager, event_queue):
6868
"""Test tapping an existing queue returns the tapped queue."""
69-
task_id = 'test_task_id'
69+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
7070
await queue_manager.add(task_id, event_queue)
7171

7272
result = await queue_manager.tap(task_id)
@@ -82,7 +82,7 @@ async def test_tap_nonexistent_queue(self, queue_manager):
8282
@pytest.mark.asyncio
8383
async def test_close_existing_queue(self, queue_manager, event_queue):
8484
"""Test closing an existing queue removes it from the manager."""
85-
task_id = 'test_task_id'
85+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
8686
await queue_manager.add(task_id, event_queue)
8787

8888
await queue_manager.close(task_id)
@@ -97,7 +97,7 @@ async def test_close_nonexistent_queue(self, queue_manager):
9797
@pytest.mark.asyncio
9898
async def test_create_or_tap_new_queue(self, queue_manager):
9999
"""Test create_or_tap with a new task_id creates and returns a new queue."""
100-
task_id = 'test_task_id'
100+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
101101

102102
result = await queue_manager.create_or_tap(task_id)
103103
assert isinstance(result, EventQueue)
@@ -108,7 +108,7 @@ async def test_create_or_tap_existing_queue(
108108
self, queue_manager, event_queue
109109
):
110110
"""Test create_or_tap with an existing task_id taps and returns the existing queue."""
111-
task_id = 'test_task_id'
111+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
112112
await queue_manager.add(task_id, event_queue)
113113

114114
result = await queue_manager.create_or_tap(task_id)

tests/server/request_handlers/test_grpc_handler.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ async def test_send_message_success(
7070
)
7171
response_model = types.Task(
7272
id='task-1',
73-
context_id='ctx-1',
73+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
7474
status=types.TaskStatus(state=types.TaskState.completed),
7575
)
7676
mock_request_handler.on_message_send.return_value = response_model
@@ -111,7 +111,7 @@ async def test_get_task_success(
111111
request_proto = a2a_pb2.GetTaskRequest(name='tasks/task-1')
112112
response_model = types.Task(
113113
id='task-1',
114-
context_id='ctx-1',
114+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
115115
status=types.TaskStatus(state=types.TaskState.working),
116116
)
117117
mock_request_handler.on_get_task.return_value = response_model
@@ -170,7 +170,7 @@ async def test_send_streaming_message(
170170
async def mock_stream():
171171
yield types.Task(
172172
id='task-1',
173-
context_id='ctx-1',
173+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
174174
status=types.TaskStatus(state=types.TaskState.working),
175175
)
176176

@@ -333,7 +333,7 @@ def side_effect(request, context: ServerCallContext):
333333
context.activated_extensions.add('baz')
334334
return types.Task(
335335
id='task-1',
336-
context_id='ctx-1',
336+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
337337
status=types.TaskStatus(state=types.TaskState.completed),
338338
)
339339

@@ -398,7 +398,7 @@ async def side_effect(request, context: ServerCallContext):
398398
context.activated_extensions.add('baz')
399399
yield types.Task(
400400
id='task-1',
401-
context_id='ctx-1',
401+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
402402
status=types.TaskStatus(state=types.TaskState.working),
403403
)
404404

tests/server/request_handlers/test_jsonrpc_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ async def test_on_get_task_success(self) -> None:
104104
)
105105
call_context = ServerCallContext(state={'foo': 'bar'})
106106
handler = JSONRPCHandler(self.mock_agent_card, request_handler)
107-
task_id = 'test_task_id'
107+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
108108
mock_task = Task(**MINIMAL_TASK)
109109
mock_task_store.get.return_value = mock_task
110110
request = GetTaskRequest(id='1', params=TaskQueryParams(id=task_id))
@@ -142,7 +142,7 @@ async def test_on_cancel_task_success(self) -> None:
142142
mock_agent_executor, mock_task_store
143143
)
144144
handler = JSONRPCHandler(self.mock_agent_card, request_handler)
145-
task_id = 'test_task_id'
145+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
146146
mock_task = Task(**MINIMAL_TASK)
147147
mock_task_store.get.return_value = mock_task
148148
mock_agent_executor.cancel.return_value = None
@@ -169,7 +169,7 @@ async def test_on_cancel_task_not_supported(self) -> None:
169169
mock_agent_executor, mock_task_store
170170
)
171171
handler = JSONRPCHandler(self.mock_agent_card, request_handler)
172-
task_id = 'test_task_id'
172+
task_id = '1c3a35ab-e35c-49d8-a37b-7988f5a2ecb7'
173173
mock_task = Task(**MINIMAL_TASK)
174174
mock_task_store.get.return_value = mock_task
175175
mock_agent_executor.cancel.return_value = None

tests/test_types.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1534,7 +1534,7 @@ def test_subclass_enums() -> None:
15341534
def test_get_task_push_config_params() -> None:
15351535
"""Tests successful validation of GetTaskPushNotificationConfigParams."""
15361536
# Minimal valid data
1537-
params = {'id': 'task-1234'}
1537+
params = {'id': 'cf810732-24e2-4f81-a235-f318de16d6f7'}
15381538
TaskIdParams.model_validate(params)
15391539
GetTaskPushNotificationConfigParams.model_validate(params)
15401540

@@ -1545,7 +1545,10 @@ def test_use_get_task_push_notification_params_for_request() -> None:
15451545
'id': 1,
15461546
'jsonrpc': '2.0',
15471547
'method': 'tasks/pushNotificationConfig/get',
1548-
'params': {'id': 'task-1234', 'pushNotificationConfigId': 'c1'},
1548+
'params': {
1549+
'id': 'cf810732-24e2-4f81-a235-f318de16d6f7',
1550+
'pushNotificationConfigId': 'c1',
1551+
},
15491552
}
15501553
a2a_req_get_push_req = A2ARequest.model_validate(get_push_notif_req_data)
15511554
assert isinstance(

tests/utils/test_proto_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
def sample_message() -> types.Message:
1616
return types.Message(
1717
message_id='15957e91-63e6-40ac-8205-1d1ffb09a5b2',
18-
context_id='ctx-1',
18+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
1919
task_id='task-1',
2020
role=types.Role.user,
2121
parts=[
@@ -35,7 +35,7 @@ def sample_message() -> types.Message:
3535
def sample_task(sample_message: types.Message) -> types.Task:
3636
return types.Task(
3737
id='task-1',
38-
context_id='ctx-1',
38+
context_id='e1bbdfd5-8818-4200-873f-8124135770fe',
3939
status=types.TaskStatus(
4040
state=types.TaskState.working, message=sample_message
4141
),

0 commit comments

Comments
 (0)