add ai feature
This commit is contained in:
46
tests/integration/ai_preview_regen_test.go
Normal file
46
tests/integration/ai_preview_regen_test.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package integration
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/rogeecn/renamer/cmd"
|
||||
)
|
||||
|
||||
func TestAICommandSupportsPromptRefinement(t *testing.T) {
|
||||
t.Setenv("RENAMER_AI_KEY", "test-key")
|
||||
|
||||
tmp := t.TempDir()
|
||||
createFile(t, filepath.Join(tmp, "IMG_1024.jpg"))
|
||||
createFile(t, filepath.Join(tmp, "notes/day1.txt"))
|
||||
|
||||
root := cmd.NewRootCommand()
|
||||
root.SetArgs([]string{"ai", "--path", tmp})
|
||||
|
||||
// Simulate editing the prompt then quitting.
|
||||
var output bytes.Buffer
|
||||
input := strings.NewReader("e\nVacation Highlights\nq\n")
|
||||
root.SetIn(input)
|
||||
root.SetOut(&output)
|
||||
root.SetErr(&output)
|
||||
|
||||
if err := root.Execute(); err != nil {
|
||||
t.Fatalf("ai command returned error: %v\noutput: %s", err, output.String())
|
||||
}
|
||||
|
||||
got := output.String()
|
||||
|
||||
if !strings.Contains(got, "Current prompt: \"Vacation Highlights\"") {
|
||||
t.Fatalf("expected updated prompt in output, got: %s", got)
|
||||
}
|
||||
|
||||
if !strings.Contains(got, "01.vacation-highlights-img-1024.jpg") {
|
||||
t.Fatalf("expected regenerated suggestion with new prefix, got: %s", got)
|
||||
}
|
||||
|
||||
if !strings.Contains(got, "Session ended without applying changes.") {
|
||||
t.Fatalf("expected session completion message, got: %s", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user