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

Revert changes made to ShardSyncTaskIntegrationTest

Browse source
This commit is contained in:
Gayan Weerakutti 2021-02-01 16:00:43 +05:30
parent ea54fdd6b9
commit d1bc673a96
No known key found for this signature in database
GPG key ID: 71DB6958626BE6CD
1 changed files with 5 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/ShardSyncTaskIntegrationTest.java
View file

@ -18,15 +18,9 @@ import java.util.HashSet;
import java.util.List;
import java.util.Set;
import com.amazonaws.auth.AWSStaticCredentialsProvider;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.client.builder.AwsClientBuilder;
import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClientBuilder;
import com.amazonaws.services.dynamodbv2.model.BillingMode;
import com.amazonaws.services.dynamodbv2.model.DescribeTableResult;
import com.amazonaws.services.dynamodbv2.model.ListTablesResult;
import com.amazonaws.services.kinesis.AmazonKinesisClientBuilder;
import com.amazonaws.services.kinesis.model.CreateStreamRequest;
import org.joda.time.DateTime;
import org.junit.AfterClass;
import org.junit.Assert;
@ -57,7 +51,7 @@ import static junit.framework.TestCase.fail;
public class ShardSyncTaskIntegrationTest {
private static final String STREAM_NAME = "IntegrationTestStream02";
private static final String KINESIS_ENDPOINT = "http://localhost:4566";
private static final String KINESIS_ENDPOINT = "https://kinesis.us-east-1.amazonaws.com";
private static AWSCredentialsProvider credentialsProvider;
private IKinesisClientLeaseManager leaseManager;
@ -69,13 +63,8 @@ public class ShardSyncTaskIntegrationTest {
*/
@BeforeClass
public static void setUpBeforeClass() throws Exception {
System.setProperty("com.amazonaws.sdk.disableCbor", "true");
credentialsProvider = new DefaultAWSCredentialsProviderChain();
AmazonKinesis kinesis = AmazonKinesisClientBuilder.standard()
.withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration("http://localhost:4566","us-east-1"))
.withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials("accessKey", "secretKey")))
.build();
AmazonKinesis kinesis = new AmazonKinesisClient(credentialsProvider);
try {
kinesis.createStream(STREAM_NAME, 1);
@ -103,10 +92,7 @@ public class ShardSyncTaskIntegrationTest {
boolean useConsistentReads = true;
leaseManager =
new KinesisClientLeaseManager(tableName,
AmazonDynamoDBClientBuilder.standard()
.withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration("http://localhost:4566", "us-east-1"))
.withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials("accessKey", "secretKey")))
.build(),
new AmazonDynamoDBClient(credentialsProvider),
useConsistentReads,
billingMode);
@ -154,10 +140,7 @@ public class ShardSyncTaskIntegrationTest {
}
private void cleanUpTable(String tableName) throws DependencyException {
AmazonDynamoDBClient client = (AmazonDynamoDBClient) AmazonDynamoDBClientBuilder.standard()
.withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration("http://localhost:4566","us-east-1"))
.withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials("accessKey", "secretKey")))
.build();
AmazonDynamoDBClient client = new AmazonDynamoDBClient(DefaultAWSCredentialsProviderChain.getInstance());
ListTablesResult tables = client.listTables();
if(tables.getTableNames().contains(tableName)){
leaseManager.waitUntilLeaseTableExists(2,20);
@ -177,10 +160,7 @@ public class ShardSyncTaskIntegrationTest {
}
private void checkBillingMode(BillingMode billingMode, String tableName) {
AmazonDynamoDBClient client = (AmazonDynamoDBClient) AmazonDynamoDBClientBuilder.standard()
.withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration("http://localhost:4566","us-east-1"))
.withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials("accessKey", "secretKey")))
.build();
AmazonDynamoDBClient client = new AmazonDynamoDBClient(DefaultAWSCredentialsProviderChain.getInstance());
DescribeTableResult tableDetails = client.describeTable(tableName);
if(BillingMode.PAY_PER_REQUEST.equals(billingMode)) {
Assert.assertEquals(tableDetails.getTable().getBillingModeSummary().getBillingMode(), billingMode.name());