@@ -33,6 +33,10 @@ func (m mockMessage) Bytes() []byte {
33
33
return []byte (m .Message )
34
34
}
35
35
36
+ func (m mockMessage ) Payload () []byte {
37
+ return []byte (m .Message )
38
+ }
39
+
36
40
func setupRedisSentinelContainer (
37
41
ctx context.Context ,
38
42
t * testing.T ,
@@ -339,7 +343,7 @@ func TestJobReachTimeout(t *testing.T) {
339
343
for {
340
344
select {
341
345
case <- ctx .Done ():
342
- log .Println ("get data:" , string (m .Bytes ()))
346
+ log .Println ("get data:" , string (m .Payload ()))
343
347
if errors .Is (ctx .Err (), context .Canceled ) {
344
348
log .Println ("queue has been shutdown and cancel the job" )
345
349
} else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
@@ -382,7 +386,7 @@ func TestCancelJobAfterShutdown(t *testing.T) {
382
386
for {
383
387
select {
384
388
case <- ctx .Done ():
385
- log .Println ("get data:" , string (m .Bytes ()))
389
+ log .Println ("get data:" , string (m .Payload ()))
386
390
if errors .Is (ctx .Err (), context .Canceled ) {
387
391
log .Println ("queue has been shutdown and cancel the job" )
388
392
} else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
@@ -425,15 +429,15 @@ func TestGoroutineLeak(t *testing.T) {
425
429
for {
426
430
select {
427
431
case <- ctx .Done ():
428
- log .Println ("get data:" , string (m .Bytes ()))
432
+ log .Println ("get data:" , string (m .Payload ()))
429
433
if errors .Is (ctx .Err (), context .Canceled ) {
430
434
log .Println ("queue has been shutdown and cancel the job" )
431
435
} else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
432
436
log .Println ("job deadline exceeded" )
433
437
}
434
438
return nil
435
439
default :
436
- log .Println ("get data:" , string (m .Bytes ()))
440
+ log .Println ("get data:" , string (m .Payload ()))
437
441
time .Sleep (50 * time .Millisecond )
438
442
return nil
439
443
}
0 commit comments