Update RDB.Dequeue to return deadline as time.Time

This commit is contained in:
Ken Hibino
2020-06-18 11:25:33 -07:00
parent 113451ce6a
commit 92af00f9fd
2 changed files with 15 additions and 15 deletions

View File

@@ -144,7 +144,7 @@ func TestDequeue(t *testing.T) {
enqueued map[string][]*base.TaskMessage
args []string // list of queues to query
wantMsg *base.TaskMessage
wantDeadline int
wantDeadline time.Time
err error
wantEnqueued map[string][]*base.TaskMessage
wantInProgress []*base.TaskMessage
@@ -156,7 +156,7 @@ func TestDequeue(t *testing.T) {
},
args: []string{"default"},
wantMsg: t1,
wantDeadline: t1Deadline,
wantDeadline: time.Unix(int64(t1Deadline), 0),
err: nil,
wantEnqueued: map[string][]*base.TaskMessage{
"default": {},
@@ -175,7 +175,7 @@ func TestDequeue(t *testing.T) {
},
args: []string{"default"},
wantMsg: nil,
wantDeadline: 0,
wantDeadline: time.Time{},
err: ErrNoProcessableTask,
wantEnqueued: map[string][]*base.TaskMessage{
"default": {},
@@ -191,7 +191,7 @@ func TestDequeue(t *testing.T) {
},
args: []string{"critical", "default", "low"},
wantMsg: t2,
wantDeadline: t2Deadline,
wantDeadline: time.Unix(int64(t2Deadline), 0),
err: nil,
wantEnqueued: map[string][]*base.TaskMessage{
"default": {t1},
@@ -214,7 +214,7 @@ func TestDequeue(t *testing.T) {
},
args: []string{"critical", "default", "low"},
wantMsg: t3,
wantDeadline: t3Deadline,
wantDeadline: time.Unix(int64(t3Deadline), 0),
err: nil,
wantEnqueued: map[string][]*base.TaskMessage{
"default": {},
@@ -237,7 +237,7 @@ func TestDequeue(t *testing.T) {
},
args: []string{"critical", "default", "low"},
wantMsg: nil,
wantDeadline: 0,
wantDeadline: time.Time{},
err: ErrNoProcessableTask,
wantEnqueued: map[string][]*base.TaskMessage{
"default": {},