| 63 | } |
| 64 | |
| 65 | func TestFixingQueue_WithParts(t *testing.T) { |
| 66 | t.Parallel() |
| 67 | mock := &mockQueue{} |
| 68 | fq := &fixingQueue{queue: mock} |
| 69 | |
| 70 | // Create an artifact update event with actual parts |
| 71 | event := &a2a.TaskArtifactUpdateEvent{ |
| 72 | ContextID: "test-context", |
| 73 | TaskID: "test-task", |
| 74 | Append: true, |
| 75 | Artifact: &a2a.Artifact{ |
| 76 | ID: "test-artifact", |
| 77 | Parts: []a2a.Part{ |
| 78 | a2a.TextPart{Text: "Hello"}, |
| 79 | }, |
| 80 | }, |
| 81 | } |
| 82 | |
| 83 | // Write the event through the fixing queue |
| 84 | err := fq.Write(t.Context(), event) |
| 85 | require.NoError(t, err) |
| 86 | |
| 87 | // Verify the event was written unchanged |
| 88 | require.Len(t, mock.events, 1) |
| 89 | written := mock.events[0].(*a2a.TaskArtifactUpdateEvent) |
| 90 | |
| 91 | assert.Len(t, written.Artifact.Parts, 1) |
| 92 | } |
| 93 | |
| 94 | func TestFixingQueue_NonArtifactEvent(t *testing.T) { |
| 95 | t.Parallel() |