@@ -120,14 +120,12 @@ func (s *queueBaseSuite) TestNewProcessBase_NoPreviousState() {
120
120
121
121
mockShard := shard .NewTestContext (
122
122
s .controller ,
123
- & persistence.ShardInfoWithFailover {
124
- ShardInfo : & persistencespb.ShardInfo {
125
- ShardId : 0 ,
126
- RangeId : rangeID ,
127
- QueueAckLevels : map [int32 ]* persistencespb.QueueAckLevel {
128
- tasks .CategoryIDTransfer : {
129
- AckLevel : ackLevel ,
130
- },
123
+ & persistencespb.ShardInfo {
124
+ ShardId : 0 ,
125
+ RangeId : rangeID ,
126
+ QueueAckLevels : map [int32 ]* persistencespb.QueueAckLevel {
127
+ tasks .CategoryIDTransfer : {
128
+ AckLevel : ackLevel ,
131
129
},
132
130
},
133
131
},
@@ -207,13 +205,11 @@ func (s *queueBaseSuite) TestNewProcessBase_WithPreviousState() {
207
205
208
206
mockShard := shard .NewTestContext (
209
207
s .controller ,
210
- & persistence.ShardInfoWithFailover {
211
- ShardInfo : & persistencespb.ShardInfo {
212
- ShardId : 0 ,
213
- RangeId : 10 ,
214
- QueueStates : map [int32 ]* persistencespb.QueueState {
215
- tasks .CategoryIDTransfer : persistenceState ,
216
- },
208
+ & persistencespb.ShardInfo {
209
+ ShardId : 0 ,
210
+ RangeId : 10 ,
211
+ QueueStates : map [int32 ]* persistencespb.QueueState {
212
+ tasks .CategoryIDTransfer : persistenceState ,
217
213
},
218
214
},
219
215
s .config ,
@@ -248,14 +244,12 @@ func (s *queueBaseSuite) TestNewProcessBase_WithPreviousState() {
248
244
func (s * queueBaseSuite ) TestStartStop () {
249
245
mockShard := shard .NewTestContext (
250
246
s .controller ,
251
- & persistence.ShardInfoWithFailover {
252
- ShardInfo : & persistencespb.ShardInfo {
253
- ShardId : 0 ,
254
- RangeId : 10 ,
255
- QueueAckLevels : map [int32 ]* persistencespb.QueueAckLevel {
256
- tasks .CategoryIDTransfer : {
257
- AckLevel : 1024 ,
258
- },
247
+ & persistencespb.ShardInfo {
248
+ ShardId : 0 ,
249
+ RangeId : 10 ,
250
+ QueueAckLevels : map [int32 ]* persistencespb.QueueAckLevel {
251
+ tasks .CategoryIDTransfer : {
252
+ AckLevel : 1024 ,
259
253
},
260
254
},
261
255
},
@@ -318,13 +312,11 @@ func (s *queueBaseSuite) TestProcessNewRange() {
318
312
319
313
mockShard := shard .NewTestContext (
320
314
s .controller ,
321
- & persistence.ShardInfoWithFailover {
322
- ShardInfo : & persistencespb.ShardInfo {
323
- ShardId : 0 ,
324
- RangeId : 10 ,
325
- QueueStates : map [int32 ]* persistencespb.QueueState {
326
- tasks .CategoryIDTimer : persistenceState ,
327
- },
315
+ & persistencespb.ShardInfo {
316
+ ShardId : 0 ,
317
+ RangeId : 10 ,
318
+ QueueStates : map [int32 ]* persistencespb.QueueState {
319
+ tasks .CategoryIDTimer : persistenceState ,
328
320
},
329
321
},
330
322
s .config ,
@@ -375,13 +367,11 @@ func (s *queueBaseSuite) TestCheckPoint_WithPendingTasks() {
375
367
376
368
mockShard := shard .NewTestContext (
377
369
s .controller ,
378
- & persistence.ShardInfoWithFailover {
379
- ShardInfo : & persistencespb.ShardInfo {
380
- ShardId : 0 ,
381
- RangeId : 10 ,
382
- QueueStates : map [int32 ]* persistencespb.QueueState {
383
- tasks .CategoryIDTimer : persistenceState ,
384
- },
370
+ & persistencespb.ShardInfo {
371
+ ShardId : 0 ,
372
+ RangeId : 10 ,
373
+ QueueStates : map [int32 ]* persistencespb.QueueState {
374
+ tasks .CategoryIDTimer : persistenceState ,
385
375
},
386
376
},
387
377
s .config ,
@@ -449,13 +439,11 @@ func (s *queueBaseSuite) TestCheckPoint_NoPendingTasks() {
449
439
450
440
mockShard := shard .NewTestContext (
451
441
s .controller ,
452
- & persistence.ShardInfoWithFailover {
453
- ShardInfo : & persistencespb.ShardInfo {
454
- ShardId : 0 ,
455
- RangeId : 10 ,
456
- QueueStates : map [int32 ]* persistencespb.QueueState {
457
- tasks .CategoryIDTimer : persistenceState ,
458
- },
442
+ & persistencespb.ShardInfo {
443
+ ShardId : 0 ,
444
+ RangeId : 10 ,
445
+ QueueStates : map [int32 ]* persistencespb.QueueState {
446
+ tasks .CategoryIDTimer : persistenceState ,
459
447
},
460
448
},
461
449
s .config ,
@@ -538,13 +526,11 @@ func (s *queueBaseSuite) TestCheckPoint_MoveSlices() {
538
526
539
527
mockShard := shard .NewTestContext (
540
528
s .controller ,
541
- & persistence.ShardInfoWithFailover {
542
- ShardInfo : & persistencespb.ShardInfo {
543
- ShardId : 0 ,
544
- RangeId : 10 ,
545
- QueueStates : map [int32 ]* persistencespb.QueueState {
546
- tasks .CategoryIDTimer : initialPersistenceState ,
547
- },
529
+ & persistencespb.ShardInfo {
530
+ ShardId : 0 ,
531
+ RangeId : 10 ,
532
+ QueueStates : map [int32 ]* persistencespb.QueueState {
533
+ tasks .CategoryIDTimer : initialPersistenceState ,
548
534
},
549
535
},
550
536
s .config ,
0 commit comments