1 files changed, 86 insertions, 0 deletions
diff --git a/fstest/test_all/clean.go b/fstest/test_all/clean.go
new file mode 100644
index 0000000..a6f52d1
--- /dev/null
+++ b/fstest/test_all/clean.go
@@ -0,0 +1,86 @@
+// Clean the left over test files
+
+package main
+
+import (
+	"context"
+	"fmt"
+	"regexp"
+
+	"github.com/rclone/rclone/fs"
+	"github.com/rclone/rclone/fs/fspath"
+	"github.com/rclone/rclone/fs/list"
+	"github.com/rclone/rclone/fs/operations"
+	"github.com/rclone/rclone/fstest/runs"
+)
+
+// MatchTestRemote matches the remote names used for testing (copied
+// from fstest/fstest.go so we don't have to import that and get all
+// its flags)
+var MatchTestRemote = regexp.MustCompile(`^rclone-test-[abcdefghijklmnopqrstuvwxyz0123456789]{12,24}(_segments)?$`)
+
+// cleanFs runs a single clean fs for left over directories
+func cleanFs(ctx context.Context, remote string, cleanup bool, Opt runs.RunOpt) error {
+	f, err := fs.NewFs(context.Background(), remote)
+	if err != nil {
+		return err
+	}
+	var lastErr error
+	if cleanup {
+		fs.Logf(nil, "%q - running cleanup", remote)
+		err = operations.CleanUp(ctx, f)
+		if err != nil {
+			lastErr = err
+			fs.Errorf(f, "Cleanup failed: %v", err)
+		}
+	}
+	entries, err := list.DirSorted(ctx, f, true, "")
+	if err != nil {
+		return err
+	}
+	err = entries.ForDirError(func(dir fs.Directory) error {
+		dirPath := dir.Remote()
+		fullPath := fspath.JoinRootPath(remote, dirPath)
+		if MatchTestRemote.MatchString(dirPath) {
+			if Opt.DryRun {
+				fs.Logf(nil, "Not Purging %s - -dry-run", fullPath)
+				return nil
+			}
+			fs.Logf(nil, "Purging %s", fullPath)
+			dir, err := fs.NewFs(context.Background(), fullPath)
+			if err != nil {
+				err = fmt.Errorf("NewFs failed: %w", err)
+				lastErr = err
+				fs.Errorf(fullPath, "%v", err)
+				return nil
+			}
+			err = operations.Purge(ctx, dir, "")
+			if err != nil {
+				err = fmt.Errorf("purge failed: %w", err)
+				lastErr = err
+				fs.Errorf(dir, "%v", err)
+				return nil
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+	return lastErr
+}
+
+// cleanRemotes cleans the list of remotes passed in
+func cleanRemotes(conf *runs.Config, Opt runs.RunOpt) error {
+	var lastError error
+	for _, backend := range conf.Backends {
+		remote := backend.Remote
+		fs.Logf(nil, "%q - Cleaning", remote)
+		err := cleanFs(context.Background(), remote, backend.CleanUp, Opt)
+		if err != nil {
+			lastError = err
+			fs.Logf(nil, "Failed to purge %q: %v", remote, err)
+		}
+	}
+	return lastError
+}