Skip to content

Commit b64c331

Browse files
committedJan 20, 2025··
refactor: refactor logging to use Payload method in mockMessage struct
- Add `Payload` method to `mockMessage` struct - Replace `m.Bytes()` with `m.Payload()` in log statements in multiple tests Signed-off-by: Bo-Yi Wu <[email protected]>
1 parent ab65c53 commit b64c331

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed
 

‎redis_test.go

+8-4
Original file line numberDiff line numberDiff line change
@@ -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+
3640
func 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

Comments
 (0)