feat: check nil context.

Author: ghosind

all.go

@@ -19,6 +19,10 @@ func all(parent context.Context, funcs ...func(context.Context) error) error {
 		return nil
 	}
 
+	if parent == nil {
+		parent = context.Background()
+	}
+
 	ctx, canFunc := context.WithCancel(parent)
 	errCh := make(chan error)
 	retCh := make(chan struct{}, len(funcs))
@@ -82,6 +86,10 @@ func allCompleted(parent context.Context, funcs ...func(context.Context) error)
 		return
 	}
 
+	if parent == nil {
+		parent = context.Background()
+	}
+
 	wg := sync.WaitGroup{}
 	wg.Add(len(funcs))
 

all_test.go

@@ -58,6 +58,17 @@ func TestAllFailure(t *testing.T) {
 	a.EqualNow(data, []bool{true, true, false, false, false})
 }
 
+func TestAllWithNilContext(t *testing.T) {
+	a := assert.New(t)
+
+	//lint:ignore SA1012 for test case only
+	err := AllWithContext(nil, func(ctx context.Context) error {
+		time.Sleep(100 * time.Millisecond)
+		return nil
+	})
+	a.NilNow(err)
+}
+
 func TestAllWithTimeoutContext(t *testing.T) {
 	a := assert.New(t)
 
@@ -134,6 +145,18 @@ func TestAllCompletedPartialFailure(t *testing.T) {
 	a.EqualNow(errs, []error{nil, nil, errNIs2, nil, nil})
 }
 
+func TestAllCompletedWithNilContext(t *testing.T) {
+	a := assert.New(t)
+
+	//lint:ignore SA1012 for test case only
+	errs, hasError := AllCompletedWithContext(nil, func(ctx context.Context) error {
+		time.Sleep(100 * time.Millisecond)
+		return nil
+	})
+	a.NotTrueNow(hasError)
+	a.EqualNow(errs, []error{nil})
+}
+
 func TestAllCompletedWithTimeoutContext(t *testing.T) {
 	a := assert.New(t)
 

race.go

@@ -18,6 +18,10 @@ func race(ctx context.Context, funcs ...func(context.Context) error) error {
 		return nil
 	}
 
+	if ctx == nil {
+		ctx = context.Background()
+	}
+
 	finished := atomic.Bool{}
 	ch := make(chan error)
 	defer close(ch)

race_test.go

@@ -38,6 +38,17 @@ func TestRace(t *testing.T) {
 	a.EqualNow(data, []bool{true, true, true, true, true})
 }
 
+func TestRaceWithNilContext(t *testing.T) {
+	a := assert.New(t)
+
+	//lint:ignore SA1012 for test case only
+	err := RaceWithContext(nil, func(ctx context.Context) error {
+		time.Sleep(100 * time.Millisecond)
+		return nil
+	})
+	a.NilNow(err)
+}
+
 func TestRaceWithContext(t *testing.T) {
 	a := assert.New(t)