Function TestFileReferencePreservation

coderd/x/chatd/chatprompt/chatprompt_test.go:1503–1546

TestFileReferencePreservation verifies file-reference parts survive the storage round-trip and convert to text for LLMs.

(t *testing.T)

// TestFileReferencePreservation verifies file-reference parts
// survive the storage round-trip and convert to text for LLMs.
func TestFileReferencePreservation(t *testing.T) {
	t.Parallel()

	raw, err := chatprompt.MarshalParts([]codersdk.ChatMessagePart{{
		Type:      codersdk.ChatMessagePartTypeFileReference,
		FileName:  "main.go",
		StartLine: 10,
		EndLine:   20,
		Content:   "func main() {}",
	}})
	require.NoError(t, err)

	// Storage round-trip: all fields intact.
	parts, err := chatprompt.ParseContent(testMsg(codersdk.ChatMessageRoleUser, raw))
	require.NoError(t, err)
	require.Len(t, parts, 1)
	assert.Equal(t, codersdk.ChatMessagePartTypeFileReference, parts[0].Type)
	assert.Equal(t, "main.go", parts[0].FileName)
	assert.Equal(t, 10, parts[0].StartLine)
	assert.Equal(t, 20, parts[0].EndLine)
	assert.Equal(t, "func main() {}", parts[0].Content)

	// LLM dispatch: file-reference becomes a TextPart.
	prompt, err := chatprompt.ConvertMessagesWithFiles(
		context.Background(),
		[]database.ChatMessage{{
			Role:       database.ChatMessageRoleUser,
			Visibility: database.ChatMessageVisibilityBoth,
			Content:    raw,
		}},
		nil,
		slogtest.Make(t, nil),
	)
	require.NoError(t, err)
	require.Len(t, prompt, 1)
	require.Len(t, prompt[0].Content, 1)

	textPart, ok := fantasy.AsMessagePart[fantasy.TextPart](prompt[0].Content[0])
	require.True(t, ok, "file-reference should become TextPart for LLM")
	assert.Contains(t, textPart.Text, "[file-reference]")
	assert.Contains(t, textPart.Text, "main.go")
	assert.Contains(t, textPart.Text, "10-20")
	assert.Contains(t, textPart.Text, "func main() {}")
}

// TestAssistantWriteRoundTrip verifies the Stage 4 write path:
// fantasy.Content (with ProviderMetadata) → PartFromContent →