@@ -34,6 +34,7 @@ import (
34
34
"go.uber.org/mock/gomock"
35
35
36
36
"github.com/uber/cadence/common"
37
+ "github.com/uber/cadence/common/log/testlogger"
37
38
"github.com/uber/cadence/common/persistence"
38
39
"github.com/uber/cadence/common/types"
39
40
)
@@ -67,7 +68,7 @@ func TestReplicationQueueImpl_Start(t *testing.T) {
67
68
t .Run (tt .name , func (t * testing.T ) {
68
69
ctrl := gomock .NewController (t )
69
70
mockQueue := persistence .NewMockQueueManager (ctrl )
70
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil ).(* replicationQueueImpl )
71
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) ).(* replicationQueueImpl )
71
72
atomic .StoreInt32 (& rq .status , tt .initialStatus )
72
73
73
74
rq .Start ()
@@ -111,7 +112,7 @@ func TestReplicationQueueImpl_Stop(t *testing.T) {
111
112
t .Run (tt .name , func (t * testing.T ) {
112
113
ctrl := gomock .NewController (t )
113
114
mockQueue := persistence .NewMockQueueManager (ctrl )
114
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil ).(* replicationQueueImpl )
115
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) ).(* replicationQueueImpl )
115
116
atomic .StoreInt32 (& rq .status , tt .initialStatus )
116
117
117
118
rq .Stop ()
@@ -158,7 +159,7 @@ func TestReplicationQueueImpl_Publish(t *testing.T) {
158
159
t .Run (tt .name , func (t * testing.T ) {
159
160
ctrl := gomock .NewController (t )
160
161
mockQueue := persistence .NewMockQueueManager (ctrl )
161
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
162
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
162
163
tt .setupMock (mockQueue )
163
164
err := rq .Publish (context .Background (), tt .task )
164
165
if tt .wantErr {
@@ -199,7 +200,7 @@ func TestReplicationQueueImpl_PublishToDLQ(t *testing.T) {
199
200
t .Run (tt .name , func (t * testing.T ) {
200
201
ctrl := gomock .NewController (t )
201
202
mockQueue := persistence .NewMockQueueManager (ctrl )
202
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
203
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
203
204
tt .setupMock (mockQueue )
204
205
err := rq .PublishToDLQ (context .Background (), tt .task )
205
206
if tt .wantErr {
@@ -275,7 +276,7 @@ func TestGetReplicationMessages(t *testing.T) {
275
276
ctrl := gomock .NewController (t )
276
277
mockQueueManager := persistence .NewMockQueueManager (ctrl )
277
278
tc .setupMocks (mockQueueManager )
278
- replicationQueue := NewReplicationQueue (mockQueueManager , "testCluster" , nil , nil )
279
+ replicationQueue := NewReplicationQueue (mockQueueManager , "testCluster" , nil , testlogger . New ( t ) )
279
280
tasks , lastID , err := replicationQueue .GetReplicationMessages (context .Background (), 0 , 10 )
280
281
281
282
if tc .expectError {
@@ -322,7 +323,7 @@ func TestUpdateAckLevel(t *testing.T) {
322
323
ctrl := gomock .NewController (t )
323
324
mockQueue := persistence .NewMockQueueManager (ctrl )
324
325
325
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
326
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
326
327
tt .setupMock (mockQueue )
327
328
err := rq .UpdateAckLevel (context .Background (), tt .lastID , tt .cluster )
328
329
if tt .wantErr {
@@ -362,7 +363,7 @@ func TestReplicationQueueImpl_GetAckLevels(t *testing.T) {
362
363
t .Run (tt .name , func (t * testing.T ) {
363
364
ctrl := gomock .NewController (t )
364
365
mockQueue := persistence .NewMockQueueManager (ctrl )
365
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
366
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
366
367
tt .setupMock (mockQueue )
367
368
got , err := rq .GetAckLevels (context .Background ())
368
369
if tt .wantErr {
@@ -415,7 +416,7 @@ func TestGetMessagesFromDLQ(t *testing.T) {
415
416
t .Run (tt .name , func (t * testing.T ) {
416
417
ctrl := gomock .NewController (t )
417
418
mockQueue := persistence .NewMockQueueManager (ctrl )
418
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
419
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
419
420
420
421
if ! tt .wantErr {
421
422
encodedData , _ := mockEncodeReplicationTask (tt .taskID )
@@ -468,7 +469,7 @@ func TestUpdateDLQAckLevel(t *testing.T) {
468
469
t .Run (tt .name , func (t * testing.T ) {
469
470
ctrl := gomock .NewController (t )
470
471
mockQueue := persistence .NewMockQueueManager (ctrl )
471
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
472
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
472
473
tt .setupMock (mockQueue )
473
474
err := rq .UpdateDLQAckLevel (context .Background (), tt .lastID )
474
475
if tt .wantErr {
@@ -508,7 +509,7 @@ func TestGetDLQAckLevel(t *testing.T) {
508
509
t .Run (tt .name , func (t * testing.T ) {
509
510
ctrl := gomock .NewController (t )
510
511
mockQueue := persistence .NewMockQueueManager (ctrl )
511
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
512
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
512
513
tt .setupMock (mockQueue )
513
514
got , err := rq .GetDLQAckLevel (context .Background ())
514
515
if tt .wantErr {
@@ -553,7 +554,7 @@ func TestRangeDeleteMessagesFromDLQ(t *testing.T) {
553
554
t .Run (tt .name , func (t * testing.T ) {
554
555
ctrl := gomock .NewController (t )
555
556
mockQueue := persistence .NewMockQueueManager (ctrl )
556
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
557
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
557
558
tt .setupMock (mockQueue )
558
559
err := rq .RangeDeleteMessagesFromDLQ (context .Background (), tt .firstID , tt .lastID )
559
560
if tt .wantErr {
@@ -594,7 +595,7 @@ func TestDeleteMessageFromDLQ(t *testing.T) {
594
595
t .Run (tt .name , func (t * testing.T ) {
595
596
ctrl := gomock .NewController (t )
596
597
mockQueue := persistence .NewMockQueueManager (ctrl )
597
- rq := NewReplicationQueue (mockQueue , "testCluster" , nil , nil )
598
+ rq := NewReplicationQueue (mockQueue , "testCluster" , nil , testlogger . New ( t ) )
598
599
tt .setupMock (mockQueue )
599
600
err := rq .DeleteMessageFromDLQ (context .Background (), tt .messageID )
600
601
if tt .wantErr {
@@ -643,7 +644,7 @@ func TestGetDLQSize(t *testing.T) {
643
644
ctrl := gomock .NewController (t )
644
645
mockQueueManager := persistence .NewMockQueueManager (ctrl )
645
646
tt .setupMock (mockQueueManager )
646
- q := & replicationQueueImpl {queue : mockQueueManager }
647
+ q := & replicationQueueImpl {queue : mockQueueManager , logger : testlogger . New ( t ) }
647
648
size , err := q .GetDLQSize (context .Background ())
648
649
if tt .wantErr {
649
650
assert .Error (t , err )
@@ -698,7 +699,7 @@ func TestPurgeAckedMessages(t *testing.T) {
698
699
ctrl := gomock .NewController (t )
699
700
mockQueueManager := persistence .NewMockQueueManager (ctrl )
700
701
tt .setupMock (mockQueueManager )
701
- q := & replicationQueueImpl {queue : mockQueueManager }
702
+ q := & replicationQueueImpl {queue : mockQueueManager , logger : testlogger . New ( t ) }
702
703
err := q .purgeAckedMessages ()
703
704
704
705
if tt .wantErr {
0 commit comments