pool: add WithCancelOnError and panic test case for ResultContextPool

Author: bobheadxi

pool/result_context_pool_test.go

@@ -8,6 +8,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
 	"github.com/sourcegraph/sourcegraph/lib/errors"
@@ -100,6 +101,29 @@ func TestResultContextPool(t *testing.T) {
 		require.ErrorIs(t, err, err1)
 	})
 
+	t.Run("WithCancelOnError and panic", func(t *testing.T) {
+		t.Parallel()
+		p := NewWithResults[int]().
+			WithContext(context.Background()).
+			WithCancelOnError()
+		var cancelledTasks atomic.Int64
+		p.Go(func(ctx context.Context) (int, error) {
+			<-ctx.Done()
+			cancelledTasks.Add(1)
+			return 0, ctx.Err()
+		})
+		p.Go(func(ctx context.Context) (int, error) {
+			<-ctx.Done()
+			cancelledTasks.Add(1)
+			return 0, ctx.Err()
+		})
+		p.Go(func(ctx context.Context) (int, error) {
+			panic("abort!")
+		})
+		assert.Panics(t, func() { _, _ = p.Wait() })
+		assert.EqualValues(t, 2, cancelledTasks.Load())
+	})
+
 	t.Run("no WithCancelOnError", func(t *testing.T) {
 		t.Parallel()
 		g := NewWithResults[int]().WithContext(context.Background())