Refine transfer progress and cancellation flow

internal/fs/ops_test.go

@@ -0,0 +1,88 @@
+package vfs
+
+import (
+	"context"
+	"errors"
+	"os"
+	"path/filepath"
+	"strconv"
+	"testing"
+)
+
+func TestCopyPathWithProgressContextRemovesPartialTargetOnCancel(t *testing.T) {
+	t.Parallel()
+
+	root := t.TempDir()
+	srcDir := filepath.Join(root, "src")
+	dstDir := filepath.Join(root, "dst")
+	if err := os.MkdirAll(srcDir, 0o755); err != nil {
+		t.Fatalf("mkdir src: %v", err)
+	}
+	if err := os.MkdirAll(dstDir, 0o755); err != nil {
+		t.Fatalf("mkdir dst: %v", err)
+	}
+
+	for idx := 0; idx < 64; idx++ {
+		path := filepath.Join(srcDir, "file-"+strconv.Itoa(idx)+".txt")
+		if err := os.WriteFile(path, []byte("payload-"+strconv.Itoa(idx)), 0o644); err != nil {
+			t.Fatalf("write source file %d: %v", idx, err)
+		}
+	}
+
+	stats, err := CopyStats(srcDir)
+	if err != nil {
+		t.Fatalf("copy stats: %v", err)
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	_, err = CopyPathWithProgressContext(ctx, srcDir, dstDir, false, stats, func(progress CopyProgress) {
+		if progress.FilesDone >= 1 {
+			cancel()
+		}
+	})
+	if !errors.Is(err, context.Canceled) {
+		t.Fatalf("expected context cancellation, got %v", err)
+	}
+
+	targetPath := filepath.Join(dstDir, filepath.Base(srcDir))
+	if _, statErr := os.Stat(targetPath); !errors.Is(statErr, os.ErrNotExist) {
+		t.Fatalf("expected partial target to be removed, stat err=%v", statErr)
+	}
+}
+
+func TestMovePathWithProgressContextCancelledBeforeStartKeepsSource(t *testing.T) {
+	t.Parallel()
+
+	root := t.TempDir()
+	srcFile := filepath.Join(root, "source.txt")
+	dstDir := filepath.Join(root, "dst")
+	if err := os.WriteFile(srcFile, []byte("payload"), 0o644); err != nil {
+		t.Fatalf("write source: %v", err)
+	}
+	if err := os.MkdirAll(dstDir, 0o755); err != nil {
+		t.Fatalf("mkdir dst: %v", err)
+	}
+
+	stats, err := CopyStats(srcFile)
+	if err != nil {
+		t.Fatalf("copy stats: %v", err)
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	_, err = MovePathWithProgressContext(ctx, srcFile, dstDir, false, stats, nil)
+	if !errors.Is(err, context.Canceled) {
+		t.Fatalf("expected context cancellation, got %v", err)
+	}
+
+	if _, statErr := os.Stat(srcFile); statErr != nil {
+		t.Fatalf("expected source to remain in place, stat err=%v", statErr)
+	}
+	targetPath := filepath.Join(dstDir, filepath.Base(srcFile))
+	if _, statErr := os.Stat(targetPath); !errors.Is(statErr, os.ErrNotExist) {
+		t.Fatalf("expected destination file to be absent, stat err=%v", statErr)
+	}
+}