grokkit/cmd/lint_test.go

package cmd

import (
	"strings"
	"testing"

	"gmgauthier.com/grokkit/internal/linter"
)

func TestBuildLintFixMessages(t *testing.T) {
	tests := []struct {
		name     string
		result   *linter.LintResult
		code     string
		wantLen  int
		sysCheck string
		usrCheck []string
	}{
		{
			name: "go file with issues",
			result: &linter.LintResult{
				Language:   "Go",
				LinterUsed: "golangci-lint",
				Output:     "line 5: unused variable x",
			},
			code:     "package main\nfunc main() { x := 1 }",
			wantLen:  2,
			sysCheck: "code quality",
			usrCheck: []string{"Go", "golangci-lint", "unused variable x", "package main"},
		},
		{
			name: "python file with issues",
			result: &linter.LintResult{
				Language:   "Python",
				LinterUsed: "flake8",
				Output:     "E501 line too long",
			},
			code:     "def foo():\n    pass",
			wantLen:  2,
			sysCheck: "code quality",
			usrCheck: []string{"Python", "flake8", "E501 line too long", "def foo"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			msgs := buildLintFixMessages(tt.result, tt.code)

			if len(msgs) != tt.wantLen {
				t.Fatalf("expected %d messages, got %d", tt.wantLen, len(msgs))
			}
			if msgs[0]["role"] != "system" {
				t.Errorf("first message role = %q, want %q", msgs[0]["role"], "system")
			}
			if msgs[1]["role"] != "user" {
				t.Errorf("second message role = %q, want %q", msgs[1]["role"], "user")
			}
			if !strings.Contains(strings.ToLower(msgs[0]["content"]), strings.ToLower(tt.sysCheck)) {
				t.Errorf("system prompt missing %q; got: %s", tt.sysCheck, msgs[0]["content"])
			}
			for _, check := range tt.usrCheck {
				if !strings.Contains(msgs[1]["content"], check) {
					t.Errorf("user prompt missing %q", check)
				}
			}
		})
	}
}