Function TestFileTracker_RollbackAllFiles

pkg/cli/file_tracker_test.go:186–251

(t *testing.T)

}

func TestFileTracker_RollbackAllFiles(t *testing.T) {
	// Create a temporary directory for testing
	tempDir, err := os.MkdirTemp("", "file-tracker-rollback-test")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}
	defer os.RemoveAll(tempDir)

	// Initialize git repository in temp directory
	gitCmd := []string{"git", "init"}
	if err := runCommandInDir(gitCmd, tempDir); err != nil {
		t.Skipf("Skipping test - git not available or failed to init: %v", err)
	}

	// Change to temp directory
	oldWd, _ := os.Getwd()
	defer func() {
		_ = os.Chdir(oldWd)
	}()
	if err := os.Chdir(tempDir); err != nil {
		t.Fatalf("Failed to change to temp directory: %v", err)
	}

	// Create file tracker
	tracker := NewFileTracker()

	// Create an existing file
	existingFile := filepath.Join(tempDir, "existing.md")
	originalContent := "# Original Content"
	if err := os.WriteFile(existingFile, []byte(originalContent), 0644); err != nil {
		t.Fatalf("Failed to write existing file: %v", err)
	}

	// Track modification before modifying
	tracker.TrackModified(existingFile)
	modifiedContent := "# Modified Content"
	if err := os.WriteFile(existingFile, []byte(modifiedContent), 0644); err != nil {
		t.Fatalf("Failed to modify existing file: %v", err)
	}

	// Create a new file
	newFile := filepath.Join(tempDir, "new.md")
	tracker.TrackCreated(newFile)
	if err := os.WriteFile(newFile, []byte("# New Content"), 0644); err != nil {
		t.Fatalf("Failed to write new file: %v", err)
	}

	// Rollback all files
	if err := tracker.RollbackAllFiles(false); err != nil {
		t.Errorf("Failed to rollback all files: %v", err)
	}

	// Verify new file was deleted
	if _, err := os.Stat(newFile); !os.IsNotExist(err) {
		t.Errorf("New file %s should have been deleted", newFile)
	}

	// Verify existing file was restored to original content