@@ -24,58 +24,87 @@ var (
24
24
func TestBroker_Init (t * testing.T ) {
25
25
b := & Broker {}
26
26
ok , err := b .Init (cfg )
27
+ if err != nil {
28
+ t .Fatal (err )
29
+ }
27
30
assert .True (t , ok )
28
31
assert .NoError (t , err )
29
32
}
30
33
31
34
func TestBroker_StopNotStarted (t * testing.T ) {
32
35
b := & Broker {}
33
- b .Init (cfg )
36
+ _ , err := b .Init (cfg )
37
+ if err != nil {
38
+ t .Fatal (err )
39
+ }
34
40
35
41
b .Stop ()
36
42
}
37
43
38
44
func TestBroker_Register (t * testing.T ) {
39
45
b := & Broker {}
40
- b .Init (cfg )
46
+ _ , err := b .Init (cfg )
47
+ if err != nil {
48
+ t .Fatal (err )
49
+ }
41
50
assert .NoError (t , b .Register (pipe ))
42
51
}
43
52
44
53
func TestBroker_Register_Twice (t * testing.T ) {
45
54
b := & Broker {}
46
- b .Init (cfg )
55
+ _ , err := b .Init (cfg )
56
+ if err != nil {
57
+ t .Fatal (err )
58
+ }
47
59
assert .NoError (t , b .Register (pipe ))
48
60
assert .Error (t , b .Register (pipe ))
49
61
}
50
62
51
63
func TestBroker_Consume_Nil_BeforeServe (t * testing.T ) {
52
64
b := & Broker {}
53
- b .Init (cfg )
54
- b .Register (pipe )
65
+ _ , err := b .Init (cfg )
66
+ if err != nil {
67
+ t .Fatal (err )
68
+ }
69
+ err = b .Register (pipe )
70
+ if err != nil {
71
+ t .Fatal (err )
72
+ }
55
73
assert .NoError (t , b .Consume (pipe , nil , nil ))
56
74
}
57
75
58
76
func TestBroker_Consume_Undefined (t * testing.T ) {
59
77
b := & Broker {}
60
- b .Init (cfg )
78
+ _ , err := b .Init (cfg )
79
+ if err != nil {
80
+ t .Fatal (err )
81
+ }
61
82
62
83
assert .Error (t , b .Consume (pipe , nil , nil ))
63
84
}
64
85
65
86
func TestBroker_Consume_BeforeServe (t * testing.T ) {
66
87
b := & Broker {}
67
- b .Init (cfg )
68
- b .Register (pipe )
69
-
88
+ _ , err := b .Init (cfg )
89
+ if err != nil {
90
+ t .Fatal (err )
91
+ }
92
+ err = b .Register (pipe )
93
+ if err != nil {
94
+ t .Fatal (err )
95
+ }
70
96
exec := make (chan jobs.Handler )
71
- err := func (id string , j * jobs.Job , err error ) {}
97
+ errf := func (id string , j * jobs.Job , err error ) {}
72
98
73
- assert .NoError (t , b .Consume (pipe , exec , err ))
99
+ assert .NoError (t , b .Consume (pipe , exec , errf ))
74
100
}
75
101
76
102
func TestBroker_Consume_BadPipeline (t * testing.T ) {
77
103
b := & Broker {}
78
- b .Init (cfg )
104
+ _ , err := b .Init (cfg )
105
+ if err != nil {
106
+ t .Fatal (err )
107
+ }
79
108
assert .Error (t , b .Register (& jobs.Pipeline {
80
109
"broker" : "amqp" ,
81
110
"name" : "default" ,
@@ -86,10 +115,18 @@ func TestBroker_Consume_BadPipeline(t *testing.T) {
86
115
87
116
func TestBroker_Consume_Serve_Nil_Stop (t * testing.T ) {
88
117
b := & Broker {}
89
- b .Init (cfg )
90
- b .Register (pipe )
91
-
92
- b .Consume (pipe , nil , nil )
118
+ _ , err := b .Init (cfg )
119
+ if err != nil {
120
+ t .Fatal (err )
121
+ }
122
+ err = b .Register (pipe )
123
+ if err != nil {
124
+ t .Fatal (err )
125
+ }
126
+ err = b .Consume (pipe , nil , nil )
127
+ if err != nil {
128
+ t .Fatal (err )
129
+ }
93
130
94
131
wait := make (chan interface {})
95
132
go func () {
@@ -104,22 +141,34 @@ func TestBroker_Consume_Serve_Nil_Stop(t *testing.T) {
104
141
105
142
func TestBroker_Consume_CantStart (t * testing.T ) {
106
143
b := & Broker {}
107
- b .Init (& Config {
144
+ _ , err := b .Init (& Config {
108
145
Addr : "amqp://guest:guest@localhost:15672/" ,
109
146
})
147
+ if err != nil {
148
+ t .Fatal (err )
149
+ }
110
150
111
151
assert .Error (t , b .Serve ())
112
152
}
113
153
114
154
func TestBroker_Consume_Serve_Stop (t * testing.T ) {
115
155
b := & Broker {}
116
- b .Init (cfg )
117
- b .Register (pipe )
156
+ _ , err := b .Init (cfg )
157
+ if err != nil {
158
+ t .Fatal (err )
159
+ }
160
+ err = b .Register (pipe )
161
+ if err != nil {
162
+ t .Fatal (err )
163
+ }
118
164
119
165
exec := make (chan jobs.Handler )
120
- err := func (id string , j * jobs.Job , err error ) {}
166
+ errf := func (id string , j * jobs.Job , err error ) {}
121
167
122
- b .Consume (pipe , exec , err )
168
+ err = b .Consume (pipe , exec , errf )
169
+ if err != nil {
170
+ t .Fatal ()
171
+ }
123
172
124
173
wait := make (chan interface {})
125
174
go func () {
@@ -134,25 +183,40 @@ func TestBroker_Consume_Serve_Stop(t *testing.T) {
134
183
135
184
func TestBroker_PushToNotRunning (t * testing.T ) {
136
185
b := & Broker {}
137
- b .Init (cfg )
138
- b .Register (pipe )
186
+ _ , err := b .Init (cfg )
187
+ if err != nil {
188
+ t .Fatal (err )
189
+ }
190
+ err = b .Register (pipe )
191
+ if err != nil {
192
+ t .Fatal (err )
193
+ }
139
194
140
- _ , err : = b .Push (pipe , & jobs.Job {})
195
+ _ , err = b .Push (pipe , & jobs.Job {})
141
196
assert .Error (t , err )
142
197
}
143
198
144
199
func TestBroker_StatNotRunning (t * testing.T ) {
145
200
b := & Broker {}
146
- b .Init (cfg )
147
- b .Register (pipe )
201
+ _ , err := b .Init (cfg )
202
+ if err != nil {
203
+ t .Fatal (err )
204
+ }
205
+ err = b .Register (pipe )
206
+ if err != nil {
207
+ t .Fatal (err )
208
+ }
148
209
149
- _ , err : = b .Stat (pipe )
210
+ _ , err = b .Stat (pipe )
150
211
assert .Error (t , err )
151
212
}
152
213
153
214
func TestBroker_PushToNotRegistered (t * testing.T ) {
154
215
b := & Broker {}
155
- b .Init (cfg )
216
+ _ , err := b .Init (cfg )
217
+ if err != nil {
218
+ t .Fatal (err )
219
+ }
156
220
157
221
ready := make (chan interface {})
158
222
b .Listen (func (event int , ctx interface {}) {
@@ -166,13 +230,16 @@ func TestBroker_PushToNotRegistered(t *testing.T) {
166
230
167
231
<- ready
168
232
169
- _ , err : = b .Push (pipe , & jobs.Job {})
233
+ _ , err = b .Push (pipe , & jobs.Job {})
170
234
assert .Error (t , err )
171
235
}
172
236
173
237
func TestBroker_StatNotRegistered (t * testing.T ) {
174
238
b := & Broker {}
175
- b .Init (cfg )
239
+ _ , err := b .Init (cfg )
240
+ if err != nil {
241
+ t .Fatal (err )
242
+ }
176
243
177
244
ready := make (chan interface {})
178
245
b .Listen (func (event int , ctx interface {}) {
@@ -186,6 +253,6 @@ func TestBroker_StatNotRegistered(t *testing.T) {
186
253
187
254
<- ready
188
255
189
- _ , err : = b .Stat (pipe )
256
+ _ , err = b .Stat (pipe )
190
257
assert .Error (t , err )
191
258
}
0 commit comments