mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-26 15:52:18 +08:00
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
// Copyright 2021 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"
|
|
"github.com/hibiken/asynq/internal/base"
|
|
"github.com/hibiken/asynq/internal/rdb"
|
|
h "github.com/hibiken/asynq/internal/testutil"
|
|
)
|
|
|
|
func newCompletedTask(qname, tasktype string, payload []byte, completedAt time.Time) *base.TaskMessage {
|
|
msg := h.NewTaskMessageWithQueue(tasktype, payload, qname)
|
|
msg.CompletedAt = completedAt.Unix()
|
|
return msg
|
|
}
|
|
|
|
func TestJanitor(t *testing.T) {
|
|
r := setup(t)
|
|
defer r.Close()
|
|
rdbClient := rdb.NewRDB(r)
|
|
const interval = 1 * time.Second
|
|
janitor := newJanitor(janitorParams{
|
|
logger: testLogger,
|
|
broker: rdbClient,
|
|
queues: []string{"default", "custom"},
|
|
interval: interval,
|
|
})
|
|
|
|
now := time.Now()
|
|
hourAgo := now.Add(-1 * time.Hour)
|
|
minuteAgo := now.Add(-1 * time.Minute)
|
|
halfHourAgo := now.Add(-30 * time.Minute)
|
|
halfHourFromNow := now.Add(30 * time.Minute)
|
|
fiveMinFromNow := now.Add(5 * time.Minute)
|
|
msg1 := newCompletedTask("default", "task1", nil, hourAgo)
|
|
msg2 := newCompletedTask("default", "task2", nil, minuteAgo)
|
|
msg3 := newCompletedTask("custom", "task3", nil, hourAgo)
|
|
msg4 := newCompletedTask("custom", "task4", nil, minuteAgo)
|
|
|
|
tests := []struct {
|
|
completed map[string][]base.Z // initial completed sets
|
|
wantCompleted map[string][]base.Z // expected completed sets after janitor runs
|
|
}{
|
|
{
|
|
completed: map[string][]base.Z{
|
|
"default": {
|
|
{Message: msg1, Score: halfHourAgo.Unix()},
|
|
{Message: msg2, Score: fiveMinFromNow.Unix()},
|
|
},
|
|
"custom": {
|
|
{Message: msg3, Score: halfHourFromNow.Unix()},
|
|
{Message: msg4, Score: minuteAgo.Unix()},
|
|
},
|
|
},
|
|
wantCompleted: map[string][]base.Z{
|
|
"default": {
|
|
{Message: msg2, Score: fiveMinFromNow.Unix()},
|
|
},
|
|
"custom": {
|
|
{Message: msg3, Score: halfHourFromNow.Unix()},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
h.FlushDB(t, r)
|
|
h.SeedAllCompletedQueues(t, r, tc.completed)
|
|
|
|
var wg sync.WaitGroup
|
|
janitor.start(&wg)
|
|
time.Sleep(2 * interval) // make sure to let janitor run at least one time
|
|
janitor.shutdown()
|
|
|
|
for qname, want := range tc.wantCompleted {
|
|
got := h.GetCompletedEntries(t, r, qname)
|
|
if diff := cmp.Diff(want, got, h.SortZSetEntryOpt); diff != "" {
|
|
t.Errorf("diff found in %q after running janitor: (-want, +got)\n%s", base.CompletedKey(qname), diff)
|
|
}
|
|
}
|
|
}
|
|
}
|