fix: testing

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

Author: appleboy

queue_test.go

@@ -100,7 +100,7 @@ func TestWorkerPanic(t *testing.T) {
 	}
 	q, err := NewQueue(
 		WithWorker(w),
-		WithWorkerCount(2),
+		WithWorkerCount(5),
 	)
 	assert.NoError(t, err)
 	assert.NotNil(t, q)
@@ -116,7 +116,7 @@ func TestWorkerPanic(t *testing.T) {
 	}))
 	q.Start()
 	time.Sleep(100 * time.Millisecond)
-	assert.Equal(t, 2, q.Workers())
+	assert.Equal(t, 5, q.Workers())
 	q.Shutdown()
 	q.Wait()
 	assert.Equal(t, 0, q.Workers())

worker.go

@@ -8,12 +8,19 @@ import (
 
 // Worker interface
 type Worker interface {
+	// BeforeRun is called before starting the worker
 	BeforeRun() error
+	// Run is called to start the worker
 	Run(chan struct{}) error
+	// BeforeRun is called after starting the worker
 	AfterRun() error
+	// Shutdown is called if stop all worker
 	Shutdown() error
+	// Queue to send message in Queue (single channel, NSQ or AWS SQS)
 	Queue(job QueuedMessage) error
+	// Capacity queue capacity = cap(channel name)
 	Capacity() int
+	// Usage is how many message in queue
 	Usage() int
 }
 
@@ -49,7 +56,7 @@ func (w *queueWorker) Run(chan struct{}) error {
 		if string(msg.Bytes()) == "panic" {
 			panic("show panic")
 		}
-		time.Sleep(100 * time.Millisecond)
+		time.Sleep(20 * time.Millisecond)
 	}
 	return nil
 }