2
0
mirror of https://github.com/hibiken/asynq.git synced 2024-11-15 20:08:46 +08:00
asynq/inspector_test.go

2223 lines
59 KiB
Go
Raw Permalink Normal View History

2020-07-13 21:29:41 +08:00
// 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 (
"fmt"
2020-07-27 04:16:13 +08:00
"math"
"sort"
2020-07-13 21:29:41 +08:00
"testing"
"time"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"github.com/hibiken/asynq/internal/asynqtest"
h "github.com/hibiken/asynq/internal/asynqtest"
"github.com/hibiken/asynq/internal/base"
"github.com/hibiken/asynq/internal/rdb"
2020-07-13 21:29:41 +08:00
)
2020-08-21 12:17:44 +08:00
func TestInspectorQueues(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
inspector := NewInspector(getRedisConnOpt(t))
2020-08-21 12:17:44 +08:00
tests := []struct {
queues []string
}{
{queues: []string{"default"}},
{queues: []string{"custom1", "custom2"}},
{queues: []string{"default", "custom1", "custom2"}},
{queues: []string{}},
}
for _, tc := range tests {
h.FlushDB(t, r)
for _, qname := range tc.queues {
if err := r.SAdd(base.AllQueues, qname).Err(); err != nil {
t.Fatalf("could not initialize all queue set: %v", err)
}
}
got, err := inspector.Queues()
if err != nil {
t.Errorf("Queues() returned an error: %v", err)
continue
}
if diff := cmp.Diff(tc.queues, got, h.SortStringSliceOpt); diff != "" {
t.Errorf("Queues() = %v, want %v; (-want, +got)\n%s", got, tc.queues, diff)
}
}
}
func TestInspectorDeleteQueue(t *testing.T) {
r := setup(t)
defer r.Close()
inspector := NewInspector(getRedisConnOpt(t))
defer inspector.Close()
m1 := h.NewTaskMessage("task1", nil)
m2 := h.NewTaskMessage("task2", nil)
m3 := h.NewTaskMessageWithQueue("task3", nil, "custom")
m4 := h.NewTaskMessageWithQueue("task4", nil, "custom")
tests := []struct {
pending map[string][]*base.TaskMessage
active map[string][]*base.TaskMessage
scheduled map[string][]base.Z
retry map[string][]base.Z
archived map[string][]base.Z
qname string // queue to remove
force bool
}{
{
pending: map[string][]*base.TaskMessage{
"default": {m1, m2},
"custom": {},
},
active: map[string][]*base.TaskMessage{
"default": {},
"custom": {},
},
scheduled: map[string][]base.Z{
"default": {},
"custom": {},
},
retry: map[string][]base.Z{
"default": {},
"custom": {},
},
archived: map[string][]base.Z{
"default": {},
"custom": {},
},
qname: "custom",
force: false,
},
{
pending: map[string][]*base.TaskMessage{
"default": {m1, m2},
"custom": {m3},
},
active: map[string][]*base.TaskMessage{
"default": {},
"custom": {},
},
scheduled: map[string][]base.Z{
"default": {},
"custom": {{Message: m4, Score: time.Now().Unix()}},
},
retry: map[string][]base.Z{
"default": {},
"custom": {},
},
archived: map[string][]base.Z{
"default": {},
"custom": {},
},
qname: "custom",
force: true, // allow removing non-empty queue
},
}
for _, tc := range tests {
h.FlushDB(t, r)
h.SeedAllPendingQueues(t, r, tc.pending)
h.SeedAllActiveQueues(t, r, tc.active)
h.SeedAllScheduledQueues(t, r, tc.scheduled)
h.SeedAllRetryQueues(t, r, tc.retry)
h.SeedAllArchivedQueues(t, r, tc.archived)
err := inspector.DeleteQueue(tc.qname, tc.force)
if err != nil {
t.Errorf("DeleteQueue(%q, %t) = %v, want nil",
tc.qname, tc.force, err)
continue
}
if r.SIsMember(base.AllQueues, tc.qname).Val() {
t.Errorf("%q is a member of %q", tc.qname, base.AllQueues)
}
}
}
func TestInspectorDeleteQueueErrorQueueNotEmpty(t *testing.T) {
r := setup(t)
defer r.Close()
inspector := NewInspector(getRedisConnOpt(t))
defer inspector.Close()
m1 := h.NewTaskMessage("task1", nil)
m2 := h.NewTaskMessage("task2", nil)
m3 := h.NewTaskMessageWithQueue("task3", nil, "custom")
m4 := h.NewTaskMessageWithQueue("task4", nil, "custom")
tests := []struct {
pending map[string][]*base.TaskMessage
active map[string][]*base.TaskMessage
scheduled map[string][]base.Z
retry map[string][]base.Z
archived map[string][]base.Z
qname string // queue to remove
force bool
}{
{
pending: map[string][]*base.TaskMessage{
"default": {m1, m2},
},
active: map[string][]*base.TaskMessage{
"default": {m3, m4},
},
scheduled: map[string][]base.Z{
"default": {},
},
retry: map[string][]base.Z{
"default": {},
},
archived: map[string][]base.Z{
"default": {},
},
qname: "default",
force: false,
},
}
for _, tc := range tests {
h.FlushDB(t, r)
h.SeedAllPendingQueues(t, r, tc.pending)
h.SeedAllActiveQueues(t, r, tc.active)
h.SeedAllScheduledQueues(t, r, tc.scheduled)
h.SeedAllRetryQueues(t, r, tc.retry)
h.SeedAllArchivedQueues(t, r, tc.archived)
err := inspector.DeleteQueue(tc.qname, tc.force)
if _, ok := err.(*ErrQueueNotEmpty); !ok {
t.Errorf("DeleteQueue(%v, %t) did not return ErrQueueNotEmpty",
tc.qname, tc.force)
}
}
}
func TestInspectorDeleteQueueErrorQueueNotFound(t *testing.T) {
r := setup(t)
defer r.Close()
inspector := NewInspector(getRedisConnOpt(t))
defer inspector.Close()
m1 := h.NewTaskMessage("task1", nil)
m2 := h.NewTaskMessage("task2", nil)
m3 := h.NewTaskMessageWithQueue("task3", nil, "custom")
m4 := h.NewTaskMessageWithQueue("task4", nil, "custom")
tests := []struct {
pending map[string][]*base.TaskMessage
active map[string][]*base.TaskMessage
scheduled map[string][]base.Z
retry map[string][]base.Z
archived map[string][]base.Z
qname string // queue to remove
force bool
}{
{
pending: map[string][]*base.TaskMessage{
"default": {m1, m2},
},
active: map[string][]*base.TaskMessage{
"default": {m3, m4},
},
scheduled: map[string][]base.Z{
"default": {},
},
retry: map[string][]base.Z{
"default": {},
},
archived: map[string][]base.Z{
"default": {},
},
qname: "nonexistent",
force: false,
},
}
for _, tc := range tests {
h.FlushDB(t, r)
h.SeedAllPendingQueues(t, r, tc.pending)
h.SeedAllActiveQueues(t, r, tc.active)
h.SeedAllScheduledQueues(t, r, tc.scheduled)
h.SeedAllRetryQueues(t, r, tc.retry)
h.SeedAllArchivedQueues(t, r, tc.archived)
err := inspector.DeleteQueue(tc.qname, tc.force)
if _, ok := err.(*ErrQueueNotFound); !ok {
t.Errorf("DeleteQueue(%v, %t) did not return ErrQueueNotFound",
tc.qname, tc.force)
}
}
}
2020-07-13 21:29:41 +08:00
func TestInspectorCurrentStats(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
m4 := asynqtest.NewTaskMessage("task4", nil)
m5 := asynqtest.NewTaskMessageWithQueue("task5", nil, "critical")
m6 := h.NewTaskMessageWithQueue("task6", nil, "low")
now := time.Now()
timeCmpOpt := cmpopts.EquateApproxTime(time.Second)
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
pending map[string][]*base.TaskMessage
active map[string][]*base.TaskMessage
scheduled map[string][]base.Z
retry map[string][]base.Z
archived map[string][]base.Z
processed map[string]int
failed map[string]int
qname string
want *QueueStats
2020-07-13 21:29:41 +08:00
}{
{
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m1},
"critical": {m5},
"low": {m6},
},
active: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m2},
"critical": {},
"low": {},
},
scheduled: map[string][]base.Z{
"default": {
{Message: m3, Score: now.Add(time.Hour).Unix()},
{Message: m4, Score: now.Unix()},
},
"critical": {},
"low": {},
},
2020-08-18 21:01:38 +08:00
retry: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"critical": {},
"low": {},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"critical": {},
"low": {},
},
processed: map[string]int{
"default": 120,
"critical": 100,
"low": 42,
},
failed: map[string]int{
"default": 2,
"critical": 0,
"low": 5,
},
qname: "default",
want: &QueueStats{
2020-09-06 03:43:15 +08:00
Queue: "default",
Size: 4,
Pending: 1,
Active: 1,
Scheduled: 2,
Retry: 0,
Archived: 0,
2020-09-06 03:43:15 +08:00
Processed: 120,
Failed: 2,
Paused: false,
Timestamp: now,
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
asynqtest.SeedAllActiveQueues(t, r, tc.active)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-08-17 05:51:56 +08:00
for qname, n := range tc.processed {
processedKey := base.ProcessedKey(qname, now)
2020-08-18 21:01:38 +08:00
r.Set(processedKey, n, 0)
2020-08-17 05:51:56 +08:00
}
for qname, n := range tc.failed {
failedKey := base.FailedKey(qname, now)
2020-08-18 21:01:38 +08:00
r.Set(failedKey, n, 0)
2020-08-17 05:51:56 +08:00
}
got, err := inspector.CurrentStats(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("r.CurrentStats(%q) = %v, %v, want %v, nil",
tc.qname, got, err, tc.want)
2020-07-13 21:29:41 +08:00
continue
}
if diff := cmp.Diff(tc.want, got, timeCmpOpt); diff != "" {
2020-08-17 05:51:56 +08:00
t.Errorf("r.CurrentStats(%q) = %v, %v, want %v, nil; (-want, +got)\n%s",
tc.qname, got, err, tc.want, diff)
2020-07-13 21:29:41 +08:00
continue
}
}
}
func TestInspectorHistory(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
now := time.Now().UTC()
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
qname string // queue of interest
n int // number of days
2020-07-13 21:29:41 +08:00
}{
2020-08-17 05:51:56 +08:00
{"default", 90},
{"custom", 7},
{"default", 1},
2020-07-13 21:29:41 +08:00
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
r.SAdd(base.AllQueues, tc.qname)
2020-07-13 21:29:41 +08:00
// populate last n days data
for i := 0; i < tc.n; i++ {
ts := now.Add(-time.Duration(i) * 24 * time.Hour)
2020-08-17 05:51:56 +08:00
processedKey := base.ProcessedKey(tc.qname, ts)
failedKey := base.FailedKey(tc.qname, ts)
2020-07-13 21:29:41 +08:00
r.Set(processedKey, (i+1)*1000, 0)
r.Set(failedKey, (i+1)*10, 0)
}
2020-08-17 05:51:56 +08:00
got, err := inspector.History(tc.qname, tc.n)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("Inspector.History(%q, %d) returned error: %v", tc.qname, tc.n, err)
2020-07-13 21:29:41 +08:00
continue
}
if len(got) != tc.n {
2020-08-17 05:51:56 +08:00
t.Errorf("Inspector.History(%q, %d) returned %d daily stats, want %d",
tc.qname, tc.n, len(got), tc.n)
2020-07-13 21:29:41 +08:00
continue
}
for i := 0; i < tc.n; i++ {
want := &DailyStats{
2020-08-17 05:51:56 +08:00
Queue: tc.qname,
2020-07-13 21:29:41 +08:00
Processed: (i + 1) * 1000,
Failed: (i + 1) * 10,
Date: now.Add(-time.Duration(i) * 24 * time.Hour),
}
2020-09-10 10:17:56 +08:00
// Allow 2 seconds difference in timestamp.
timeCmpOpt := cmpopts.EquateApproxTime(2 * time.Second)
2020-07-13 21:29:41 +08:00
if diff := cmp.Diff(want, got[i], timeCmpOpt); diff != "" {
t.Errorf("Inspector.History %d days ago data; got %+v, want %+v; (-want,+got):\n%s",
i, got[i], want, diff)
}
}
}
}
2020-09-05 22:03:43 +08:00
func createPendingTask(msg *base.TaskMessage) *PendingTask {
return &PendingTask{
2020-07-13 21:29:41 +08:00
Task: NewTask(msg.Type, msg.Payload),
ID: msg.ID.String(),
Queue: msg.Queue,
}
}
2020-09-05 22:03:43 +08:00
func TestInspectorListPendingTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
m4 := asynqtest.NewTaskMessage("task4", nil)
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-09-05 22:03:43 +08:00
desc string
pending map[string][]*base.TaskMessage
qname string
want []*PendingTask
2020-07-13 21:29:41 +08:00
}{
{
desc: "with default queue",
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m2},
},
qname: "default",
2020-09-05 22:03:43 +08:00
want: []*PendingTask{
createPendingTask(m1),
createPendingTask(m2),
2020-07-13 21:29:41 +08:00
},
},
{
desc: "with named queue",
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m2},
"critical": {m3},
"low": {m4},
},
qname: "critical",
2020-09-05 22:03:43 +08:00
want: []*PendingTask{
createPendingTask(m3),
2020-07-13 21:29:41 +08:00
},
},
{
desc: "with empty queue",
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
},
qname: "default",
2020-09-05 22:03:43 +08:00
want: []*PendingTask(nil),
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-09-05 22:03:43 +08:00
for q, msgs := range tc.pending {
asynqtest.SeedPendingQueue(t, r, msgs, q)
2020-07-13 21:29:41 +08:00
}
2020-09-05 22:03:43 +08:00
got, err := inspector.ListPendingTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-09-05 22:03:43 +08:00
t.Errorf("%s; ListPendingTasks(%q) returned error: %v",
2020-07-13 21:29:41 +08:00
tc.desc, tc.qname, err)
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{})
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
2020-09-05 22:03:43 +08:00
t.Errorf("%s; ListPendingTasks(%q) = %v, want %v; (-want,+got)\n%s",
2020-07-13 21:29:41 +08:00
tc.desc, tc.qname, got, tc.want, diff)
}
}
}
2020-09-06 03:43:15 +08:00
func TestInspectorListActiveTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
2020-08-17 05:51:56 +08:00
m3 := asynqtest.NewTaskMessage("task3", nil)
m4 := asynqtest.NewTaskMessage("task4", nil)
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
2020-09-06 03:43:15 +08:00
createActiveTask := func(msg *base.TaskMessage) *ActiveTask {
return &ActiveTask{
2020-08-17 05:51:56 +08:00
Task: NewTask(msg.Type, msg.Payload),
ID: msg.ID.String(),
Queue: msg.Queue,
2020-07-13 21:29:41 +08:00
}
}
tests := []struct {
desc string
active map[string][]*base.TaskMessage
qname string
want []*ActiveTask
2020-07-13 21:29:41 +08:00
}{
{
2020-09-06 03:43:15 +08:00
desc: "with a few active tasks",
active: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m1, m2},
"custom": {m3, m4},
},
qname: "default",
2020-09-06 03:43:15 +08:00
want: []*ActiveTask{
createActiveTask(m1),
createActiveTask(m2),
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllActiveQueues(t, r, tc.active)
2020-07-13 21:29:41 +08:00
2020-09-06 03:43:15 +08:00
got, err := inspector.ListActiveTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-09-06 03:43:15 +08:00
t.Errorf("%s; ListActiveTasks(%q) returned error: %v", tc.qname, tc.desc, err)
2020-07-13 21:29:41 +08:00
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{})
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
2020-09-06 03:43:15 +08:00
t.Errorf("%s; ListActiveTask(%q) = %v, want %v; (-want,+got)\n%s",
2020-08-17 05:51:56 +08:00
tc.desc, tc.qname, got, tc.want, diff)
2020-07-13 21:29:41 +08:00
}
}
}
2020-08-17 05:51:56 +08:00
func createScheduledTask(z base.Z) *ScheduledTask {
msg := z.Message
return &ScheduledTask{
Task: NewTask(msg.Type, msg.Payload),
ID: msg.ID.String(),
Queue: msg.Queue,
2020-09-05 22:11:55 +08:00
NextProcessAt: time.Unix(z.Score, 0),
2020-08-19 20:45:02 +08:00
score: z.Score,
2020-08-17 05:51:56 +08:00
}
}
2020-07-13 21:29:41 +08:00
func TestInspectorListScheduledTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-18 21:01:38 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
desc string
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
qname string
2020-07-13 21:29:41 +08:00
want []*ScheduledTask
}{
{
2020-08-17 05:51:56 +08:00
desc: "with a few scheduled tasks",
scheduled: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
2020-09-05 22:11:55 +08:00
// Should be sorted by NextProcessAt.
2020-07-13 21:29:41 +08:00
want: []*ScheduledTask{
createScheduledTask(z3),
createScheduledTask(z1),
createScheduledTask(z2),
},
},
{
2020-08-17 05:51:56 +08:00
desc: "with empty scheduled queue",
scheduled: map[string][]base.Z{
"default": {},
},
qname: "default",
want: []*ScheduledTask(nil),
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
2020-07-13 21:29:41 +08:00
2020-08-18 21:01:38 +08:00
got, err := inspector.ListScheduledTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("%s; ListScheduledTasks(%q) returned error: %v", tc.desc, tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{}, ScheduledTask{})
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
2020-08-17 05:51:56 +08:00
t.Errorf("%s; ListScheduledTask(%q) = %v, want %v; (-want,+got)\n%s",
tc.desc, tc.qname, got, tc.want, diff)
2020-07-13 21:29:41 +08:00
}
}
}
2020-08-17 05:51:56 +08:00
func createRetryTask(z base.Z) *RetryTask {
msg := z.Message
return &RetryTask{
Task: NewTask(msg.Type, msg.Payload),
ID: msg.ID.String(),
Queue: msg.Queue,
2020-09-05 22:11:55 +08:00
NextProcessAt: time.Unix(z.Score, 0),
2020-08-17 05:51:56 +08:00
MaxRetry: msg.Retry,
Retried: msg.Retried,
ErrorMsg: msg.ErrorMsg,
2020-08-19 20:45:02 +08:00
score: z.Score,
2020-08-17 05:51:56 +08:00
}
}
2020-07-13 21:29:41 +08:00
func TestInspectorListRetryTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
desc string
2020-08-17 05:51:56 +08:00
retry map[string][]base.Z
qname string
2020-07-13 21:29:41 +08:00
want []*RetryTask
}{
{
2020-08-17 05:51:56 +08:00
desc: "with a few retry tasks",
retry: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
2020-09-05 22:11:55 +08:00
// Should be sorted by NextProcessAt.
2020-07-13 21:29:41 +08:00
want: []*RetryTask{
createRetryTask(z3),
createRetryTask(z1),
createRetryTask(z2),
},
},
{
2020-08-17 05:51:56 +08:00
desc: "with empty retry queue",
retry: map[string][]base.Z{
"default": {},
},
qname: "default",
2020-07-13 21:29:41 +08:00
want: []*RetryTask(nil),
},
2020-08-17 05:51:56 +08:00
// TODO(hibiken): ErrQueueNotFound when queue doesn't exist
2020-07-13 21:29:41 +08:00
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
2020-07-13 21:29:41 +08:00
2020-08-18 21:01:38 +08:00
got, err := inspector.ListRetryTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("%s; ListRetryTasks(%q) returned error: %v", tc.desc, tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{}, RetryTask{})
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
2020-08-17 05:51:56 +08:00
t.Errorf("%s; ListRetryTask(%q) = %v, want %v; (-want,+got)\n%s",
tc.desc, tc.qname, got, tc.want, diff)
2020-07-13 21:29:41 +08:00
}
}
}
func createArchivedTask(z base.Z) *ArchivedTask {
2020-08-17 05:51:56 +08:00
msg := z.Message
return &ArchivedTask{
2020-08-17 05:51:56 +08:00
Task: NewTask(msg.Type, msg.Payload),
ID: msg.ID.String(),
Queue: msg.Queue,
MaxRetry: msg.Retry,
Retried: msg.Retried,
LastFailedAt: time.Unix(z.Score, 0),
ErrorMsg: msg.ErrorMsg,
2020-08-19 20:45:02 +08:00
score: z.Score,
2020-08-17 05:51:56 +08:00
}
}
func TestInspectorListArchivedTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(-5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(-15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(-2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(-2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
desc string
archived map[string][]base.Z
qname string
want []*ArchivedTask
2020-07-13 21:29:41 +08:00
}{
{
desc: "with a few archived tasks",
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
2020-07-13 21:29:41 +08:00
// Should be sorted by LastFailedAt.
want: []*ArchivedTask{
createArchivedTask(z2),
createArchivedTask(z1),
createArchivedTask(z3),
2020-07-13 21:29:41 +08:00
},
},
{
desc: "with empty archived queue",
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
},
qname: "default",
want: []*ArchivedTask(nil),
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
got, err := inspector.ListArchivedTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
t.Errorf("%s; ListArchivedTasks(%q) returned error: %v", tc.desc, tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{}, ArchivedTask{})
2020-07-13 21:29:41 +08:00
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
t.Errorf("%s; ListArchivedTask(%q) = %v, want %v; (-want,+got)\n%s",
2020-08-17 05:51:56 +08:00
tc.desc, tc.qname, got, tc.want, diff)
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorListPagination(t *testing.T) {
// Create 100 tasks.
var msgs []*base.TaskMessage
for i := 0; i <= 99; i++ {
msgs = append(msgs,
asynqtest.NewTaskMessage(fmt.Sprintf("task%d", i), nil))
}
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-09-05 22:03:43 +08:00
asynqtest.SeedPendingQueue(t, r, msgs, base.DefaultQueueName)
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
page int
pageSize int
2020-09-05 22:03:43 +08:00
want []*PendingTask
2020-07-13 21:29:41 +08:00
}{
{
page: 1,
pageSize: 5,
2020-09-05 22:03:43 +08:00
want: []*PendingTask{
createPendingTask(msgs[0]),
createPendingTask(msgs[1]),
createPendingTask(msgs[2]),
createPendingTask(msgs[3]),
createPendingTask(msgs[4]),
2020-07-13 21:29:41 +08:00
},
},
{
page: 3,
pageSize: 10,
2020-09-05 22:03:43 +08:00
want: []*PendingTask{
createPendingTask(msgs[20]),
createPendingTask(msgs[21]),
createPendingTask(msgs[22]),
createPendingTask(msgs[23]),
createPendingTask(msgs[24]),
createPendingTask(msgs[25]),
createPendingTask(msgs[26]),
createPendingTask(msgs[27]),
createPendingTask(msgs[28]),
createPendingTask(msgs[29]),
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
2020-09-05 22:03:43 +08:00
got, err := inspector.ListPendingTasks("default", Page(tc.page), PageSize(tc.pageSize))
2020-07-13 21:29:41 +08:00
if err != nil {
2020-09-05 22:03:43 +08:00
t.Errorf("ListPendingTask('default') returned error: %v", err)
2020-07-13 21:29:41 +08:00
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{})
if diff := cmp.Diff(tc.want, got, ignoreOpt); diff != "" {
2020-09-05 22:03:43 +08:00
t.Errorf("ListPendingTask('default') = %v, want %v; (-want,+got)\n%s",
2020-07-13 21:29:41 +08:00
got, tc.want, diff)
}
}
}
func TestInspectorDeleteAllScheduledTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-18 21:46:19 +08:00
scheduled map[string][]base.Z
qname string
want int
wantScheduled map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
want: 3,
2020-08-18 21:46:19 +08:00
wantScheduled: map[string][]base.Z{
"default": {},
"custom": {z4},
},
2020-07-13 21:29:41 +08:00
},
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {},
},
qname: "default",
want: 0,
2020-08-18 21:46:19 +08:00
wantScheduled: map[string][]base.Z{
"default": {},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
2020-07-13 21:29:41 +08:00
2020-08-17 05:51:56 +08:00
got, err := inspector.DeleteAllScheduledTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("DeleteAllScheduledTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
2020-08-17 05:51:56 +08:00
t.Errorf("DeleteAllScheduledTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-18 21:46:19 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s", qname, diff)
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorDeleteAllRetryTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-18 21:46:19 +08:00
retry map[string][]base.Z
qname string
want int
wantRetry map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
2020-07-13 21:29:41 +08:00
want: 3,
2020-08-18 21:46:19 +08:00
wantRetry: map[string][]base.Z{
"default": {},
"custom": {z4},
},
2020-07-13 21:29:41 +08:00
},
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {},
},
qname: "default",
2020-07-13 21:29:41 +08:00
want: 0,
2020-08-18 21:46:19 +08:00
wantRetry: map[string][]base.Z{
"default": {},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
2020-07-13 21:29:41 +08:00
2020-08-17 05:51:56 +08:00
got, err := inspector.DeleteAllRetryTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("DeleteAllRetryTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
2020-08-17 05:51:56 +08:00
t.Errorf("DeleteAllRetryTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-18 21:46:19 +08:00
for qname, want := range tc.wantRetry {
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s", qname, diff)
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorDeleteAllArchivedTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
archived map[string][]base.Z
qname string
want int
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z2, z3},
"custom": {z4},
},
qname: "default",
want: 3,
wantArchived: map[string][]base.Z{
2020-08-18 21:46:19 +08:00
"default": {},
"custom": {z4},
},
2020-07-13 21:29:41 +08:00
},
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
},
qname: "default",
want: 0,
wantArchived: map[string][]base.Z{
2020-08-18 21:46:19 +08:00
"default": {},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
got, err := inspector.DeleteAllArchivedTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
t.Errorf("DeleteAllArchivedTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
t.Errorf("DeleteAllArchivedTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
for qname, want := range tc.wantArchived {
gotArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, gotArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-18 21:46:19 +08:00
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorKillAllScheduledTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
archived map[string][]base.Z
2020-08-17 05:51:56 +08:00
qname string
want int
wantScheduled map[string][]base.Z
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "default",
want: 3,
wantScheduled: map[string][]base.Z{
"default": {},
"custom": {z4},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {
base.Z{Message: m1, Score: now.Unix()},
base.Z{Message: m2, Score: now.Unix()},
base.Z{Message: m3, Score: now.Unix()},
},
"custom": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-18 21:01:38 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z2},
},
archived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {z3},
},
qname: "default",
want: 2,
wantScheduled: map[string][]base.Z{
"default": {},
},
wantArchived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {
z3,
base.Z{Message: m1, Score: now.Unix()},
base.Z{Message: m2, Score: now.Unix()},
},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-18 21:01:38 +08:00
scheduled: map[string][]base.Z{
"default": {},
},
archived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {},
},
qname: "default",
want: 0,
wantScheduled: map[string][]base.Z{
"default": {},
},
wantArchived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {},
},
2020-07-13 21:29:41 +08:00
},
{
2020-08-18 21:01:38 +08:00
scheduled: map[string][]base.Z{
"default": {},
},
archived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {z1, z2},
},
qname: "default",
want: 0,
wantScheduled: map[string][]base.Z{
"default": {},
},
wantArchived: map[string][]base.Z{
2020-08-18 21:01:38 +08:00
"default": {z1, z2},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
got, err := inspector.ArchiveAllScheduledTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("KillAllScheduledTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
2020-08-17 05:51:56 +08:00
t.Errorf("KillAllScheduledTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
for qname, want := range tc.wantArchived {
2020-08-17 05:51:56 +08:00
// Allow Z.Score to differ by up to 2.
approxOpt := cmp.Comparer(func(a, b int64) bool {
return math.Abs(float64(a-b)) < 2
})
gotArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, gotArchived, asynqtest.SortZSetEntryOpt, approxOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorKillAllRetryTasks(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessage("task3", nil)
2020-08-17 05:51:56 +08:00
m4 := asynqtest.NewTaskMessageWithQueue("task4", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
retry map[string][]base.Z
archived map[string][]base.Z
qname string
want int
wantRetry map[string][]base.Z
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1, z2, z3},
"custom": {z4},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "default",
2020-07-13 21:29:41 +08:00
want: 3,
2020-08-17 05:51:56 +08:00
wantRetry: map[string][]base.Z{
"default": {},
"custom": {z4},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {
base.Z{Message: m1, Score: now.Unix()},
base.Z{Message: m2, Score: now.Unix()},
base.Z{Message: m3, Score: now.Unix()},
},
"custom": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1, z2},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z3},
},
qname: "default",
2020-07-13 21:29:41 +08:00
want: 2,
2020-08-17 05:51:56 +08:00
wantRetry: map[string][]base.Z{
"default": {},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {
z3,
base.Z{Message: m1, Score: now.Unix()},
base.Z{Message: m2, Score: now.Unix()},
},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z2},
},
qname: "default",
want: 0,
2020-08-17 05:51:56 +08:00
wantRetry: map[string][]base.Z{
"default": {},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z2},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
got, err := inspector.ArchiveAllRetryTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
2020-08-17 05:51:56 +08:00
t.Errorf("KillAllRetryTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
2020-08-17 05:51:56 +08:00
t.Errorf("KillAllRetryTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantRetry {
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
cmpOpt := asynqtest.EquateInt64Approx(2) // allow for 2 seconds difference in Z.Score
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt, cmpOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorRunAllScheduledTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "critical")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "low")
m4 := asynqtest.NewTaskMessage("task4", nil)
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
2020-09-05 22:03:43 +08:00
pending map[string][]*base.TaskMessage
2020-08-17 05:51:56 +08:00
qname string
want int
wantScheduled map[string][]base.Z
2020-09-05 22:03:43 +08:00
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z4},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 2,
wantScheduled: map[string][]base.Z{
"default": {},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m1, m4},
"critical": {},
"low": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 1,
wantScheduled: map[string][]base.Z{
"default": {},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4, m1},
2020-08-17 05:51:56 +08:00
"critical": {},
"low": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 0,
wantScheduled: map[string][]base.Z{
"default": {},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
got, err := inspector.RunAllScheduledTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
t.Errorf("RunAllScheduledTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
t.Errorf("RunAllScheduledTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorRunAllRetryTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "critical")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "low")
m4 := asynqtest.NewTaskMessage("task2", nil)
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-09-05 22:03:43 +08:00
retry map[string][]base.Z
pending map[string][]*base.TaskMessage
qname string
want int
wantRetry map[string][]base.Z
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1, z4},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 2,
wantRetry: map[string][]base.Z{
"default": {},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m1, m4},
"critical": {},
"low": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 1,
wantRetry: map[string][]base.Z{
"default": {},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4, m1},
2020-08-19 20:45:02 +08:00
"critical": {},
"low": {},
2020-07-13 21:29:41 +08:00
},
},
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 0,
wantRetry: map[string][]base.Z{
"default": {},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
got, err := inspector.RunAllRetryTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
t.Errorf("RunAllRetryTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
t.Errorf("RunAllRetryTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantRetry {
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
2020-08-17 05:51:56 +08:00
func TestInspectorRunAllArchivedTasks(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "critical")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "low")
m4 := asynqtest.NewTaskMessage("task2", nil)
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(-5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(-15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(-2 * time.Minute).Unix()}
2020-08-17 05:51:56 +08:00
z4 := base.Z{Message: m4, Score: now.Add(-2 * time.Minute).Unix()}
2020-07-13 21:29:41 +08:00
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
archived map[string][]base.Z
pending map[string][]*base.TaskMessage
qname string
want int
wantArchived map[string][]base.Z
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z4},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 2,
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m1, m4},
"critical": {},
"low": {},
2020-07-13 21:29:41 +08:00
},
},
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1},
"critical": {z2},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4},
"critical": {},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 1,
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"critical": {z2},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m4, m1},
2020-08-17 05:51:56 +08:00
"critical": {},
2020-07-13 21:29:41 +08:00
},
},
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
2020-08-17 05:51:56 +08:00
qname: "default",
want: 0,
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {m1, m4},
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
got, err := inspector.RunAllArchivedTasks(tc.qname)
2020-07-13 21:29:41 +08:00
if err != nil {
t.Errorf("RunAllArchivedTasks(%q) returned error: %v", tc.qname, err)
2020-07-13 21:29:41 +08:00
continue
}
if got != tc.want {
t.Errorf("RunAllArchivedTasks(%q) = %d, want %d", tc.qname, got, tc.want)
2020-07-13 21:29:41 +08:00
}
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorDeleteTaskByKeyDeletesScheduledTask(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
2020-08-17 05:51:56 +08:00
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
qname string
key string
wantScheduled map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z2},
"custom": {z3},
},
qname: "default",
key: createScheduledTask(z2).Key(),
wantScheduled: map[string][]base.Z{
"default": {z1},
"custom": {z3},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
2020-07-13 21:29:41 +08:00
2020-08-17 05:51:56 +08:00
if err := inspector.DeleteTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("DeleteTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorDeleteTaskByKeyDeletesRetryTask(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
2020-08-17 05:51:56 +08:00
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
retry map[string][]base.Z
qname string
key string
wantRetry map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1, z2},
"custom": {z3},
},
qname: "default",
key: createRetryTask(z2).Key(),
wantRetry: map[string][]base.Z{
"default": {z1},
"custom": {z3},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
2020-07-13 21:29:41 +08:00
2020-08-17 05:51:56 +08:00
if err := inspector.DeleteTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("DeleteTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantRetry {
2020-08-18 21:46:19 +08:00
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
2020-08-17 05:51:56 +08:00
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
2020-08-18 21:46:19 +08:00
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorDeleteTaskByKeyDeletesArchivedTask(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessage("task2", nil)
2020-08-17 05:51:56 +08:00
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(-5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(-15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(-2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
archived map[string][]base.Z
qname string
key string
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1, z2},
"custom": {z3},
},
qname: "default",
key: createArchivedTask(z2).Key(),
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1},
"custom": {z3},
},
2020-07-13 21:29:41 +08:00
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
2020-08-17 05:51:56 +08:00
if err := inspector.DeleteTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("DeleteTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s", qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorRunTaskByKeyRunsScheduledTask(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
2020-08-17 05:51:56 +08:00
m2 := asynqtest.NewTaskMessage("task2", nil)
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
2020-09-05 22:03:43 +08:00
pending map[string][]*base.TaskMessage
2020-08-17 05:51:56 +08:00
qname string
key string
wantScheduled map[string][]base.Z
2020-09-05 22:03:43 +08:00
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1, z2},
"custom": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "default",
key: createScheduledTask(z2).Key(),
wantScheduled: map[string][]base.Z{
"default": {z1},
"custom": {z3},
2020-07-13 21:29:41 +08:00
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {m2},
"custom": {},
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
if err := inspector.RunTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("RunTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
2020-07-13 21:29:41 +08:00
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorRunTaskByKeyRunsRetryTask(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
2020-08-17 05:51:56 +08:00
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "custom")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-09-05 22:03:43 +08:00
retry map[string][]base.Z
pending map[string][]*base.TaskMessage
qname string
key string
wantRetry map[string][]base.Z
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1},
"custom": {z2, z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "custom",
key: createRetryTask(z2).Key(),
wantRetry: map[string][]base.Z{
"default": {z1},
"custom": {z3},
2020-07-13 21:29:41 +08:00
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {m2},
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
if err := inspector.RunTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("RunTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantRetry {
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorRunTaskByKeyRunsArchivedTask(t *testing.T) {
2020-07-13 21:29:41 +08:00
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "critical")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "low")
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(-5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(-15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(-2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
archived map[string][]base.Z
pending map[string][]*base.TaskMessage
qname string
key string
wantArchived map[string][]base.Z
wantPending map[string][]*base.TaskMessage
2020-07-13 21:29:41 +08:00
}{
{
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1},
"critical": {z2},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
pending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
"critical": {},
"low": {},
},
2020-08-17 05:51:56 +08:00
qname: "critical",
key: createArchivedTask(z2).Key(),
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1},
"critical": {},
"low": {z3},
},
2020-09-05 22:03:43 +08:00
wantPending: map[string][]*base.TaskMessage{
2020-07-13 21:29:41 +08:00
"default": {},
"critical": {m2},
"low": {},
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-09-05 22:03:43 +08:00
asynqtest.SeedAllPendingQueues(t, r, tc.pending)
2020-07-13 21:29:41 +08:00
if err := inspector.RunTaskByKey(tc.qname, tc.key); err != nil {
t.Errorf("RunTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
2020-09-05 22:03:43 +08:00
for qname, want := range tc.wantPending {
gotPending := asynqtest.GetPendingMessages(t, r, qname)
if diff := cmp.Diff(want, gotPending, asynqtest.SortMsgOpt); diff != "" {
t.Errorf("unexpected pending tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
}
}
func TestInspectorKillTaskByKeyKillsScheduledTask(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
2020-08-17 05:51:56 +08:00
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "custom")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
2020-08-17 05:51:56 +08:00
scheduled map[string][]base.Z
archived map[string][]base.Z
2020-08-17 05:51:56 +08:00
qname string
2020-08-18 21:01:38 +08:00
key string
2020-08-17 05:51:56 +08:00
want string
wantScheduled map[string][]base.Z
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
scheduled: map[string][]base.Z{
"default": {z1},
"custom": {z2, z3},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "custom",
key: createScheduledTask(z2).Key(),
2020-08-18 21:01:38 +08:00
wantScheduled: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {z1},
"custom": {z3},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
2020-12-20 22:09:51 +08:00
"custom": {
{
Message: m2,
Score: now.Unix(),
},
},
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllScheduledQueues(t, r, tc.scheduled)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
if err := inspector.ArchiveTaskByKey(tc.qname, tc.key); err != nil {
2020-08-18 21:46:19 +08:00
t.Errorf("KillTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantScheduled {
gotScheduled := asynqtest.GetScheduledEntries(t, r, qname)
if diff := cmp.Diff(want, gotScheduled, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected scheduled tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
2020-08-17 05:51:56 +08:00
2020-07-13 21:29:41 +08:00
}
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
}
}
func TestInspectorKillTaskByKeyKillsRetryTask(t *testing.T) {
r := setup(t)
2020-09-08 21:51:01 +08:00
defer r.Close()
2020-07-13 21:29:41 +08:00
m1 := asynqtest.NewTaskMessage("task1", nil)
2020-08-17 05:51:56 +08:00
m2 := asynqtest.NewTaskMessageWithQueue("task2", nil, "custom")
m3 := asynqtest.NewTaskMessageWithQueue("task3", nil, "custom")
2020-07-13 21:29:41 +08:00
now := time.Now()
z1 := base.Z{Message: m1, Score: now.Add(5 * time.Minute).Unix()}
z2 := base.Z{Message: m2, Score: now.Add(15 * time.Minute).Unix()}
z3 := base.Z{Message: m3, Score: now.Add(2 * time.Minute).Unix()}
inspector := NewInspector(getRedisConnOpt(t))
2020-07-13 21:29:41 +08:00
tests := []struct {
retry map[string][]base.Z
archived map[string][]base.Z
qname string
key string
wantRetry map[string][]base.Z
wantArchived map[string][]base.Z
2020-07-13 21:29:41 +08:00
}{
{
2020-08-17 05:51:56 +08:00
retry: map[string][]base.Z{
"default": {z1},
"custom": {z2, z3},
},
archived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
"custom": {},
},
qname: "custom",
key: createRetryTask(z2).Key(),
wantRetry: map[string][]base.Z{
"default": {z1},
"custom": {z3},
},
wantArchived: map[string][]base.Z{
2020-08-17 05:51:56 +08:00
"default": {},
2020-12-20 22:09:51 +08:00
"custom": {
{
Message: m2,
Score: now.Unix(),
},
},
2020-07-13 21:29:41 +08:00
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
2020-08-17 05:51:56 +08:00
asynqtest.SeedAllRetryQueues(t, r, tc.retry)
asynqtest.SeedAllArchivedQueues(t, r, tc.archived)
2020-07-13 21:29:41 +08:00
if err := inspector.ArchiveTaskByKey(tc.qname, tc.key); err != nil {
2020-08-18 21:46:19 +08:00
t.Errorf("KillTaskByKey(%q, %q) returned error: %v", tc.qname, tc.key, err)
2020-07-13 21:29:41 +08:00
continue
}
2020-08-17 05:51:56 +08:00
for qname, want := range tc.wantRetry {
gotRetry := asynqtest.GetRetryEntries(t, r, qname)
if diff := cmp.Diff(want, gotRetry, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected retry tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-07-13 21:29:41 +08:00
}
}
for qname, want := range tc.wantArchived {
wantArchived := asynqtest.GetArchivedEntries(t, r, qname)
if diff := cmp.Diff(want, wantArchived, asynqtest.SortZSetEntryOpt); diff != "" {
t.Errorf("unexpected archived tasks in queue %q: (-want, +got)\n%s",
2020-08-18 21:46:19 +08:00
qname, diff)
2020-08-17 05:51:56 +08:00
}
2020-07-13 21:29:41 +08:00
}
}
}
var sortSchedulerEntry = cmp.Transformer("SortSchedulerEntry", func(in []*SchedulerEntry) []*SchedulerEntry {
out := append([]*SchedulerEntry(nil), in...)
sort.Slice(out, func(i, j int) bool {
return out[i].Spec < out[j].Spec
})
return out
})
func TestInspectorSchedulerEntries(t *testing.T) {
r := setup(t)
rdbClient := rdb.NewRDB(r)
inspector := NewInspector(getRedisConnOpt(t))
now := time.Now().UTC()
schedulerID := "127.0.0.1:9876:abc123"
tests := []struct {
data []*base.SchedulerEntry // data to seed redis
want []*SchedulerEntry
}{
{
data: []*base.SchedulerEntry{
&base.SchedulerEntry{
Spec: "* * * * *",
Type: "foo",
Payload: nil,
Opts: nil,
Next: now.Add(5 * time.Hour),
Prev: now.Add(-2 * time.Hour),
},
&base.SchedulerEntry{
Spec: "@every 20m",
Type: "bar",
Payload: map[string]interface{}{"fiz": "baz"},
Opts: []string{`Queue("bar")`, `MaxRetry(20)`},
Next: now.Add(1 * time.Minute),
Prev: now.Add(-19 * time.Minute),
},
},
want: []*SchedulerEntry{
&SchedulerEntry{
Spec: "* * * * *",
Task: NewTask("foo", nil),
Opts: nil,
Next: now.Add(5 * time.Hour),
Prev: now.Add(-2 * time.Hour),
},
&SchedulerEntry{
Spec: "@every 20m",
Task: NewTask("bar", map[string]interface{}{"fiz": "baz"}),
Opts: []Option{Queue("bar"), MaxRetry(20)},
Next: now.Add(1 * time.Minute),
Prev: now.Add(-19 * time.Minute),
},
},
},
}
for _, tc := range tests {
asynqtest.FlushDB(t, r)
err := rdbClient.WriteSchedulerEntries(schedulerID, tc.data, time.Minute)
if err != nil {
t.Fatalf("could not write data: %v", err)
}
got, err := inspector.SchedulerEntries()
if err != nil {
t.Errorf("SchedulerEntries() returned error: %v", err)
continue
}
ignoreOpt := cmpopts.IgnoreUnexported(Payload{})
if diff := cmp.Diff(tc.want, got, sortSchedulerEntry, ignoreOpt); diff != "" {
t.Errorf("SchedulerEntries() = %v, want %v; (-want,+got)\n%s",
got, tc.want, diff)
}
}
}