Skip to content

Commit 6c17632

Browse files
committed
refactor test messages
1 parent bb87035 commit 6c17632

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

a2asrv/handler_test.go

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,7 @@ func TestDefaultRequestHandler_OnSendMessage(t *testing.T) {
393393
t.Errorf("OnSendMessageStream() received %d events, want %d", eventI, len(tt.agentEvents))
394394
}
395395
if tt.wantErr != nil && streamErr == nil {
396-
t.Errorf("expected OnSendMessageStream() stream to return error %v", tt.wantErr)
396+
t.Errorf("OnSendMessageStream() error = nil, want %v", tt.wantErr)
397397
}
398398
if tt.wantErr != nil && (streamErr.Error() != tt.wantErr.Error() && !errors.Is(streamErr, tt.wantErr)) {
399399
t.Errorf("OnSendMessageStream() error = %v, wantErr %v", streamErr, tt.wantErr)
@@ -415,7 +415,7 @@ func TestDefaultRequestHandler_OnSendMessage_QueueCreationFails(t *testing.T) {
415415
result, err := handler.OnSendMessage(ctx, &a2a.MessageSendParams{Message: &a2a.Message{}})
416416

417417
if result != nil || !errors.Is(err, wantErr) {
418-
t.Fatalf("expected OnSendMessage() to fail with %v, got: %v, %v", wantErr, result, err)
418+
t.Fatalf("handler.OnSendMessage() = (%v, %v), want error %v", result, err, wantErr)
419419
}
420420
}
421421

@@ -437,7 +437,7 @@ func TestDefaultRequestHandler_OnSendMessage_QueueReadFails(t *testing.T) {
437437
result, err := handler.OnSendMessage(ctx, &a2a.MessageSendParams{Message: &a2a.Message{}})
438438

439439
if result != nil || !errors.Is(err, wantErr) {
440-
t.Fatalf("expected OnSendMessage() to fail with %v, got: %v, %v", wantErr, result, err)
440+
t.Fatalf("handler.OnSendMessage() = (%v, %v), want error %v", result, err, wantErr)
441441
}
442442
}
443443

@@ -452,12 +452,12 @@ func TestDefaultRequestHandler_OnSendMessage_RelatedTaskLoading(t *testing.T) {
452452
request := &a2a.MessageSendParams{Message: &a2a.Message{ReferenceTasks: []a2a.TaskID{a2a.NewTaskID(), existingTask.ID}}}
453453
_, err := handler.OnSendMessage(ctx, request)
454454
if err != nil {
455-
t.Fatalf("OnSendMessage() failed with %v", err)
455+
t.Fatalf("handler.OnSendMessage() failed: %v", err)
456456
}
457457

458458
capturedReqContext := executor.capturedReqContext
459459
if len(capturedReqContext.RelatedTasks) != 1 || capturedReqContext.RelatedTasks[0].ID != existingTask.ID {
460-
t.Fatalf("expected to load existing task %v, got %v", existingTask, capturedReqContext.RelatedTasks)
460+
t.Fatalf("RequestContext.RelatedTasks = %v, want [%v]", capturedReqContext.RelatedTasks, existingTask)
461461
}
462462
}
463463

@@ -482,12 +482,12 @@ func TestDefaultRequestHandler_MultipleRequestContextInterceptors(t *testing.T)
482482

483483
_, err := handler.OnSendMessage(ctx, &a2a.MessageSendParams{Message: &a2a.Message{}})
484484
if err != nil {
485-
t.Fatalf("OnSendMessage() failed with %v", err)
485+
t.Fatalf("handler.OnSendMessage() failed: %v", err)
486486
}
487487

488488
capturedContext := executor.capturedContext
489489
if capturedContext.Value(key1) != val1 || capturedContext.Value(key2) != val2 {
490-
t.Fatalf("expected to find interceptor-attached values, got %v", capturedContext)
490+
t.Fatalf("Execute() context = %+v, want to have interceptor attached values", capturedContext)
491491
}
492492
}
493493

@@ -503,10 +503,10 @@ func TestDefaultRequestHandler_RequestContextInterceptorRejectsRequest(t *testin
503503
_, err := handler.OnSendMessage(ctx, &a2a.MessageSendParams{Message: &a2a.Message{}})
504504

505505
if !errors.Is(err, wantErr) {
506-
t.Fatalf("expected OnSendMessage to failed with %v, got %v", wantErr, err)
506+
t.Fatalf("handler.OnSendMessage() error = %v, want %v", err, wantErr)
507507
}
508508
if executor.executeCalled {
509-
t.Fatalf("expected agent executor to no be called")
509+
t.Fatal("want agent executor to no be called")
510510
}
511511
}
512512

@@ -519,7 +519,7 @@ func TestDefaultRequestHandler_OnSendMessage_AgentExecutionFails(t *testing.T) {
519519
result, err := handler.OnSendMessage(ctx, &a2a.MessageSendParams{Message: &a2a.Message{}})
520520

521521
if result != nil || !errors.Is(err, wantErr) {
522-
t.Fatalf("expected OnSendMessage() to fail with %v, got: %v, %v", wantErr, result, err)
522+
t.Fatalf("handler.OnSendMessage() = (%v, %v), want error %v", result, err, wantErr)
523523
}
524524
}
525525

0 commit comments

Comments
 (0)