mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-25 23:32:17 +08:00
9c95c41651
* Rename ServerStatus to ServerState internally * Rename terminate to shutdown internally * Update Scheduler API to match Server API
138 lines
4.2 KiB
Go
138 lines
4.2 KiB
Go
// Copyright 2020 Kentaro Hibino. All rights reserved.
|
|
// Use of this source code is governed by a MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package asynq
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
h "github.com/hibiken/asynq/internal/asynqtest"
|
|
"github.com/hibiken/asynq/internal/base"
|
|
"github.com/hibiken/asynq/internal/rdb"
|
|
)
|
|
|
|
func TestForwarder(t *testing.T) {
|
|
r := setup(t)
|
|
defer r.Close()
|
|
rdbClient := rdb.NewRDB(r)
|
|
const pollInterval = time.Second
|
|
s := newForwarder(forwarderParams{
|
|
logger: testLogger,
|
|
broker: rdbClient,
|
|
queues: []string{"default", "critical"},
|
|
interval: pollInterval,
|
|
})
|
|
t1 := h.NewTaskMessageWithQueue("gen_thumbnail", nil, "default")
|
|
t2 := h.NewTaskMessageWithQueue("send_email", nil, "critical")
|
|
t3 := h.NewTaskMessageWithQueue("reindex", nil, "default")
|
|
t4 := h.NewTaskMessageWithQueue("sync", nil, "critical")
|
|
now := time.Now()
|
|
|
|
tests := []struct {
|
|
initScheduled map[string][]base.Z // scheduled queue initial state
|
|
initRetry map[string][]base.Z // retry queue initial state
|
|
initPending map[string][]*base.TaskMessage // default queue initial state
|
|
wait time.Duration // wait duration before checking for final state
|
|
wantScheduled map[string][]*base.TaskMessage // schedule queue final state
|
|
wantRetry map[string][]*base.TaskMessage // retry queue final state
|
|
wantPending map[string][]*base.TaskMessage // default queue final state
|
|
}{
|
|
{
|
|
initScheduled: map[string][]base.Z{
|
|
"default": {{Message: t1, Score: now.Add(time.Hour).Unix()}},
|
|
"critical": {{Message: t2, Score: now.Add(-2 * time.Second).Unix()}},
|
|
},
|
|
initRetry: map[string][]base.Z{
|
|
"default": {{Message: t3, Score: time.Now().Add(-500 * time.Millisecond).Unix()}},
|
|
"critical": {},
|
|
},
|
|
initPending: map[string][]*base.TaskMessage{
|
|
"default": {},
|
|
"critical": {t4},
|
|
},
|
|
wait: pollInterval * 2,
|
|
wantScheduled: map[string][]*base.TaskMessage{
|
|
"default": {t1},
|
|
"critical": {},
|
|
},
|
|
wantRetry: map[string][]*base.TaskMessage{
|
|
"default": {},
|
|
"critical": {},
|
|
},
|
|
wantPending: map[string][]*base.TaskMessage{
|
|
"default": {t3},
|
|
"critical": {t2, t4},
|
|
},
|
|
},
|
|
{
|
|
initScheduled: map[string][]base.Z{
|
|
"default": {
|
|
{Message: t1, Score: now.Unix()},
|
|
{Message: t3, Score: now.Add(-500 * time.Millisecond).Unix()},
|
|
},
|
|
"critical": {
|
|
{Message: t2, Score: now.Add(-2 * time.Second).Unix()},
|
|
},
|
|
},
|
|
initRetry: map[string][]base.Z{
|
|
"default": {},
|
|
"critical": {},
|
|
},
|
|
initPending: map[string][]*base.TaskMessage{
|
|
"default": {},
|
|
"critical": {t4},
|
|
},
|
|
wait: pollInterval * 2,
|
|
wantScheduled: map[string][]*base.TaskMessage{
|
|
"default": {},
|
|
"critical": {},
|
|
},
|
|
wantRetry: map[string][]*base.TaskMessage{
|
|
"default": {},
|
|
"critical": {},
|
|
},
|
|
wantPending: map[string][]*base.TaskMessage{
|
|
"default": {t1, t3},
|
|
"critical": {t2, t4},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
h.FlushDB(t, r) // clean up db before each test case.
|
|
h.SeedAllScheduledQueues(t, r, tc.initScheduled) // initialize scheduled queue
|
|
h.SeedAllRetryQueues(t, r, tc.initRetry) // initialize retry queue
|
|
h.SeedAllPendingQueues(t, r, tc.initPending) // initialize default queue
|
|
|
|
var wg sync.WaitGroup
|
|
s.start(&wg)
|
|
time.Sleep(tc.wait)
|
|
s.shutdown()
|
|
|
|
for qname, want := range tc.wantScheduled {
|
|
gotScheduled := h.GetScheduledMessages(t, r, qname)
|
|
if diff := cmp.Diff(want, gotScheduled, h.SortMsgOpt); diff != "" {
|
|
t.Errorf("mismatch found in %q after running forwarder: (-want, +got)\n%s", base.ScheduledKey(qname), diff)
|
|
}
|
|
}
|
|
|
|
for qname, want := range tc.wantRetry {
|
|
gotRetry := h.GetRetryMessages(t, r, qname)
|
|
if diff := cmp.Diff(want, gotRetry, h.SortMsgOpt); diff != "" {
|
|
t.Errorf("mismatch found in %q after running forwarder: (-want, +got)\n%s", base.RetryKey(qname), diff)
|
|
}
|
|
}
|
|
|
|
for qname, want := range tc.wantPending {
|
|
gotPending := h.GetPendingMessages(t, r, qname)
|
|
if diff := cmp.Diff(want, gotPending, h.SortMsgOpt); diff != "" {
|
|
t.Errorf("mismatch found in %q after running forwarder: (-want, +got)\n%s", base.PendingKey(qname), diff)
|
|
}
|
|
}
|
|
}
|
|
}
|