2
0
mirror of https://github.com/hibiken/asynq.git synced 2024-12-27 00:02:19 +08:00
asynq/internal/rdb/inspect_test.go

1471 lines
38 KiB
Go
Raw Normal View History

2020-01-03 10:13:16 +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.
2019-12-05 12:30:37 +08:00
package rdb
import (
"sort"
"testing"
"time"
"github.com/google/go-cmp/cmp"
h "github.com/hibiken/asynq/internal/asynqtest"
2019-12-22 23:15:45 +08:00
"github.com/hibiken/asynq/internal/base"
"github.com/rs/xid"
2019-12-05 12:30:37 +08:00
)
// TODO(hibiken): Replace this with cmpopts.EquateApproxTime once it becomes availalble.
// https://github.com/google/go-cmp/issues/166
//
// EquateApproxTime returns a Comparer options that
// determine two time.Time values to be equal if they
// are within the given time interval of one another.
// Note that if both times have a monotonic clock reading,
// the monotonic time difference will be used.
//
// The zero time is treated specially: it is only considered
// equal to another zero time value.
//
// It will panic if margin is negative.
func EquateApproxTime(margin time.Duration) cmp.Option {
if margin < 0 {
panic("negative duration in EquateApproxTime")
}
return cmp.FilterValues(func(x, y time.Time) bool {
return !x.IsZero() && !y.IsZero()
}, cmp.Comparer(timeApproximator{margin}.compare))
}
type timeApproximator struct {
margin time.Duration
}
func (a timeApproximator) compare(x, y time.Time) bool {
// Avoid subtracting times to avoid overflow when the
// difference is larger than the largest representible duration.
if x.After(y) {
// Ensure x is always before y
x, y = y, x
}
// We're within the margin if x+margin >= y.
// Note: time.Time doesn't have AfterOrEqual method hence the negation.
return !x.Add(a.margin).Before(y)
}
2019-12-05 12:30:37 +08:00
func TestCurrentStats(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", map[string]interface{}{"subject": "hello"})
m2 := h.NewTaskMessage("reindex", nil)
m3 := h.NewTaskMessage("gen_thumbnail", map[string]interface{}{"src": "some/path/to/img"})
m4 := h.NewTaskMessage("sync", nil)
now := time.Now()
2019-12-05 12:30:37 +08:00
tests := []struct {
2019-12-22 23:15:45 +08:00
enqueued []*base.TaskMessage
inProgress []*base.TaskMessage
scheduled []h.ZSetEntry
retry []h.ZSetEntry
dead []h.ZSetEntry
processed int
failed int
2019-12-05 12:30:37 +08:00
want *Stats
}{
{
2019-12-22 23:15:45 +08:00
enqueued: []*base.TaskMessage{m1},
inProgress: []*base.TaskMessage{m2},
scheduled: []h.ZSetEntry{
{Msg: m3, Score: now.Add(time.Hour).Unix()},
{Msg: m4, Score: now.Unix()}},
retry: []h.ZSetEntry{},
dead: []h.ZSetEntry{},
processed: 120,
failed: 2,
2019-12-05 12:30:37 +08:00
want: &Stats{
Enqueued: 1,
InProgress: 1,
Scheduled: 2,
Retry: 0,
Dead: 0,
Processed: 120,
Failed: 2,
Timestamp: now,
2019-12-05 12:30:37 +08:00
},
},
{
2019-12-22 23:15:45 +08:00
enqueued: []*base.TaskMessage{},
inProgress: []*base.TaskMessage{},
scheduled: []h.ZSetEntry{
{Msg: m3, Score: now.Unix()},
{Msg: m4, Score: now.Unix()}},
retry: []h.ZSetEntry{
{Msg: m1, Score: now.Add(time.Minute).Unix()}},
dead: []h.ZSetEntry{
{Msg: m2, Score: now.Add(-time.Hour).Unix()}},
processed: 90,
failed: 10,
2019-12-05 12:30:37 +08:00
want: &Stats{
Enqueued: 0,
InProgress: 0,
Scheduled: 2,
Retry: 1,
Dead: 1,
Processed: 90,
Failed: 10,
Timestamp: now,
2019-12-05 12:30:37 +08:00
},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedEnqueuedQueue(t, r.client, tc.enqueued)
h.SeedInProgressQueue(t, r.client, tc.inProgress)
h.SeedScheduledQueue(t, r.client, tc.scheduled)
h.SeedRetryQueue(t, r.client, tc.retry)
h.SeedDeadQueue(t, r.client, tc.dead)
processedKey := base.ProcessedKey(now)
failedKey := base.FailureKey(now)
r.client.Set(processedKey, tc.processed, 0)
r.client.Set(failedKey, tc.failed, 0)
2019-12-05 12:30:37 +08:00
got, err := r.CurrentStats()
if err != nil {
t.Errorf("r.CurrentStats() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
if diff := cmp.Diff(tc.want, got, timeCmpOpt); diff != "" {
t.Errorf("r.CurrentStats() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
2019-12-23 01:09:57 +08:00
}
func TestCurrentStatsWithoutData(t *testing.T) {
r := setup(t)
want := &Stats{
Enqueued: 0,
InProgress: 0,
Scheduled: 0,
Retry: 0,
Dead: 0,
Processed: 0,
Failed: 0,
Timestamp: time.Now(),
}
got, err := r.CurrentStats()
if err != nil {
t.Fatalf("r.CurrentStats() = %v, %v, want %+v, nil", got, err, want)
}
if diff := cmp.Diff(want, got, timeCmpOpt); diff != "" {
t.Errorf("r.CurrentStats() = %v, %v, want %+v, nil; (-want, +got)\n%s", got, err, want, diff)
}
}
2020-01-05 01:41:05 +08:00
func TestHistoricalStats(t *testing.T) {
r := setup(t)
now := time.Now().UTC()
tests := []struct {
n int // number of days
}{
{90},
{7},
{0},
}
for _, tc := range tests {
h.FlushDB(t, r.client)
// populate last n days data
for i := 0; i < tc.n; i++ {
ts := now.Add(-time.Duration(i) * 24 * time.Hour)
processedKey := base.ProcessedKey(ts)
failedKey := base.FailureKey(ts)
r.client.Set(processedKey, (i+1)*1000, 0)
r.client.Set(failedKey, (i+1)*10, 0)
}
got, err := r.HistoricalStats(tc.n)
if err != nil {
t.Errorf("RDB.HistoricalStats(%v) returned error: %v", tc.n, err)
continue
}
if len(got) != tc.n {
t.Errorf("RDB.HistorycalStats(%v) returned %d daily stats, want %d", tc.n, len(got), tc.n)
continue
}
for i := 0; i < tc.n; i++ {
want := &DailyStats{
Processed: (i + 1) * 1000,
Failed: (i + 1) * 10,
Time: now.Add(-time.Duration(i) * 24 * time.Hour),
}
if diff := cmp.Diff(want, got[i], timeCmpOpt); diff != "" {
t.Errorf("RDB.HistoricalStats %d days ago data; got %+v, want %+v; (-want,+got):\n%s", i, got[i], want, diff)
}
}
}
}
2019-12-23 01:09:57 +08:00
func TestRedisInfo(t *testing.T) {
r := setup(t)
info, err := r.RedisInfo()
if err != nil {
t.Fatalf("RDB.RedisInfo() returned error: %v", err)
}
2019-12-05 12:30:37 +08:00
2019-12-23 01:09:57 +08:00
wantKeys := []string{
"redis_version",
"uptime_in_days",
"connected_clients",
"used_memory_human",
"used_memory_peak_human",
"used_memory_peak_perc",
}
for _, key := range wantKeys {
if _, ok := info[key]; !ok {
t.Errorf("RDB.RedisInfo() = %v is missing entry for %q", info, key)
}
}
2019-12-05 12:30:37 +08:00
}
func TestListEnqueued(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", map[string]interface{}{"subject": "hello"})
m2 := h.NewTaskMessage("reindex", nil)
2019-12-05 12:30:37 +08:00
t1 := &EnqueuedTask{ID: m1.ID, Type: m1.Type, Payload: m1.Payload}
t2 := &EnqueuedTask{ID: m2.ID, Type: m2.Type, Payload: m2.Payload}
tests := []struct {
2019-12-22 23:15:45 +08:00
enqueued []*base.TaskMessage
2019-12-05 12:30:37 +08:00
want []*EnqueuedTask
}{
{
2019-12-22 23:15:45 +08:00
enqueued: []*base.TaskMessage{m1, m2},
2019-12-05 12:30:37 +08:00
want: []*EnqueuedTask{t1, t2},
},
{
2019-12-22 23:15:45 +08:00
enqueued: []*base.TaskMessage{},
2019-12-05 12:30:37 +08:00
want: []*EnqueuedTask{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedEnqueuedQueue(t, r.client, tc.enqueued)
2019-12-13 11:49:41 +08:00
2019-12-05 12:30:37 +08:00
got, err := r.ListEnqueued()
if err != nil {
t.Errorf("r.ListEnqueued() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
sortOpt := cmp.Transformer("SortMsg", func(in []*EnqueuedTask) []*EnqueuedTask {
out := append([]*EnqueuedTask(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() < out[j].ID.String()
})
return out
})
if diff := cmp.Diff(tc.want, got, sortOpt); diff != "" {
t.Errorf("r.ListEnqueued() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
}
func TestListInProgress(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", map[string]interface{}{"subject": "hello"})
m2 := h.NewTaskMessage("reindex", nil)
2019-12-05 12:30:37 +08:00
t1 := &InProgressTask{ID: m1.ID, Type: m1.Type, Payload: m1.Payload}
t2 := &InProgressTask{ID: m2.ID, Type: m2.Type, Payload: m2.Payload}
tests := []struct {
2019-12-22 23:15:45 +08:00
inProgress []*base.TaskMessage
2019-12-13 11:49:41 +08:00
want []*InProgressTask
2019-12-05 12:30:37 +08:00
}{
{
2019-12-22 23:15:45 +08:00
inProgress: []*base.TaskMessage{m1, m2},
2019-12-13 11:49:41 +08:00
want: []*InProgressTask{t1, t2},
2019-12-05 12:30:37 +08:00
},
{
2019-12-22 23:15:45 +08:00
inProgress: []*base.TaskMessage{},
2019-12-13 11:49:41 +08:00
want: []*InProgressTask{},
2019-12-05 12:30:37 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedInProgressQueue(t, r.client, tc.inProgress)
2019-12-13 11:49:41 +08:00
2019-12-05 12:30:37 +08:00
got, err := r.ListInProgress()
if err != nil {
t.Errorf("r.ListInProgress() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
sortOpt := cmp.Transformer("SortMsg", func(in []*InProgressTask) []*InProgressTask {
out := append([]*InProgressTask(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() < out[j].ID.String()
})
return out
})
if diff := cmp.Diff(tc.want, got, sortOpt); diff != "" {
t.Errorf("r.ListInProgress() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
}
func TestListScheduled(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", map[string]interface{}{"subject": "hello"})
m2 := h.NewTaskMessage("reindex", nil)
2019-12-05 12:30:37 +08:00
p1 := time.Now().Add(30 * time.Minute)
p2 := time.Now().Add(24 * time.Hour)
2019-12-09 08:36:08 +08:00
t1 := &ScheduledTask{ID: m1.ID, Type: m1.Type, Payload: m1.Payload, ProcessAt: p1, Score: p1.Unix()}
t2 := &ScheduledTask{ID: m2.ID, Type: m2.Type, Payload: m2.Payload, ProcessAt: p2, Score: p2.Unix()}
2019-12-05 12:30:37 +08:00
tests := []struct {
scheduled []h.ZSetEntry
2019-12-05 12:30:37 +08:00
want []*ScheduledTask
}{
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: p1.Unix()},
{Msg: m2, Score: p2.Unix()},
2019-12-05 12:30:37 +08:00
},
want: []*ScheduledTask{t1, t2},
},
{
scheduled: []h.ZSetEntry{},
2019-12-05 12:30:37 +08:00
want: []*ScheduledTask{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedScheduledQueue(t, r.client, tc.scheduled)
2019-12-05 12:30:37 +08:00
got, err := r.ListScheduled()
if err != nil {
t.Errorf("r.ListScheduled() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
sortOpt := cmp.Transformer("SortMsg", func(in []*ScheduledTask) []*ScheduledTask {
out := append([]*ScheduledTask(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() < out[j].ID.String()
})
return out
})
if diff := cmp.Diff(tc.want, got, sortOpt, timeCmpOpt); diff != "" {
t.Errorf("r.ListScheduled() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
}
func TestListRetry(t *testing.T) {
r := setup(t)
2019-12-22 23:15:45 +08:00
m1 := &base.TaskMessage{
ID: xid.New(),
2019-12-05 12:30:37 +08:00
Type: "send_email",
Queue: "default",
Payload: map[string]interface{}{"subject": "hello"},
ErrorMsg: "email server not responding",
Retry: 25,
Retried: 10,
}
2019-12-22 23:15:45 +08:00
m2 := &base.TaskMessage{
ID: xid.New(),
2019-12-05 12:30:37 +08:00
Type: "reindex",
Queue: "default",
Payload: nil,
ErrorMsg: "search engine not responding",
Retry: 25,
Retried: 2,
}
p1 := time.Now().Add(5 * time.Minute)
p2 := time.Now().Add(24 * time.Hour)
2019-12-13 11:49:41 +08:00
t1 := &RetryTask{
ID: m1.ID,
Type: m1.Type,
Payload: m1.Payload,
ProcessAt: p1,
ErrorMsg: m1.ErrorMsg,
Retried: m1.Retried,
Retry: m1.Retry,
Score: p1.Unix(),
}
t2 := &RetryTask{
ID: m2.ID,
Type: m2.Type,
Payload: m2.Payload,
ProcessAt: p2,
ErrorMsg: m2.ErrorMsg,
Retried: m2.Retried,
Retry: m2.Retry,
Score: p2.Unix(),
2019-12-05 12:30:37 +08:00
}
tests := []struct {
retry []h.ZSetEntry
2019-12-13 11:49:41 +08:00
want []*RetryTask
2019-12-05 12:30:37 +08:00
}{
{
retry: []h.ZSetEntry{
{Msg: m1, Score: p1.Unix()},
{Msg: m2, Score: p2.Unix()},
2019-12-05 12:30:37 +08:00
},
want: []*RetryTask{t1, t2},
},
{
retry: []h.ZSetEntry{},
2019-12-13 11:49:41 +08:00
want: []*RetryTask{},
2019-12-05 12:30:37 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedRetryQueue(t, r.client, tc.retry)
2019-12-05 12:30:37 +08:00
got, err := r.ListRetry()
if err != nil {
t.Errorf("r.ListRetry() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
sortOpt := cmp.Transformer("SortMsg", func(in []*RetryTask) []*RetryTask {
out := append([]*RetryTask(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() < out[j].ID.String()
})
return out
})
if diff := cmp.Diff(tc.want, got, sortOpt, timeCmpOpt); diff != "" {
t.Errorf("r.ListRetry() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
}
func TestListDead(t *testing.T) {
r := setup(t)
2019-12-22 23:15:45 +08:00
m1 := &base.TaskMessage{
ID: xid.New(),
2019-12-05 12:30:37 +08:00
Type: "send_email",
Queue: "default",
Payload: map[string]interface{}{"subject": "hello"},
ErrorMsg: "email server not responding",
}
2019-12-22 23:15:45 +08:00
m2 := &base.TaskMessage{
ID: xid.New(),
2019-12-05 12:30:37 +08:00
Type: "reindex",
Queue: "default",
Payload: nil,
ErrorMsg: "search engine not responding",
}
f1 := time.Now().Add(-5 * time.Minute)
f2 := time.Now().Add(-24 * time.Hour)
2019-12-13 11:49:41 +08:00
t1 := &DeadTask{
ID: m1.ID,
Type: m1.Type,
Payload: m1.Payload,
LastFailedAt: f1,
ErrorMsg: m1.ErrorMsg,
Score: f1.Unix(),
}
t2 := &DeadTask{
ID: m2.ID,
Type: m2.Type,
Payload: m2.Payload,
LastFailedAt: f2,
ErrorMsg: m2.ErrorMsg,
Score: f2.Unix(),
2019-12-05 12:30:37 +08:00
}
tests := []struct {
dead []h.ZSetEntry
2019-12-05 12:30:37 +08:00
want []*DeadTask
}{
{
dead: []h.ZSetEntry{
{Msg: m1, Score: f1.Unix()},
{Msg: m2, Score: f2.Unix()},
2019-12-05 12:30:37 +08:00
},
want: []*DeadTask{t1, t2},
},
{
dead: []h.ZSetEntry{},
2019-12-05 12:30:37 +08:00
want: []*DeadTask{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedDeadQueue(t, r.client, tc.dead)
2019-12-05 12:30:37 +08:00
got, err := r.ListDead()
if err != nil {
t.Errorf("r.ListDead() = %v, %v, want %v, nil", got, err, tc.want)
continue
}
sortOpt := cmp.Transformer("SortMsg", func(in []*DeadTask) []*DeadTask {
out := append([]*DeadTask(nil), in...) // Copy input to avoid mutating it
sort.Slice(out, func(i, j int) bool {
return out[i].ID.String() < out[j].ID.String()
})
return out
})
if diff := cmp.Diff(tc.want, got, sortOpt, timeCmpOpt); diff != "" {
t.Errorf("r.ListDead() = %v, %v, want %v, nil; (-want, +got)\n%s", got, err, tc.want, diff)
continue
}
}
}
var timeCmpOpt = EquateApproxTime(time.Second)
func TestEnqueueDeadTask(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-10 12:37:30 +08:00
s1 := time.Now().Add(-5 * time.Minute).Unix()
s2 := time.Now().Add(-time.Hour).Unix()
2019-12-13 11:49:41 +08:00
tests := []struct {
dead []h.ZSetEntry
2019-12-10 12:37:30 +08:00
score int64
id xid.ID
want error // expected return value from calling EnqueueDeadTask
2019-12-22 23:15:45 +08:00
wantDead []*base.TaskMessage
wantEnqueued []*base.TaskMessage
}{
{
dead: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
score: s2,
2019-12-10 12:37:30 +08:00
id: t2.ID,
want: nil,
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{t1},
wantEnqueued: []*base.TaskMessage{t2},
},
{
dead: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
2019-12-10 12:37:30 +08:00
score: 123,
id: t2.ID,
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{t1, t2},
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedDeadQueue(t, r.client, tc.dead)
got := r.EnqueueDeadTask(tc.id, tc.score)
if got != tc.want {
2019-12-10 12:37:30 +08:00
t.Errorf("r.EnqueueDeadTask(%s, %d) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.DefaultQueue, diff)
}
gotDead := h.GetDeadMessages(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q, (-want, +got)\n%s", base.DeadQueue, diff)
}
2019-12-05 12:30:37 +08:00
}
}
func TestEnqueueRetryTask(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-10 12:37:30 +08:00
s1 := time.Now().Add(-5 * time.Minute).Unix()
s2 := time.Now().Add(-time.Hour).Unix()
tests := []struct {
retry []h.ZSetEntry
2019-12-10 12:37:30 +08:00
score int64
id xid.ID
want error // expected return value from calling EnqueueRetryTask
2019-12-22 23:15:45 +08:00
wantRetry []*base.TaskMessage
wantEnqueued []*base.TaskMessage
}{
{
retry: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
score: s2,
2019-12-10 12:37:30 +08:00
id: t2.ID,
want: nil,
2019-12-22 23:15:45 +08:00
wantRetry: []*base.TaskMessage{t1},
wantEnqueued: []*base.TaskMessage{t2},
},
{
retry: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
2019-12-10 12:37:30 +08:00
score: 123,
id: t2.ID,
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantRetry: []*base.TaskMessage{t1, t2},
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedRetryQueue(t, r.client, tc.retry) // initialize retry queue
got := r.EnqueueRetryTask(tc.id, tc.score)
if got != tc.want {
2019-12-10 12:37:30 +08:00
t.Errorf("r.EnqueueRetryTask(%s, %d) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.DefaultQueue, diff)
}
gotRetry := h.GetRetryMessages(t, r.client)
if diff := cmp.Diff(tc.wantRetry, gotRetry, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q, (-want, +got)\n%s", base.RetryQueue, diff)
}
}
2019-12-05 12:30:37 +08:00
}
func TestEnqueueScheduledTask(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-10 12:37:30 +08:00
s1 := time.Now().Add(-5 * time.Minute).Unix()
s2 := time.Now().Add(-time.Hour).Unix()
2019-12-13 11:49:41 +08:00
tests := []struct {
scheduled []h.ZSetEntry
2019-12-10 12:37:30 +08:00
score int64
id xid.ID
want error // expected return value from calling EnqueueScheduledTask
2019-12-22 23:15:45 +08:00
wantScheduled []*base.TaskMessage
wantEnqueued []*base.TaskMessage
}{
{
scheduled: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
score: s2,
2019-12-10 12:37:30 +08:00
id: t2.ID,
want: nil,
2019-12-22 23:15:45 +08:00
wantScheduled: []*base.TaskMessage{t1},
wantEnqueued: []*base.TaskMessage{t2},
},
{
scheduled: []h.ZSetEntry{
{Msg: t1, Score: s1},
{Msg: t2, Score: s2},
},
2019-12-10 12:37:30 +08:00
score: 123,
id: t2.ID,
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantScheduled: []*base.TaskMessage{t1, t2},
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedScheduledQueue(t, r.client, tc.scheduled)
got := r.EnqueueScheduledTask(tc.id, tc.score)
if got != tc.want {
2019-12-10 12:37:30 +08:00
t.Errorf("r.EnqueueRetryTask(%s, %d) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.DefaultQueue, diff)
}
gotScheduled := h.GetScheduledMessages(t, r.client)
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q, (-want, +got)\n%s", base.ScheduledQueue, diff)
}
2019-12-05 12:30:37 +08:00
}
}
func TestEnqueueAllScheduledTasks(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
t3 := h.NewTaskMessage("reindex", nil)
tests := []struct {
2019-12-13 11:49:41 +08:00
desc string
scheduled []h.ZSetEntry
2019-12-11 13:38:25 +08:00
want int64
2019-12-22 23:15:45 +08:00
wantEnqueued []*base.TaskMessage
}{
{
2019-12-13 11:49:41 +08:00
desc: "with tasks in scheduled queue",
scheduled: []h.ZSetEntry{
{Msg: t1, Score: time.Now().Add(time.Hour).Unix()},
{Msg: t2, Score: time.Now().Add(time.Hour).Unix()},
{Msg: t3, Score: time.Now().Add(time.Hour).Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-11 13:38:25 +08:00
want: 3,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{t1, t2, t3},
},
{
2019-12-13 11:49:41 +08:00
desc: "with empty scheduled queue",
scheduled: []h.ZSetEntry{},
2019-12-11 13:38:25 +08:00
want: 0,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedScheduledQueue(t, r.client, tc.scheduled)
2019-12-11 13:38:25 +08:00
got, err := r.EnqueueAllScheduledTasks()
if err != nil {
2019-12-11 13:38:25 +08:00
t.Errorf("%s; r.EnqueueAllScheduledTasks = %v, %v; want %v, nil",
2019-12-13 11:49:41 +08:00
tc.desc, got, err, tc.want)
2019-12-11 13:38:25 +08:00
continue
}
if got != tc.want {
t.Errorf("%s; r.EnqueueAllScheduledTasks = %v, %v; want %v, nil",
2019-12-13 11:49:41 +08:00
tc.desc, got, err, tc.want)
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("%s; mismatch found in %q; (-want, +got)\n%s", tc.desc, base.DefaultQueue, diff)
}
}
}
func TestEnqueueAllRetryTasks(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
t3 := h.NewTaskMessage("reindex", nil)
tests := []struct {
description string
retry []h.ZSetEntry
2019-12-11 13:38:25 +08:00
want int64
2019-12-22 23:15:45 +08:00
wantEnqueued []*base.TaskMessage
}{
{
2019-12-13 11:49:41 +08:00
description: "with tasks in retry queue",
retry: []h.ZSetEntry{
{Msg: t1, Score: time.Now().Add(time.Hour).Unix()},
{Msg: t2, Score: time.Now().Add(time.Hour).Unix()},
{Msg: t3, Score: time.Now().Add(time.Hour).Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-11 13:38:25 +08:00
want: 3,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{t1, t2, t3},
},
{
description: "with empty retry queue",
retry: []h.ZSetEntry{},
2019-12-11 13:38:25 +08:00
want: 0,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedRetryQueue(t, r.client, tc.retry)
2019-12-11 13:38:25 +08:00
got, err := r.EnqueueAllRetryTasks()
if err != nil {
2019-12-11 13:38:25 +08:00
t.Errorf("%s; r.EnqueueAllRetryTasks = %v, %v; want %v, nil",
tc.description, got, err, tc.want)
continue
}
if got != tc.want {
t.Errorf("%s; r.EnqueueAllRetryTasks = %v, %v; want %v, nil",
tc.description, got, err, tc.want)
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("%s; mismatch found in %q; (-want, +got)\n%s", tc.description, base.DefaultQueue, diff)
}
}
}
func TestEnqueueAllDeadTasks(t *testing.T) {
r := setup(t)
t1 := h.NewTaskMessage("send_email", nil)
t2 := h.NewTaskMessage("gen_thumbnail", nil)
t3 := h.NewTaskMessage("reindex", nil)
tests := []struct {
2019-12-13 11:49:41 +08:00
desc string
dead []h.ZSetEntry
2019-12-11 13:38:25 +08:00
want int64
2019-12-22 23:15:45 +08:00
wantEnqueued []*base.TaskMessage
}{
{
2019-12-13 11:49:41 +08:00
desc: "with tasks in dead queue",
dead: []h.ZSetEntry{
{Msg: t1, Score: time.Now().Add(-time.Minute).Unix()},
{Msg: t2, Score: time.Now().Add(-time.Minute).Unix()},
{Msg: t3, Score: time.Now().Add(-time.Minute).Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-11 13:38:25 +08:00
want: 3,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{t1, t2, t3},
},
{
2019-12-13 11:49:41 +08:00
desc: "with empty dead queue",
dead: []h.ZSetEntry{},
2019-12-11 13:38:25 +08:00
want: 0,
2019-12-22 23:15:45 +08:00
wantEnqueued: []*base.TaskMessage{},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedDeadQueue(t, r.client, tc.dead)
2019-12-11 13:38:25 +08:00
got, err := r.EnqueueAllDeadTasks()
if err != nil {
2019-12-11 13:38:25 +08:00
t.Errorf("%s; r.EnqueueAllDeadTasks = %v, %v; want %v, nil",
2019-12-13 11:49:41 +08:00
tc.desc, got, err, tc.want)
2019-12-11 13:38:25 +08:00
continue
}
if got != tc.want {
t.Errorf("%s; r.EnqueueAllDeadTasks = %v, %v; want %v, nil",
2019-12-13 11:49:41 +08:00
tc.desc, got, err, tc.want)
}
gotEnqueued := h.GetEnqueuedMessages(t, r.client)
if diff := cmp.Diff(tc.wantEnqueued, gotEnqueued, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("%s; mismatch found in %q; (-want, +got)\n%s", tc.desc, base.DefaultQueue, diff)
}
}
}
2019-12-12 11:56:19 +08:00
func TestKillRetryTask(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
t1 := time.Now().Add(time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
retry []h.ZSetEntry
dead []h.ZSetEntry
id xid.ID
score int64
want error
wantRetry []h.ZSetEntry
wantDead []h.ZSetEntry
}{
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
dead: []h.ZSetEntry{},
id: m1.ID,
score: t1.Unix(),
want: nil,
wantRetry: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
},
},
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
dead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
id: m2.ID,
score: t2.Unix(),
want: ErrTaskNotFound,
wantRetry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
wantDead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client)
h.SeedRetryQueue(t, r.client, tc.retry)
h.SeedDeadQueue(t, r.client, tc.dead)
got := r.KillRetryTask(tc.id, tc.score)
if got != tc.want {
t.Errorf("(*RDB).KillRetryTask(%v, %v) = %v, want %v",
tc.id, tc.score, got, tc.want)
continue
}
gotRetry := h.GetRetryEntries(t, r.client)
if diff := cmp.Diff(tc.wantRetry, gotRetry, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.RetryQueue, diff)
}
gotDead := h.GetDeadEntries(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.DeadQueue, diff)
}
}
}
func TestKillScheduledTask(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
t1 := time.Now().Add(time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
scheduled []h.ZSetEntry
dead []h.ZSetEntry
id xid.ID
score int64
want error
wantScheduled []h.ZSetEntry
wantDead []h.ZSetEntry
}{
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
dead: []h.ZSetEntry{},
id: m1.ID,
score: t1.Unix(),
want: nil,
wantScheduled: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
},
},
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
dead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
id: m2.ID,
score: t2.Unix(),
want: ErrTaskNotFound,
wantScheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
wantDead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client)
h.SeedScheduledQueue(t, r.client, tc.scheduled)
h.SeedDeadQueue(t, r.client, tc.dead)
got := r.KillScheduledTask(tc.id, tc.score)
if got != tc.want {
t.Errorf("(*RDB).KillScheduledTask(%v, %v) = %v, want %v",
tc.id, tc.score, got, tc.want)
continue
}
gotScheduled := h.GetScheduledEntries(t, r.client)
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.ScheduledQueue, diff)
}
gotDead := h.GetDeadEntries(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.DeadQueue, diff)
}
}
}
func TestKillAllRetryTasks(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
t1 := time.Now().Add(time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
retry []h.ZSetEntry
dead []h.ZSetEntry
want int64
wantRetry []h.ZSetEntry
wantDead []h.ZSetEntry
}{
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
dead: []h.ZSetEntry{},
want: 2,
wantRetry: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: time.Now().Unix()},
},
},
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
dead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
want: 1,
wantRetry: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: t2.Unix()},
},
},
{
retry: []h.ZSetEntry{},
dead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
want: 0,
wantRetry: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client)
h.SeedRetryQueue(t, r.client, tc.retry)
h.SeedDeadQueue(t, r.client, tc.dead)
got, err := r.KillAllRetryTasks()
if got != tc.want || err != nil {
t.Errorf("(*RDB).KillAllRetryTasks() = %v, %v; want %v, nil",
got, err, tc.want)
continue
}
gotRetry := h.GetRetryEntries(t, r.client)
if diff := cmp.Diff(tc.wantRetry, gotRetry, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.RetryQueue, diff)
}
gotDead := h.GetDeadEntries(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.DeadQueue, diff)
}
}
}
func TestKillAllScheduledTasks(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
t1 := time.Now().Add(time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
scheduled []h.ZSetEntry
dead []h.ZSetEntry
want int64
wantScheduled []h.ZSetEntry
wantDead []h.ZSetEntry
}{
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
dead: []h.ZSetEntry{},
want: 2,
wantScheduled: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: time.Now().Unix()},
},
},
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
},
dead: []h.ZSetEntry{
{Msg: m2, Score: t2.Unix()},
},
want: 1,
wantScheduled: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: t2.Unix()},
},
},
{
scheduled: []h.ZSetEntry{},
dead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
want: 0,
wantScheduled: []h.ZSetEntry{},
wantDead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
},
},
}
for _, tc := range tests {
h.FlushDB(t, r.client)
h.SeedScheduledQueue(t, r.client, tc.scheduled)
h.SeedDeadQueue(t, r.client, tc.dead)
got, err := r.KillAllScheduledTasks()
if got != tc.want || err != nil {
t.Errorf("(*RDB).KillAllScheduledTasks() = %v, %v; want %v, nil",
got, err, tc.want)
continue
}
gotScheduled := h.GetScheduledEntries(t, r.client)
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.ScheduledQueue, diff)
}
gotDead := h.GetDeadEntries(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortZSetEntryOpt, timeCmpOpt); diff != "" {
t.Errorf("mismatch found in %q; (-want,+got)\n%s",
base.DeadQueue, diff)
}
}
}
2019-12-12 11:56:19 +08:00
func TestDeleteDeadTask(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
2019-12-12 11:56:19 +08:00
t1 := time.Now().Add(-5 * time.Minute)
t2 := time.Now().Add(-time.Hour)
tests := []struct {
dead []h.ZSetEntry
2019-12-12 11:56:19 +08:00
id xid.ID
score int64
want error
2019-12-22 23:15:45 +08:00
wantDead []*base.TaskMessage
2019-12-12 11:56:19 +08:00
}{
{
dead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m1.ID,
score: t1.Unix(),
want: nil,
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{m2},
2019-12-12 11:56:19 +08:00
},
{
dead: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m1.ID,
score: t2.Unix(), // id and score mismatch
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{m1, m2},
2019-12-12 11:56:19 +08:00
},
{
dead: []h.ZSetEntry{},
2019-12-12 11:56:19 +08:00
id: m1.ID,
score: t1.Unix(),
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{},
2019-12-12 11:56:19 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedDeadQueue(t, r.client, tc.dead)
2019-12-12 11:56:19 +08:00
got := r.DeleteDeadTask(tc.id, tc.score)
if got != tc.want {
t.Errorf("r.DeleteDeadTask(%v, %v) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotDead := h.GetDeadMessages(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.DeadQueue, diff)
2019-12-12 11:56:19 +08:00
}
}
}
func TestDeleteRetryTask(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
2019-12-12 11:56:19 +08:00
t1 := time.Now().Add(5 * time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
retry []h.ZSetEntry
2019-12-12 11:56:19 +08:00
id xid.ID
score int64
want error
2019-12-22 23:15:45 +08:00
wantRetry []*base.TaskMessage
2019-12-12 11:56:19 +08:00
}{
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m1.ID,
score: t1.Unix(),
want: nil,
2019-12-22 23:15:45 +08:00
wantRetry: []*base.TaskMessage{m2},
2019-12-12 11:56:19 +08:00
},
{
retry: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m2.ID,
score: t2.Unix(),
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantRetry: []*base.TaskMessage{m1},
2019-12-12 11:56:19 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedRetryQueue(t, r.client, tc.retry)
2019-12-12 11:56:19 +08:00
got := r.DeleteRetryTask(tc.id, tc.score)
if got != tc.want {
t.Errorf("r.DeleteRetryTask(%v, %v) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotRetry := h.GetRetryMessages(t, r.client)
if diff := cmp.Diff(tc.wantRetry, gotRetry, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.RetryQueue, diff)
2019-12-12 11:56:19 +08:00
}
}
}
func TestDeleteScheduledTask(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
2019-12-12 11:56:19 +08:00
t1 := time.Now().Add(5 * time.Minute)
t2 := time.Now().Add(time.Hour)
tests := []struct {
scheduled []h.ZSetEntry
2019-12-12 11:56:19 +08:00
id xid.ID
score int64
want error
2019-12-22 23:15:45 +08:00
wantScheduled []*base.TaskMessage
2019-12-12 11:56:19 +08:00
}{
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
{Msg: m2, Score: t2.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m1.ID,
score: t1.Unix(),
want: nil,
2019-12-22 23:15:45 +08:00
wantScheduled: []*base.TaskMessage{m2},
2019-12-12 11:56:19 +08:00
},
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: t1.Unix()},
2019-12-12 11:56:19 +08:00
},
id: m2.ID,
score: t2.Unix(),
want: ErrTaskNotFound,
2019-12-22 23:15:45 +08:00
wantScheduled: []*base.TaskMessage{m1},
2019-12-12 11:56:19 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedScheduledQueue(t, r.client, tc.scheduled)
2019-12-12 11:56:19 +08:00
got := r.DeleteScheduledTask(tc.id, tc.score)
if got != tc.want {
t.Errorf("r.DeleteScheduledTask(%v, %v) = %v, want %v", tc.id, tc.score, got, tc.want)
continue
}
gotScheduled := h.GetScheduledMessages(t, r.client)
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.ScheduledQueue, diff)
2019-12-12 11:56:19 +08:00
}
}
}
2019-12-12 22:38:01 +08:00
func TestDeleteAllDeadTasks(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
m3 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-12 22:38:01 +08:00
tests := []struct {
dead []h.ZSetEntry
2019-12-22 23:15:45 +08:00
wantDead []*base.TaskMessage
2019-12-12 22:38:01 +08:00
}{
{
dead: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: time.Now().Unix()},
{Msg: m3, Score: time.Now().Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-22 23:15:45 +08:00
wantDead: []*base.TaskMessage{},
2019-12-12 22:38:01 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedDeadQueue(t, r.client, tc.dead)
2019-12-12 22:38:01 +08:00
err := r.DeleteAllDeadTasks()
if err != nil {
t.Errorf("r.DeleteAllDeaadTasks = %v, want nil", err)
}
gotDead := h.GetDeadMessages(t, r.client)
if diff := cmp.Diff(tc.wantDead, gotDead, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.DeadQueue, diff)
2019-12-12 22:38:01 +08:00
}
}
}
func TestDeleteAllRetryTasks(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
m3 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-12 22:38:01 +08:00
tests := []struct {
retry []h.ZSetEntry
2019-12-22 23:15:45 +08:00
wantRetry []*base.TaskMessage
2019-12-12 22:38:01 +08:00
}{
{
retry: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Unix()},
{Msg: m2, Score: time.Now().Unix()},
{Msg: m3, Score: time.Now().Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-22 23:15:45 +08:00
wantRetry: []*base.TaskMessage{},
2019-12-12 22:38:01 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedRetryQueue(t, r.client, tc.retry)
2019-12-12 22:38:01 +08:00
err := r.DeleteAllRetryTasks()
if err != nil {
t.Errorf("r.DeleteAllDeaadTasks = %v, want nil", err)
}
gotRetry := h.GetRetryMessages(t, r.client)
if diff := cmp.Diff(tc.wantRetry, gotRetry, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.RetryQueue, diff)
2019-12-12 22:38:01 +08:00
}
}
}
func TestDeleteAllScheduledTasks(t *testing.T) {
r := setup(t)
m1 := h.NewTaskMessage("send_email", nil)
m2 := h.NewTaskMessage("reindex", nil)
m3 := h.NewTaskMessage("gen_thumbnail", nil)
2019-12-12 22:38:01 +08:00
tests := []struct {
scheduled []h.ZSetEntry
2019-12-22 23:15:45 +08:00
wantScheduled []*base.TaskMessage
2019-12-12 22:38:01 +08:00
}{
{
scheduled: []h.ZSetEntry{
{Msg: m1, Score: time.Now().Add(time.Minute).Unix()},
{Msg: m2, Score: time.Now().Add(time.Minute).Unix()},
{Msg: m3, Score: time.Now().Add(time.Minute).Unix()},
2019-12-13 11:49:41 +08:00
},
2019-12-22 23:15:45 +08:00
wantScheduled: []*base.TaskMessage{},
2019-12-12 22:38:01 +08:00
},
}
for _, tc := range tests {
h.FlushDB(t, r.client) // clean up db before each test case
h.SeedScheduledQueue(t, r.client, tc.scheduled)
2019-12-12 22:38:01 +08:00
err := r.DeleteAllScheduledTasks()
if err != nil {
t.Errorf("r.DeleteAllDeaadTasks = %v, want nil", err)
}
gotScheduled := h.GetScheduledMessages(t, r.client)
if diff := cmp.Diff(tc.wantScheduled, gotScheduled, h.SortMsgOpt); diff != "" {
2019-12-22 23:15:45 +08:00
t.Errorf("mismatch found in %q; (-want, +got)\n%s", base.ScheduledQueue, diff)
2019-12-12 22:38:01 +08:00
}
}
}