luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adding unit test changes

Browse source
This commit is contained in:
Chunxue Yang 2020-03-02 09:48:00 -08:00
parent 5760a27962
commit e95344b13e
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskManagerTest.java
View file

@ -23,6 +23,8 @@ import java.util.concurrent.Executors;
import static org.mockito.AdditionalAnswers.delegatesTo; import static org.mockito.AdditionalAnswers.delegatesTo;
import static org.mockito.Matchers.anyBoolean; import static org.mockito.Matchers.anyBoolean;
import static org.mockito.Matchers.matches;
import static org.mockito.Matchers.notNull;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times; import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
@ -91,10 +93,14 @@ public class ShardSyncTaskManagerTest {
pausableNoOpShardSyncer.blockShardSyncLatch.countDown(); pausableNoOpShardSyncer.blockShardSyncLatch.countDown();
// Wait for ShardSyncer to be initialized. // Wait for ShardSyncer to be initialized.
pausableNoOpShardSyncer.waitForShardSyncerInitializationLatch.await(); pausableNoOpShardSyncer.waitForShardSyncerInitializationLatch.await();
// There should be 1 more shardSyncer invocation after the previous shardSync completes. // There should be totally 2 invocation of shardSyncer. The first one should be triggered with an empty list the latestShards.
verify(mockShardSyncer, times(2)) // The second invocation should be the pending shard sync task, which should have null as the latestShards.
verify(mockShardSyncer, times(1))
.checkAndCreateLeasesForNewShards(Matchers.any(), Matchers.any(), Matchers.any(), anyBoolean(), .checkAndCreateLeasesForNewShards(Matchers.any(), Matchers.any(), Matchers.any(), anyBoolean(),
anyBoolean(), Matchers.any()); anyBoolean(), Matchers.eq(new ArrayList<>()));
verify(mockShardSyncer, times(1))
.checkAndCreateLeasesForNewShards(Matchers.any(), Matchers.any(), Matchers.any(), anyBoolean(),
anyBoolean(), Matchers.eq(null));
} }
private static class PausableNoOpShardSyncer implements ShardSyncer { private static class PausableNoOpShardSyncer implements ShardSyncer {