Examples with AmazonKinesis com.amazonaws.services.kinesis.AmazonKinesis used on opensource projects

Example 11 with AmazonKinesis

use of com.amazonaws.services.kinesis.AmazonKinesis in project druid by druid-io.

the class KinesisRecordSupplierTest method testIsClosedShardEmpty.

@Test
public void testIsClosedShardEmpty() {
    AmazonKinesis mockKinesis = EasyMock.mock(AmazonKinesis.class);
    KinesisRecordSupplier target = new KinesisRecordSupplier(mockKinesis, recordsPerFetch, 0, 2, false, 100, 5000, 5000, 60000, 5, true);
    Record record = new Record();
    final String shardWithoutRecordsAndNullNextIterator = "0";
    setupMockKinesisForShardId(mockKinesis, shardWithoutRecordsAndNullNextIterator, new ArrayList<>(), null);
    final String shardWithRecordsAndNullNextIterator = "1";
    setupMockKinesisForShardId(mockKinesis, shardWithRecordsAndNullNextIterator, Collections.singletonList(record), null);
    final String shardWithoutRecordsAndNonNullNextIterator = "2";
    setupMockKinesisForShardId(mockKinesis, shardWithoutRecordsAndNonNullNextIterator, new ArrayList<>(), "nextIterator");
    final String shardWithRecordsAndNonNullNextIterator = "3";
    setupMockKinesisForShardId(mockKinesis, shardWithRecordsAndNonNullNextIterator, Collections.singletonList(record), "nextIterator");
    EasyMock.replay(mockKinesis);
    // A closed shard is empty only when the records are empty and the next iterator is null
    Assert.assertTrue(target.isClosedShardEmpty(STREAM, shardWithoutRecordsAndNullNextIterator));
    Assert.assertFalse(target.isClosedShardEmpty(STREAM, shardWithRecordsAndNullNextIterator));
    Assert.assertFalse(target.isClosedShardEmpty(STREAM, shardWithoutRecordsAndNonNullNextIterator));
    Assert.assertFalse(target.isClosedShardEmpty(STREAM, shardWithRecordsAndNonNullNextIterator));
}