test: add benchmarks.

Author: ghosind

all_test.go

@@ -97,6 +97,17 @@ func TestAllWithTimeoutContext(t *testing.T) {
 	a.EqualNow(data, []bool{true, true, false, false, false})
 }
 
+func BenchmarkAll(b *testing.B) {
+	tasks := make([]AsyncFn, 0, 1000)
+	for i := 0; i < 1000; i++ {
+		tasks = append(tasks, func(ctx context.Context) error {
+			return nil
+		})
+	}
+
+	All(tasks...)
+}
+
 func TestAllCompletedWithoutFuncs(t *testing.T) {
 	a := assert.New(t)
 
@@ -191,3 +202,14 @@ func TestAllCompletedWithTimeoutContext(t *testing.T) {
 	a.EqualNow(data, []bool{true, true, false, false, false})
 	a.EqualNow(errs, []error{nil, nil, errTimeout, errTimeout, errTimeout})
 }
+
+func BenchmarkAllCompleted(b *testing.B) {
+	tasks := make([]AsyncFn, 0, 1000)
+	for i := 0; i < 1000; i++ {
+		tasks = append(tasks, func(ctx context.Context) error {
+			return nil
+		})
+	}
+
+	AllCompleted(tasks...)
+}

parallel_test.go

@@ -138,6 +138,17 @@ func TestParallelWithTimedOutContext(t *testing.T) {
 	a.EqualNow(finishedNum, 2)
 }
 
+func BenchmarkParallel(b *testing.B) {
+	tasks := make([]AsyncFn, 0, 1000)
+	for i := 0; i < 1000; i++ {
+		tasks = append(tasks, func(ctx context.Context) error {
+			return nil
+		})
+	}
+
+	Parallel(5, tasks...)
+}
+
 func TestParallelCompleted(t *testing.T) {
 	a := assert.New(t)
 
@@ -283,3 +294,14 @@ func TestParallelCompletedWithTimedOutContext(t *testing.T) {
 	}
 	a.EqualNow(finishedNum, 2)
 }
+
+func BenchmarkParallelCompleted(b *testing.B) {
+	tasks := make([]AsyncFn, 0, 1000)
+	for i := 0; i < 1000; i++ {
+		tasks = append(tasks, func(ctx context.Context) error {
+			return nil
+		})
+	}
+
+	ParallelCompleted(5, tasks...)
+}

race_test.go

@@ -82,3 +82,14 @@ func TestRaceWithContext(t *testing.T) {
 	time.Sleep(300 * time.Millisecond)
 	a.EqualNow(data, []bool{true, true, true, false, false})
 }
+
+func BenchmarkRace(b *testing.B) {
+	tasks := make([]AsyncFn, 0, 1000)
+	for i := 0; i < 1000; i++ {
+		tasks = append(tasks, func(ctx context.Context) error {
+			return nil
+		})
+	}
+
+	Race(tasks...)
+}