Function TestEngineIngestAssemblePreservesParts

pkg/seahorse/short_engine_test.go:665–704

(t *testing.T)

}

func TestEngineIngestAssemblePreservesParts(t *testing.T) {
	eng := newTestEngine(t)
	ctx := context.Background()

	// Ingest a message with tool_use parts
	eng.Ingest(ctx, "agent:parts-roundtrip", []Message{
		{Role: "user", Content: "list files", TokenCount: 3},
		{
			Role:       "assistant",
			Content:    "",
			TokenCount: 5,
			Parts: []MessagePart{
				{Type: "tool_use", Name: "bash", Arguments: `{"cmd":"ls"}`, ToolCallID: "tc_1"},
				{Type: "text", Text: "found 3 files"},
			},
		},
	})

	// Assemble should return messages with parts intact
	result, err := eng.Assemble(ctx, "agent:parts-roundtrip", AssembleInput{Budget: 1000})
	if err != nil {
		t.Fatalf("Assemble: %v", err)
	}

	if len(result.Messages) != 2 {
		t.Fatalf("Assemble returned %d messages, want 2", len(result.Messages))
	}

	// The second message should have Parts populated
	assistantMsg := result.Messages[1]
	if len(assistantMsg.Parts) != 2 {
		t.Fatalf("Assembled assistant message Parts = %d, want 2", len(assistantMsg.Parts))
	}
	if assistantMsg.Parts[0].Type != "tool_use" {
		t.Errorf("part[0].Type = %q, want tool_use", assistantMsg.Parts[0].Type)
	}
	if assistantMsg.Parts[0].ToolCallID != "tc_1" {
		t.Errorf("part[0].ToolCallID = %q, want tc_1", assistantMsg.Parts[0].ToolCallID)
	}
}

// --- Session Mutex ---