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

DDB is considered healthy in UPDATING status

Browse source
This commit is contained in:
Cai41 2020-06-05 15:45:22 -07:00 committed by Cory-Bradshaw
parent be1e3cd112
commit 87677255e1
2 changed files with 46 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
amazon-kinesis-client/src/main/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresher.java
View file

@ -190,7 +190,8 @@ public class DynamoDBLeaseRefresher implements LeaseRefresher {
*/
@Override
public boolean leaseTableExists() throws DependencyException {
return TableStatus.ACTIVE == tableStatus();
TableStatus tableStatus = tableStatus();
return TableStatus.ACTIVE == tableStatus || TableStatus.UPDATING == tableStatus;
}
private TableStatus tableStatus() throws DependencyException {

44
amazon-kinesis-client/src/test/java/software/amazon/kinesis/leases/dynamodb/DynamoDBLeaseRefresherTest.java
View file

@ -15,6 +15,8 @@
package software.amazon.kinesis.leases.dynamodb;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyBoolean;
import static org.mockito.Matchers.anyLong;
@ -54,6 +56,8 @@ import software.amazon.awssdk.services.dynamodb.model.PutItemResponse;
import software.amazon.awssdk.services.dynamodb.model.ResourceNotFoundException;
import software.amazon.awssdk.services.dynamodb.model.ScanRequest;
import software.amazon.awssdk.services.dynamodb.model.ScanResponse;
import software.amazon.awssdk.services.dynamodb.model.TableDescription;
import software.amazon.awssdk.services.dynamodb.model.TableStatus;
import software.amazon.awssdk.services.dynamodb.model.UpdateItemRequest;
import software.amazon.awssdk.services.dynamodb.model.UpdateItemResponse;
import software.amazon.awssdk.services.dynamodb.model.BillingMode;
@ -149,6 +153,46 @@ public class DynamoDBLeaseRefresherTest {
verifyCancel(mockPutItemFuture, () -> leaseRefresher.createLeaseIfNotExists(lease));
}
@Test
public void testWaitUntilLeaseTableExistsUpdatingStatus() throws Exception {
when(dynamoDbClient.describeTable(any(DescribeTableRequest.class))).thenReturn(mockDescribeTableFuture);
when(mockDescribeTableFuture.get(anyLong(), any()))
.thenReturn(DescribeTableResponse.builder()
.table(TableDescription.builder().tableStatus(TableStatus.UPDATING).build())
.build());
assertTrue(leaseRefresher.waitUntilLeaseTableExists(0, 0));
}
@Test
public void testWaitUntilLeaseTableExistsActiveStatus() throws Exception {
when(dynamoDbClient.describeTable(any(DescribeTableRequest.class))).thenReturn(mockDescribeTableFuture);
when(mockDescribeTableFuture.get(anyLong(), any()))
.thenReturn(DescribeTableResponse.builder()
.table(TableDescription.builder().tableStatus(TableStatus.ACTIVE).build())
.build());
assertTrue(leaseRefresher.waitUntilLeaseTableExists(0, 0));
}
@Test
public void testWaitUntilLeaseTableExistsCreatingStatus() throws Exception {
when(dynamoDbClient.describeTable(any(DescribeTableRequest.class))).thenReturn(mockDescribeTableFuture);
when(mockDescribeTableFuture.get(anyLong(), any()))
.thenReturn(DescribeTableResponse.builder()
.table(TableDescription.builder().tableStatus(TableStatus.CREATING).build())
.build());
assertFalse(leaseRefresher.waitUntilLeaseTableExists(0, 0));
}
@Test
public void testWaitUntilLeaseTableExistsDeletingStatus() throws Exception {
when(dynamoDbClient.describeTable(any(DescribeTableRequest.class))).thenReturn(mockDescribeTableFuture);
when(mockDescribeTableFuture.get(anyLong(), any()))
.thenReturn(DescribeTableResponse.builder()
.table(TableDescription.builder().tableStatus(TableStatus.DELETING).build())
.build());
assertFalse(leaseRefresher.waitUntilLeaseTableExists(0, 0));
}
@Test
public void testGetLeaseTimesOut() throws Exception {
TimeoutException te = setRuleForDependencyTimeout();