updating new unit test to conform to ShardConsumer constructor change

This commit is contained in:
Shitanshu Aggarwal 2019-10-30 02:51:31 +00:00
parent 52fe45e8c3
commit 39ce21e4b0

View file

@ -470,15 +470,17 @@ public class ShardConsumerTest {
final KinesisClientLease parentLease = mock(KinesisClientLease.class);
when(shardInfo.getShardId()).thenReturn(shardId);
when(shardInfo.getParentShardIds()).thenReturn(Arrays.asList(parentShardId));
when(leaseCoordinator.getLeaseManager()).thenReturn(leaseManager);
when(leaseManager.getLease(eq(parentShardId))).thenReturn(parentLease);
when(parentLease.getCheckpoint()).thenReturn(ExtendedSequenceNumber.TRIM_HORIZON);
final ShardConsumer consumer = new ShardConsumer(shardInfo,
final ShardConsumer consumer =
new ShardConsumer(shardInfo,
streamConfig,
checkpoint,
processor,
recordProcessorCheckpointer,
leaseManager,
leaseCoordinator,
parentShardPollIntervalMillis,
cleanupLeasesOfCompletedShards,
executorService,