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

bug test

Browse source
This commit is contained in:
Joshua Kim 2021-04-29 11:25:11 -07:00
parent f38dd18ed1
commit c5bb0d6bcd
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/test/java/com/amazonaws/services/kinesis/clientlibrary/proxies/KinesisProxyTest.java
View file

@ -483,6 +483,26 @@ public class KinesisProxyTest {
verify(mockClient).listShards(any()); verify(mockClient).listShards(any());
} }
@Test
public void testNoDuplicateShardsInPartialFailure() {
proxy.setCachedShardMap(null);
ListShardsResult firstPage = new ListShardsResult().withShards(shards.subList(0, 2)).withNextToken(NEXT_TOKEN);
ListShardsResult lastPage = new ListShardsResult().withShards(shards.subList(2, shards.size())).withNextToken(null);
when(mockClient.listShards(any()))
.thenReturn(firstPage)
.thenThrow(new RuntimeException("Failed!"))
.thenReturn(firstPage)
.thenReturn(lastPage);
try {
proxy.getShardList();
} catch (Exception e) {
}
assertEquals(shards, proxy.getShardList());
}
private void mockListShardsForSingleResponse(List<Shard> shards) { private void mockListShardsForSingleResponse(List<Shard> shards) {
when(mockClient.listShards(any())).thenReturn(listShardsResult); when(mockClient.listShards(any())).thenReturn(listShardsResult);
when(listShardsResult.getShards()).thenReturn(shards); when(listShardsResult.getShards()).thenReturn(shards);