chore(benchmark): move testing to new file.

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

benchmark_test.go

@@ -0,0 +1,29 @@
+package queue
+
+import (
+	"context"
+	"testing"
+)
+
+func BenchmarkQueueTask(b *testing.B) {
+	b.ReportAllocs()
+	q := NewPool(5)
+	defer q.Release()
+	for n := 0; n < b.N; n++ {
+		_ = q.QueueTask(func(context.Context) error {
+			return nil
+		})
+	}
+}
+
+func BenchmarkQueue(b *testing.B) {
+	b.ReportAllocs()
+	m := &mockMessage{
+		message: "foo",
+	}
+	q := NewPool(5)
+	defer q.Release()
+	for n := 0; n < b.N; n++ {
+		_ = q.Queue(m)
+	}
+}

queue_test.go

@@ -1,7 +1,6 @@
 package queue
 
 import (
-	"context"
 	"testing"
 	"time"
 
@@ -141,26 +140,3 @@ func TestCloseQueueAfterShutdown(t *testing.T) {
 	assert.Error(t, err)
 	assert.Equal(t, ErrQueueShutdown, err)
 }
-
-func BenchmarkQueueTask(b *testing.B) {
-	b.ReportAllocs()
-	q := NewPool(5)
-	defer q.Release()
-	for n := 0; n < b.N; n++ {
-		_ = q.QueueTask(func(context.Context) error {
-			return nil
-		})
-	}
-}
-
-func BenchmarkQueue(b *testing.B) {
-	b.ReportAllocs()
-	m := &mockMessage{
-		message: "foo",
-	}
-	q := NewPool(5)
-	defer q.Release()
-	for n := 0; n < b.N; n++ {
-		_ = q.Queue(m)
-	}
-}