102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package integration
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
renamercmd "github.com/rogeecn/renamer/cmd"
|
|
"github.com/rogeecn/renamer/internal/history"
|
|
"github.com/rogeecn/renamer/internal/replace"
|
|
)
|
|
|
|
func TestReplaceApplyAndUndo(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
|
|
createFile(t, filepath.Join(tmp, "foo_draft.txt"))
|
|
createFile(t, filepath.Join(tmp, "bar_draft.txt"))
|
|
|
|
parsed, err := replace.ParseArgs([]string{"draft", "final"})
|
|
if err != nil {
|
|
t.Fatalf("ParseArgs error: %v", err)
|
|
}
|
|
|
|
req := &replace.ReplaceRequest{
|
|
WorkingDir: tmp,
|
|
Patterns: parsed.Patterns,
|
|
Replacement: parsed.Replacement,
|
|
}
|
|
|
|
summary, planned, err := replace.Preview(context.Background(), req, parsed, nil)
|
|
if err != nil {
|
|
t.Fatalf("Preview error: %v", err)
|
|
}
|
|
if summary.ChangedCount != 2 {
|
|
t.Fatalf("expected 2 changes, got %d", summary.ChangedCount)
|
|
}
|
|
|
|
entry, err := replace.Apply(context.Background(), req, planned, summary)
|
|
if err != nil {
|
|
t.Fatalf("apply error: %v", err)
|
|
}
|
|
|
|
if len(entry.Operations) != 2 {
|
|
t.Fatalf("expected 2 operations recorded, got %d", len(entry.Operations))
|
|
}
|
|
if entry.Metadata == nil {
|
|
t.Fatalf("expected metadata to be recorded")
|
|
}
|
|
counts, ok := entry.Metadata["patterns"].(map[string]int)
|
|
if !ok {
|
|
t.Fatalf("patterns metadata missing or wrong type: %#v", entry.Metadata)
|
|
}
|
|
if counts["draft"] != 2 {
|
|
t.Fatalf("expected pattern count for 'draft' to be 2, got %d", counts["draft"])
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(tmp, "foo_final.txt")); err != nil {
|
|
t.Fatalf("expected renamed file exists: %v", err)
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(tmp, "bar_final.txt")); err != nil {
|
|
t.Fatalf("expected renamed file exists: %v", err)
|
|
}
|
|
|
|
_, err = history.Undo(tmp)
|
|
if err != nil {
|
|
t.Fatalf("undo error: %v", err)
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(tmp, "foo_draft.txt")); err != nil {
|
|
t.Fatalf("expected original file after undo: %v", err)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(tmp, "bar_draft.txt")); err != nil {
|
|
t.Fatalf("expected original file after undo: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestReplaceCommandInvalidArgs(t *testing.T) {
|
|
root := renamercmd.NewRootCommand()
|
|
var out bytes.Buffer
|
|
root.SetOut(&out)
|
|
root.SetErr(&out)
|
|
root.SetArgs([]string{"replace", "onlyone"})
|
|
|
|
err := root.Execute()
|
|
if err == nil {
|
|
t.Fatalf("expected error for insufficient arguments")
|
|
}
|
|
}
|
|
|
|
func createFile(t *testing.T, path string) {
|
|
t.Helper()
|
|
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
|
|
t.Fatalf("mkdir %s: %v", path, err)
|
|
}
|
|
if err := os.WriteFile(path, []byte("x"), 0o644); err != nil {
|
|
t.Fatalf("write file %s: %v", path, err)
|
|
}
|
|
}
|