From 09ee8df5a0d5e9c042fc18f55789b148c35f5ad3 Mon Sep 17 00:00:00 2001
From: Ken Hibino <ken.hibino7@gmail.com>
Date: Tue, 31 Dec 2019 12:36:46 -0800
Subject: [PATCH] Add end-to-end benchmark tests

---
 asynq_test.go     |  6 ++--
 benchmark_test.go | 91 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+), 3 deletions(-)
 create mode 100644 benchmark_test.go

diff --git a/asynq_test.go b/asynq_test.go
index 95c0dda..3f96d1c 100644
--- a/asynq_test.go
+++ b/asynq_test.go
@@ -12,14 +12,14 @@ import (
 // This file defines test helper functions used by
 // other test files.
 
-func setup(t *testing.T) *redis.Client {
-	t.Helper()
+func setup(tb testing.TB) *redis.Client {
+	tb.Helper()
 	r := redis.NewClient(&redis.Options{
 		Addr: "localhost:6379",
 		DB:   14,
 	})
 	// Start each test with a clean slate.
-	h.FlushDB(t, r)
+	h.FlushDB(tb, r)
 	return r
 }
 
diff --git a/benchmark_test.go b/benchmark_test.go
new file mode 100644
index 0000000..00c4a6a
--- /dev/null
+++ b/benchmark_test.go
@@ -0,0 +1,91 @@
+package asynq
+
+import (
+	"fmt"
+	"math/rand"
+	"sync"
+	"testing"
+	"time"
+)
+
+// Simple E2E Benchmark testing with no scheduled tasks and
+// no retries.
+func BenchmarkEndToEndSimple(b *testing.B) {
+	const count = 100000
+	for n := 0; n < b.N; n++ {
+		b.StopTimer() // begin setup
+		r := setup(b)
+		client := NewClient(r)
+		bg := NewBackground(r, &Config{
+			Concurrency: 10,
+			RetryDelayFunc: func(n int, err error, t *Task) time.Duration {
+				return time.Second
+			},
+		})
+		// Create a bunch of tasks
+		for i := 0; i < count; i++ {
+			t := Task{Type: fmt.Sprintf("task%d", i), Payload: Payload{"data": i}}
+			client.Process(&t, time.Now())
+		}
+
+		var wg sync.WaitGroup
+		wg.Add(count)
+		handler := func(t *Task) error {
+			wg.Done()
+			return nil
+		}
+		b.StartTimer() // end setup
+
+		bg.start(HandlerFunc(handler))
+		wg.Wait()
+
+		b.StopTimer() // begin teardown
+		bg.stop()
+		b.StartTimer() // end teardown
+	}
+}
+
+// E2E benchmark with scheduled tasks and retries.
+func BenchmarkEndToEnd(b *testing.B) {
+	const count = 100000
+	for n := 0; n < b.N; n++ {
+		b.StopTimer() // begin setup
+		rand.Seed(time.Now().UnixNano())
+		r := setup(b)
+		client := NewClient(r)
+		bg := NewBackground(r, &Config{
+			Concurrency: 10,
+			RetryDelayFunc: func(n int, err error, t *Task) time.Duration {
+				return time.Second
+			},
+		})
+		// Create a bunch of tasks
+		for i := 0; i < count; i++ {
+			t := Task{Type: fmt.Sprintf("task%d", i), Payload: Payload{"data": i}}
+			client.Process(&t, time.Now())
+		}
+		for i := 0; i < count; i++ {
+			t := Task{Type: fmt.Sprintf("scheduled%d", i), Payload: Payload{"data": i}}
+			client.Process(&t, time.Now().Add(time.Second))
+		}
+
+		var wg sync.WaitGroup
+		wg.Add(count * 2)
+		handler := func(t *Task) error {
+			// randomly fail 1% of tasks
+			if rand.Intn(100) == 1 {
+				return fmt.Errorf(":(")
+			}
+			wg.Done()
+			return nil
+		}
+		b.StartTimer() // end setup
+
+		bg.start(HandlerFunc(handler))
+		wg.Wait()
+
+		b.StopTimer() // begin teardown
+		bg.stop()
+		b.StartTimer() // end teardown
+	}
+}