Skip to content

Commit 205f646

Browse files
holtskinnerholtskinner
committed
Update UUIDs
1 parent 5898802 commit 205f646

16 files changed

+337
-223
lines changed

‎tests/client/test_jsonrpc_client.py‎

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,9 @@ async def test_send_message_success(
365365
role=Role.agent, content='Hi there!'
366366
)
367367
rpc_response = SendMessageSuccessResponse(
368-
id='123', jsonrpc='2.0', result=success_response
368+
id='5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
369+
jsonrpc='2.0',
370+
result=success_response,
369371
)
370372
response = httpx.Response(
371373
200, json=rpc_response.model_dump(mode='json')
@@ -390,7 +392,7 @@ async def test_send_message_error_response(
390392
)
391393
error_response = InvalidParamsError()
392394
rpc_response = {
393-
'id': '123',
395+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
394396
'jsonrpc': '2.0',
395397
'error': error_response.model_dump(exclude_none=True),
396398
}
@@ -541,7 +543,7 @@ async def test_get_task_success(
541543
)
542544
params = TaskQueryParams(id='ea719c56-e398-425e-b02c-49fd77b7c156')
543545
rpc_response = {
544-
'id': '123',
546+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
545547
'jsonrpc': '2.0',
546548
'result': MINIMAL_TASK,
547549
}
@@ -569,7 +571,7 @@ async def test_cancel_task_success(
569571
)
570572
params = TaskIdParams(id='ea719c56-e398-425e-b02c-49fd77b7c156')
571573
rpc_response = {
572-
'id': '123',
574+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
573575
'jsonrpc': '2.0',
574576
'result': MINIMAL_CANCELLED_TASK,
575577
}
@@ -602,7 +604,7 @@ async def test_set_task_callback_success(
602604
),
603605
)
604606
rpc_response = {
605-
'id': '123',
607+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
606608
'jsonrpc': '2.0',
607609
'result': params.model_dump(mode='json'),
608610
}
@@ -633,7 +635,7 @@ async def test_get_task_callback_success(
633635
),
634636
)
635637
rpc_response = {
636-
'id': '123',
638+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
637639
'jsonrpc': '2.0',
638640
'result': expected_response.model_dump(mode='json'),
639641
}
@@ -763,7 +765,7 @@ async def test_get_card_with_extended_card_support(
763765
)
764766

765767
rpc_response = {
766-
'id': '123',
768+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
767769
'jsonrpc': '2.0',
768770
'result': AGENT_CARD_EXTENDED.model_dump(mode='json'),
769771
}

‎tests/server/agent_execution/test_simple_request_context_builder.py‎

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def create_sample_message(
4141

4242
# Helper to create a simple task
4343
def create_sample_task(
44-
task_id='task1',
44+
task_id='13d5b8a8-62d7-4490-98c8-d3951b42702a',
4545
status_state=TaskState.submitted,
4646
context_id='06cc947f-8946-4bde-b776-165462407e57',
4747
):
@@ -87,7 +87,7 @@ async def test_build_basic_context_no_populate(self):
8787

8888
params = MessageSendParams(message=create_sample_message())
8989
task_id = 'ea57b599-5bff-4f72-8376-6e47b1e3bac0'
90-
context_id = '1a4bd17f-952b-4c19-9d5e-603ae0cab8cd_1'
90+
context_id = '1a4bd17f-952b-4c19-9d5e-603ae0cab8cd'
9191
current_task = create_sample_task(
9292
task_id=task_id, context_id=context_id
9393
)
@@ -123,9 +123,9 @@ async def test_build_populate_true_with_reference_task_ids(self):
123123
builder = SimpleRequestContextBuilder(
124124
should_populate_referred_tasks=True, task_store=self.mock_task_store
125125
)
126-
ref_task_id1 = 'ref_task1'
126+
ref_task_id1 = '11aec17b-882d-4c14-a1c3-629e04041bb5'
127127
ref_task_id2 = 'ref_task2_missing'
128-
ref_task_id3 = 'ref_task3'
128+
ref_task_id3 = 'd6853c8e-834d-4877-be4c-aa2fec76f755'
129129

130130
mock_ref_task1 = create_sample_task(task_id=ref_task_id1)
131131
mock_ref_task3 = create_sample_task(task_id=ref_task_id3)
@@ -213,7 +213,10 @@ async def test_build_populate_true_reference_ids_empty_or_none(self):
213213
# To explicitly test None in Message, we'd have to bypass Pydantic default or modify helper.
214214
# For now, this covers the "no IDs to process" case.
215215
msg_with_no_refs = Message(
216-
message_id='m2', role=Role.user, parts=[], referenceTaskIds=None
216+
message_id='11aec17b-882d-4c14-a1c3-629e04041bb5',
217+
role=Role.user,
218+
parts=[],
219+
referenceTaskIds=None,
217220
)
218221
params_none_refs = MessageSendParams(message=msg_with_no_refs)
219222
request_context_none = await builder.build(
@@ -236,7 +239,9 @@ async def test_build_populate_true_task_store_none(self):
236239
task_store=None, # Explicitly None
237240
)
238241
params = MessageSendParams(
239-
message=create_sample_message(reference_task_ids=['ref1'])
242+
message=create_sample_message(
243+
reference_task_ids=['11aec17b-882d-4c14-a1c3-629e04041bb5']
244+
)
240245
)
241246
server_call_context = ServerCallContext(user=UnauthenticatedUser())
242247

@@ -258,7 +263,7 @@ async def test_build_populate_false_with_reference_task_ids(self):
258263
)
259264
params = MessageSendParams(
260265
message=create_sample_message(
261-
reference_task_ids=['ref_task_should_not_be_fetched']
266+
reference_task_ids=['11aec17b-882d-4c14-a1c3-629e04041bb5']
262267
)
263268
)
264269
server_call_context = ServerCallContext(user=UnauthenticatedUser())

