@@ -33,6 +33,10 @@ func (m mockMessage) Bytes() []byte {
3333 return []byte (m .Message )
3434}
3535
36+ func (m mockMessage ) Payload () []byte {
37+ return []byte (m .Message )
38+ }
39+
3640func setupRedisSentinelContainer (
3741 ctx context.Context ,
3842 t * testing.T ,
@@ -339,7 +343,7 @@ func TestJobReachTimeout(t *testing.T) {
339343 for {
340344 select {
341345 case <- ctx .Done ():
342- log .Println ("get data:" , string (m .Bytes ()))
346+ log .Println ("get data:" , string (m .Payload ()))
343347 if errors .Is (ctx .Err (), context .Canceled ) {
344348 log .Println ("queue has been shutdown and cancel the job" )
345349 } else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
@@ -382,7 +386,7 @@ func TestCancelJobAfterShutdown(t *testing.T) {
382386 for {
383387 select {
384388 case <- ctx .Done ():
385- log .Println ("get data:" , string (m .Bytes ()))
389+ log .Println ("get data:" , string (m .Payload ()))
386390 if errors .Is (ctx .Err (), context .Canceled ) {
387391 log .Println ("queue has been shutdown and cancel the job" )
388392 } else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
@@ -425,15 +429,15 @@ func TestGoroutineLeak(t *testing.T) {
425429 for {
426430 select {
427431 case <- ctx .Done ():
428- log .Println ("get data:" , string (m .Bytes ()))
432+ log .Println ("get data:" , string (m .Payload ()))
429433 if errors .Is (ctx .Err (), context .Canceled ) {
430434 log .Println ("queue has been shutdown and cancel the job" )
431435 } else if errors .Is (ctx .Err (), context .DeadlineExceeded ) {
432436 log .Println ("job deadline exceeded" )
433437 }
434438 return nil
435439 default :
436- log .Println ("get data:" , string (m .Bytes ()))
440+ log .Println ("get data:" , string (m .Payload ()))
437441 time .Sleep (50 * time .Millisecond )
438442 return nil
439443 }
0 commit comments