Function RunTestStorage

br/pkg/task/operator/test_storage.go:225–309

RunTestStorage tests all operations of an external storage.

(ctx context.Context, cfg TestStorageConfig)

// RunTestStorage tests all operations of an external storage.
func RunTestStorage(ctx context.Context, cfg TestStorageConfig) error {
	logger := logutil.Logger(ctx)
	logger.Info("Starting external storage test", zap.String("storage", cfg.StorageURI), zap.Any("options", cfg.BackendOptions))

	// Initialize test report
	report := &TestReport{
		StorageURI:  cfg.StorageURI,
		StartTime:   time.Now(),
		TestResults: make([]TestResult, 0),
	}

	// Parse and create storage
	backend, err := storage.ParseBackend(cfg.StorageURI, &cfg.BackendOptions)
	if err != nil {
		return errors.Annotate(err, "failed to parse storage backend")
	}

	store, err := storage.New(ctx, backend, &storage.ExternalStorageOptions{
		SendCredentials: true,
	})
	if err != nil {
		return errors.Annotate(err, "failed to create external storage")
	}
	defer store.Close()

	printStep("Testing external storage: %s", store.URI())
	report.StorageURI = store.URI()

	// Create test context
	tc := &TestContext{
		Ctx:           ctx,
		Report:        report,
		Store:         store,
		PauseWhenFail: cfg.PauseWhenFail,
	}

	// Generate test data
	testData := make([]byte, cfg.TestDataSize)
	if _, err := rand.Read(testData); err != nil {
		return errors.Annotate(err, "failed to generate test data")
	}
	printStep("Generated test data: %s", formatBytes(cfg.TestDataSize))

	cleanup := func() {
		if !cfg.CleanupOnSuccess {
			return
		}
		printStep("Cleaning up test files...")
		_ = store.DeleteFile(ctx, testFileName1)
		_ = store.DeleteFile(ctx, testFileName2)
		_ = store.DeleteFile(ctx, testFileNameRenamed)
		_ = store.DeleteFile(ctx, filepath.Join(testDirName, testFileName1))
		_ = store.DeleteFile(ctx, filepath.Join(testDirName, testFileName2))
	}
	defer cleanup()

	// Run all tests
	testWriteFile(tc, testFileName1, testData)