Function TestBackgroundCleanupLifecycle

pkg/export/disk/disk_test.go:1320–1492

(t *testing.T)

}

func TestBackgroundCleanupLifecycle(t *testing.T) {
	t.Run("Background cleanup starts on first CreateConnection", func(t *testing.T) {
		writer := &Writer{
			mu:                sync.RWMutex{},
			openConnections:   make(map[string]Connection),
			closedConnections: make(map[string]FailedConnection),
			cleanupDone:       make(chan struct{}),
			cleanupOnce:       sync.Once{},
			cleanupStopped:    false,
			closeAndRemoveFilesWithRetry: func(_ Connection) error {
				return nil
			},
		}

		ctx := context.Background()
		tmpDir := t.TempDir()
		config := map[string]interface{}{
			"path":            tmpDir,
			"maxAuditResults": 5.0,
		}

		err := writer.CreateConnection(ctx, "test-conn", config)
		if err != nil {
			t.Fatalf("CreateConnection failed: %v", err)
		}

		time.Sleep(100 * time.Millisecond)

		writer.mu.RLock()
		_, exists := writer.openConnections["test-conn"]
		cleanupStopped := writer.cleanupStopped
		writer.mu.RUnlock()

		if !exists {
			t.Error("Connection was not created")
		}

		if cleanupStopped {
			t.Error("Expected cleanup to be running, but it's stopped")
		}

		writer.mu.Lock()
		if !writer.cleanupStopped {
			writer.cleanupStopped = true
			close(writer.cleanupDone)
		}
		writer.mu.Unlock()
	})

	t.Run("Background cleanup stops when all connections are closed", func(t *testing.T) {
		writer := &Writer{
			mu:                sync.RWMutex{},
			openConnections:   make(map[string]Connection),
			closedConnections: make(map[string]FailedConnection),
			cleanupDone:       make(chan struct{}),
			cleanupOnce:       sync.Once{},
			cleanupStopped:    false,
			closeAndRemoveFilesWithRetry: func(_ Connection) error {