Skip to content
This repository was archived by the owner on Jan 15, 2022. It is now read-only.

Commit 4a4d8cd

Browse files
committed
Update consume_delayed test
1 parent 084fbab commit 4a4d8cd

File tree

1 file changed

+21
-6
lines changed

1 file changed

+21
-6
lines changed

broker/amqp/consume_test.go

+21-6
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,10 @@ func TestBroker_Consume_Job(t *testing.T) {
1414
if err != nil {
1515
t.Fatal(err)
1616
}
17-
b.Register(pipe)
17+
err = b.Register(pipe)
18+
if err != nil {
19+
t.Fatal(err)
20+
}
1821

1922
ready := make(chan interface{})
2023
b.Listen(func(event int, ctx interface{}) {
@@ -57,7 +60,10 @@ func TestBroker_ConsumeAfterStart_Job(t *testing.T) {
5760
if err != nil {
5861
t.Fatal(err)
5962
}
60-
b.Register(pipe)
63+
err = b.Register(pipe)
64+
if err != nil {
65+
t.Fatal(err)
66+
}
6167

6268
ready := make(chan interface{})
6369
b.Listen(func(event int, ctx interface{}) {
@@ -101,7 +107,10 @@ func TestBroker_Consume_Delayed(t *testing.T) {
101107
t.Fatal(err)
102108
}
103109

104-
b.Register(pipe)
110+
err = b.Register(pipe)
111+
if err != nil {
112+
t.Fatal(err)
113+
}
105114

106115
ready := make(chan interface{})
107116
b.Listen(func(event int, ctx interface{}) {
@@ -140,7 +149,7 @@ func TestBroker_Consume_Delayed(t *testing.T) {
140149

141150
elapsed := time.Since(start)
142151
assert.True(t, elapsed >= time.Second)
143-
assert.True(t, elapsed < 2*time.Second)
152+
assert.True(t, elapsed < 3*time.Second)
144153
}
145154

146155
func TestBroker_Consume_Errored(t *testing.T) {
@@ -149,7 +158,10 @@ func TestBroker_Consume_Errored(t *testing.T) {
149158
if err != nil {
150159
t.Fatal(err)
151160
}
152-
b.Register(pipe)
161+
err = b.Register(pipe)
162+
if err != nil {
163+
t.Fatal(err)
164+
}
153165

154166
ready := make(chan interface{})
155167
b.Listen(func(event int, ctx interface{}) {
@@ -195,7 +207,10 @@ func TestBroker_Consume_Errored_Attempts(t *testing.T) {
195207
if err != nil {
196208
t.Fatal(err)
197209
}
198-
b.Register(pipe)
210+
err = b.Register(pipe)
211+
if err != nil {
212+
t.Fatal(err)
213+
}
199214

200215
ready := make(chan interface{})
201216
b.Listen(func(event int, ctx interface{}) {

0 commit comments

Comments
 (0)