mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-25 15:22:18 +08:00
137 lines
3.5 KiB
Go
137 lines
3.5 KiB
Go
// Copyright 2020 Kentaro Hibino. All rights reserved.
|
|
// Use of this source code is governed by a MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package asynq
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
h "github.com/hibiken/asynq/internal/asynqtest"
|
|
"github.com/hibiken/asynq/internal/base"
|
|
)
|
|
|
|
func TestClient(t *testing.T) {
|
|
r := setup(t)
|
|
client := NewClient(r)
|
|
|
|
task := &Task{Type: "send_email", Payload: map[string]interface{}{"to": "customer@gmail.com", "from": "merchant@example.com"}}
|
|
|
|
tests := []struct {
|
|
desc string
|
|
task *Task
|
|
processAt time.Time
|
|
opts []Option
|
|
wantEnqueued []*base.TaskMessage
|
|
wantScheduled []h.ZSetEntry
|
|
}{
|
|
{
|
|
desc: "Process task immediately",
|
|
task: task,
|
|
processAt: time.Now(),
|
|
opts: []Option{},
|
|
wantEnqueued: []*base.TaskMessage{
|
|
&base.TaskMessage{
|
|
Type: task.Type,
|
|
Payload: task.Payload,
|
|
Retry: defaultMaxRetry,
|
|
Queue: "default",
|
|
},
|
|
},
|
|
wantScheduled: nil, // db is flushed in setup so zset does not exist hence nil
|
|
},
|
|
{
|
|
desc: "Schedule task to be processed in the future",
|
|
task: task,
|
|
processAt: time.Now().Add(2 * time.Hour),
|
|
opts: []Option{},
|
|
wantEnqueued: nil, // db is flushed in setup so list does not exist hence nil
|
|
wantScheduled: []h.ZSetEntry{
|
|
{
|
|
Msg: &base.TaskMessage{
|
|
Type: task.Type,
|
|
Payload: task.Payload,
|
|
Retry: defaultMaxRetry,
|
|
Queue: "default",
|
|
},
|
|
Score: time.Now().Add(2 * time.Hour).Unix(),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "Process task immediately with a custom retry count",
|
|
task: task,
|
|
processAt: time.Now(),
|
|
opts: []Option{
|
|
MaxRetry(3),
|
|
},
|
|
wantEnqueued: []*base.TaskMessage{
|
|
&base.TaskMessage{
|
|
Type: task.Type,
|
|
Payload: task.Payload,
|
|
Retry: 3,
|
|
Queue: "default",
|
|
},
|
|
},
|
|
wantScheduled: nil, // db is flushed in setup so zset does not exist hence nil
|
|
},
|
|
{
|
|
desc: "Negative retry count",
|
|
task: task,
|
|
processAt: time.Now(),
|
|
opts: []Option{
|
|
MaxRetry(-2),
|
|
},
|
|
wantEnqueued: []*base.TaskMessage{
|
|
&base.TaskMessage{
|
|
Type: task.Type,
|
|
Payload: task.Payload,
|
|
Retry: 0, // Retry count should be set to zero
|
|
Queue: "default",
|
|
},
|
|
},
|
|
wantScheduled: nil, // db is flushed in setup so zset does not exist hence nil
|
|
},
|
|
{
|
|
desc: "Conflicting options",
|
|
task: task,
|
|
processAt: time.Now(),
|
|
opts: []Option{
|
|
MaxRetry(2),
|
|
MaxRetry(10),
|
|
},
|
|
wantEnqueued: []*base.TaskMessage{
|
|
&base.TaskMessage{
|
|
Type: task.Type,
|
|
Payload: task.Payload,
|
|
Retry: 10, // Last option takes precedence
|
|
Queue: "default",
|
|
},
|
|
},
|
|
wantScheduled: nil, // db is flushed in setup so zset does not exist hence nil
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
h.FlushDB(t, r) // clean up db before each test case.
|
|
|
|
err := client.Process(tc.task, tc.processAt, tc.opts...)
|
|
if err != nil {
|
|
t.Error(err)
|
|
continue
|
|
}
|
|
|
|
gotEnqueued := h.GetEnqueuedMessages(t, r)
|
|
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.IgnoreIDOpt); diff != "" {
|
|
t.Errorf("%s;\nmismatch found in %q; (-want,+got)\n%s", tc.desc, base.DefaultQueue, diff)
|
|
}
|
|
|
|
gotScheduled := h.GetScheduledEntries(t, r)
|
|
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.IgnoreIDOpt); diff != "" {
|
|
t.Errorf("%s;\nmismatch found in %q; (-want,+got)\n%s", tc.desc, base.ScheduledQueue, diff)
|
|
}
|
|
}
|
|
}
|