Refactored, added comprehensive testing.

prompt/prompt_test.go

@@ -0,0 +1,214 @@
+package prompt
+
+import (
+	"testing"
+
+	"git.wisehodl.dev/jay/aicli/input"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFormatPrompts(t *testing.T) {
+	tests := []struct {
+		name    string
+		prompts []string
+		want    string
+	}{
+		{
+			name:    "empty array returns empty string",
+			prompts: []string{},
+			want:    "",
+		},
+		{
+			name:    "single prompt unchanged",
+			prompts: []string{"analyze this"},
+			want:    "analyze this",
+		},
+		{
+			name:    "multiple prompts joined with newline",
+			prompts: []string{"first", "second", "third"},
+			want:    "first\nsecond\nthird",
+		},
+		{
+			name:    "prompts with trailing newlines preserved",
+			prompts: []string{"line one\n", "line two\n"},
+			want:    "line one\n\nline two\n",
+		},
+		{
+			name:    "empty string in array produces empty line",
+			prompts: []string{"first", "", "third"},
+			want:    "first\n\nthird",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := formatPrompts(tt.prompts)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestFormatFiles(t *testing.T) {
+	tests := []struct {
+		name  string
+		files []input.FileData
+		want  string
+	}{
+		{
+			name:  "empty array returns empty string",
+			files: []input.FileData{},
+			want:  "",
+		},
+		{
+			name: "single file formatted with template",
+			files: []input.FileData{
+				{Path: "main.go", Content: "package main"},
+			},
+			want: "File: main.go\n\n```\npackage main\n```",
+		},
+		{
+			name: "multiple files separated by double newline",
+			files: []input.FileData{
+				{Path: "a.go", Content: "code a"},
+				{Path: "b.go", Content: "code b"},
+			},
+			want: "File: a.go\n\n```\ncode a\n```\n\nFile: b.go\n\n```\ncode b\n```",
+		},
+		{
+			name: "stdin path 'input' appears correctly",
+			files: []input.FileData{
+				{Path: "input", Content: "stdin content"},
+			},
+			want: "File: input\n\n```\nstdin content\n```",
+		},
+		{
+			name: "file path with directory",
+			files: []input.FileData{
+				{Path: "src/main.go", Content: "package main"},
+			},
+			want: "File: src/main.go\n\n```\npackage main\n```",
+		},
+		{
+			name: "content with backticks still wrapped",
+			files: []input.FileData{
+				{Path: "test.md", Content: "```go\nfunc main() {}\n```"},
+			},
+			want: "File: test.md\n\n```\n```go\nfunc main() {}\n```\n```",
+		},
+		{
+			name: "empty content",
+			files: []input.FileData{
+				{Path: "empty.txt", Content: ""},
+			},
+			want: "File: empty.txt\n\n```\n\n```",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := formatFiles(tt.files)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestCombineContent(t *testing.T) {
+	tests := []struct {
+		name      string
+		promptStr string
+		filesStr  string
+		want      string
+	}{
+		{
+			name:      "both empty returns empty",
+			promptStr: "",
+			filesStr:  "",
+			want:      "",
+		},
+		{
+			name:      "prompt only",
+			promptStr: "analyze this",
+			filesStr:  "",
+			want:      "analyze this",
+		},
+		{
+			name:      "files only uses default prompt",
+			promptStr: "",
+			filesStr:  "File: a.go\n\n```\ncode\n```",
+			want:      "Analyze the following:\n\nFile: a.go\n\n```\ncode\n```",
+		},
+		{
+			name:      "prompt and files combined with separator",
+			promptStr: "review this code",
+			filesStr:  "File: a.go\n\n```\ncode\n```",
+			want:      "review this code\n\nFile: a.go\n\n```\ncode\n```",
+		},
+		{
+			name:      "multiline prompt preserved",
+			promptStr: "first line\nsecond line",
+			filesStr:  "File: a.go\n\n```\ncode\n```",
+			want:      "first line\nsecond line\n\nFile: a.go\n\n```\ncode\n```",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := combineContent(tt.promptStr, tt.filesStr)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestConstructQuery(t *testing.T) {
+	tests := []struct {
+		name    string
+		prompts []string
+		files   []input.FileData
+		want    string
+	}{
+		{
+			name:    "empty inputs returns empty",
+			prompts: []string{},
+			files:   []input.FileData{},
+			want:    "",
+		},
+		{
+			name:    "prompt only",
+			prompts: []string{"analyze this"},
+			files:   []input.FileData{},
+			want:    "analyze this",
+		},
+		{
+			name:    "file only with default prompt",
+			prompts: []string{},
+			files: []input.FileData{
+				{Path: "main.go", Content: "package main"},
+			},
+			want: "Analyze the following:\n\nFile: main.go\n\n```\npackage main\n```",
+		},
+		{
+			name:    "multiple prompts and files",
+			prompts: []string{"review", "focus on bugs"},
+			files: []input.FileData{
+				{Path: "a.go", Content: "code a"},
+				{Path: "b.go", Content: "code b"},
+			},
+			want: "review\nfocus on bugs\n\nFile: a.go\n\n```\ncode a\n```\n\nFile: b.go\n\n```\ncode b\n```",
+		},
+		{
+			name:    "stdin as file with explicit prompt",
+			prompts: []string{"analyze"},
+			files: []input.FileData{
+				{Path: "input", Content: "stdin data"},
+			},
+			want: "analyze\n\nFile: input\n\n```\nstdin data\n```",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := ConstructQuery(tt.prompts, tt.files)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}