‎tests/server/apps/jsonrpc/test_jsonrpc_app.py‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def mock_handler(self):
191191
handler = AsyncMock(spec=RequestHandler)
192192
handler.on_message_send.return_value = SendMessageResponse(
193193
root=SendMessageSuccessResponse(
194-
id='1',
194+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
195195
result=Message(
196196
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
197197
role=Role.agent,
@@ -221,10 +221,10 @@ def test_request_with_single_extension(self, client, mock_handler):
221221
'/',
222222
headers=headers,
223223
json=SendMessageRequest(
224-
id='1',
224+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
225225
params=MessageSendParams(
226226
message=Message(
227-
message_id='1',
227+
message_id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
228228
role=Role.user,
229229
parts=[Part(TextPart(text='hi'))],
230230
)
@@ -246,10 +246,10 @@ def test_request_with_comma_separated_extensions(
246246
'/',
247247
headers=headers,
248248
json=SendMessageRequest(
249-
id='1',
249+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
250250
params=MessageSendParams(
251251
message=Message(
252-
message_id='1',
252+
message_id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
253253
role=Role.user,
254254
parts=[Part(TextPart(text='hi'))],
255255
)
@@ -273,10 +273,10 @@ def test_request_with_comma_separated_extensions_no_space(
273273
'/',
274274
headers=headers,
275275
json=SendMessageRequest(
276-
id='1',
276+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
277277
params=MessageSendParams(
278278
message=Message(
279-
message_id='1',
279+
message_id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
280280
role=Role.user,
281281
parts=[Part(TextPart(text='hi'))],
282282
)
@@ -300,10 +300,10 @@ def test_request_with_multiple_extension_headers(
300300
'/',
301301
headers=headers,
302302
json=SendMessageRequest(
303-
id='1',
303+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
304304
params=MessageSendParams(
305305
message=Message(
306-
message_id='1',
306+
message_id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
307307
role=Role.user,
308308
parts=[Part(TextPart(text='hi'))],
309309
)
@@ -322,7 +322,7 @@ def side_effect(request, context: ServerCallContext):
322322
context.activated_extensions.add('baz')
323323
return SendMessageResponse(
324324
root=SendMessageSuccessResponse(
325-
id='1',
325+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
326326
result=Message(
327327
message_id='d2c4511e-11e1-4cda-b3fd-c24cfe8a7119',
328328
role=Role.agent,
@@ -336,10 +336,10 @@ def side_effect(request, context: ServerCallContext):
336336
response = client.post(
337337
'/',
338338
json=SendMessageRequest(
339-
id='1',
339+
id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
340340
params=MessageSendParams(
341341
message=Message(
342-
message_id='1',
342+
message_id='8a9ca3aa-1f78-4e4d-8e8f-b8228f02ed32',
343343
role=Role.user,
344344
parts=[Part(TextPart(text='hi'))],
345345
)

‎tests/server/apps/jsonrpc/test_serialization.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ def test_handle_unicode_characters(agent_card_with_api_key: AgentCard):
172172
handler.on_message_send.return_value = Message(
173173
role=Role.agent,
174174
parts=[Part(root=TextPart(text=f'Received: {unicode_text}'))],
175-
message_id='response-unicode',
175+
message_id='d6853c8e-834d-4877-be4c-aa2fec76f755',
176176
)
177177

178178
response = client.post('/', json=unicode_payload)

‎tests/server/events/test_event_consumer.py‎

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828

2929
MINIMAL_TASK: dict[str, Any] = {
30-
'id': '123',
30+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
3131
'context_id': '598c0e6f-72c2-48fc-803a-15d693622c6f',
3232
'status': {'state': 'submitted'},
3333
'kind': 'task',
@@ -36,7 +36,7 @@
3636
MESSAGE_PAYLOAD: dict[str, Any] = {
3737
'role': 'agent',
3838
'parts': [{'text': 'test message'}],
39-
'message_id': '111',
39+
'message_id': '2e888b8b-6d81-4505-a8ec-9220dc3c508f',
4040
}
4141

4242

@@ -128,14 +128,15 @@ async def test_consume_all_multiple_events(
128128
events: list[Any] = [
129129
Task(**MINIMAL_TASK),
130130
TaskArtifactUpdateEvent(
131-
task_id='task_123',
131+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
132132
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
133133
artifact=Artifact(
134-
artifact_id='11', parts=[Part(TextPart(text='text'))]
134+
artifact_id='d2323590-71c6-4a4b-8d53-b3f92fe8d1c7',
135+
parts=[Part(TextPart(text='text'))],
135136
),
136137
),
137138
TaskStatusUpdateEvent(
138-
task_id='task_123',
139+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
139140
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
140141
status=TaskStatus(state=TaskState.working),
141142
final=True,
@@ -170,15 +171,16 @@ async def test_consume_until_message(
170171
events: list[Any] = [
171172
Task(**MINIMAL_TASK),
172173
TaskArtifactUpdateEvent(
173-
task_id='task_123',
174+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
174175
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
175176
artifact=Artifact(
176-
artifact_id='11', parts=[Part(TextPart(text='text'))]
177+
artifact_id='d2323590-71c6-4a4b-8d53-b3f92fe8d1c7',
178+
parts=[Part(TextPart(text='text'))],
177179
),
178180
),
179181
Message(**MESSAGE_PAYLOAD),
180182
TaskStatusUpdateEvent(
181-
task_id='task_123',
183+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
182184
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
183185
status=TaskStatus(state=TaskState.working),
184186
final=True,
@@ -276,7 +278,7 @@ async def test_consume_all_continues_on_queue_empty_if_not_really_closed(
276278
):
277279
"""Test that QueueClosed with is_closed=False allows loop to continue via timeout."""
278280
payload = MESSAGE_PAYLOAD.copy()
279-
payload['message_id'] = 'final_event_id'
281+
payload['message_id'] = 'cce10a17-1f2f-48c5-9afe-43f97389a476'
280282
final_event = Message(**payload)
281283

282284
# Setup dequeue_event behavior:

‎tests/server/events/test_event_queue.py‎

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,15 +27,15 @@
2727

2828

2929
MINIMAL_TASK: dict[str, Any] = {
30-
'id': '123',
30+
'id': '5bb3c918-28c9-4d1f-8ca6-8ddc85c91863',
3131
'context_id': '598c0e6f-72c2-48fc-803a-15d693622c6f',
3232
'status': {'state': 'submitted'},
3333
'kind': 'task',
3434
}
3535
MESSAGE_PAYLOAD: dict[str, Any] = {
3636
'role': 'agent',
3737
'parts': [{'text': 'test message'}],
38-
'message_id': '111',
38+
'message_id': '2e888b8b-6d81-4505-a8ec-9220dc3c508f',
3939
}
4040

4141

@@ -100,7 +100,7 @@ async def test_dequeue_event_empty_queue_no_wait(
100100
async def test_dequeue_event_wait(event_queue: EventQueue) -> None:
101101
"""Test dequeue_event with the default wait behavior."""
102102
event = TaskStatusUpdateEvent(
103-
task_id='task_123',
103+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
104104
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
105105
status=TaskStatus(state=TaskState.working),
106106
final=True,
@@ -114,10 +114,11 @@ async def test_dequeue_event_wait(event_queue: EventQueue) -> None:
114114
async def test_task_done(event_queue: EventQueue) -> None:
115115
"""Test the task_done method."""
116116
event = TaskArtifactUpdateEvent(
117-
task_id='task_123',
117+
task_id='43305029-d2b0-4494-a61b-1c7980bf259d',
118118
context_id='598c0e6f-72c2-48fc-803a-15d693622c6f',
119119
artifact=Artifact(
120-
artifact_id='11', parts=[Part(TextPart(text='text'))]
120+
artifact_id='d2323590-71c6-4a4b-8d53-b3f92fe8d1c7',
121+
parts=[Part(TextPart(text='text'))],
121122
),
122123
)
123124
await event_queue.enqueue_event(event)

0 commit comments

Comments
 (0)