From 98f2cdaf5a4a9e28a5425f1671f00961ef4c9e60 Mon Sep 17 00:00:00 2001 From: Chris Eager Date: Mon, 1 Jul 2024 17:35:23 -0500 Subject: [PATCH] Reduce DynamoDB test client API timeouts --- .../textsecuregcm/storage/DynamoDbExtension.java | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/storage/DynamoDbExtension.java b/service/src/test/java/org/whispersystems/textsecuregcm/storage/DynamoDbExtension.java index 4f2c71706..eb658ec96 100644 --- a/service/src/test/java/org/whispersystems/textsecuregcm/storage/DynamoDbExtension.java +++ b/service/src/test/java/org/whispersystems/textsecuregcm/storage/DynamoDbExtension.java @@ -10,6 +10,7 @@ import com.amazonaws.services.dynamodbv2.local.main.ServerRunner; import com.amazonaws.services.dynamodbv2.local.server.DynamoDBProxyServer; import java.net.ServerSocket; import java.net.URI; +import java.time.Duration; import java.util.List; import java.util.Optional; import java.util.concurrent.atomic.AtomicBoolean; @@ -161,12 +162,18 @@ public class DynamoDbExtension implements BeforeEachCallback, AfterEachCallback .region(Region.of("local-test-region")) .credentialsProvider(StaticCredentialsProvider.create( AwsBasicCredentials.create("accessKey", "secretKey"))) + .overrideConfiguration(builder -> + builder.apiCallTimeout(Duration.ofSeconds(1)) + .apiCallAttemptTimeout(Duration.ofSeconds(1))) .build(); dynamoAsyncDB2 = DynamoDbAsyncClient.builder() .endpointOverride(URI.create("http://localhost:" + port)) .region(Region.of("local-test-region")) .credentialsProvider(StaticCredentialsProvider.create( AwsBasicCredentials.create("accessKey", "secretKey"))) + .overrideConfiguration(builder -> + builder.apiCallTimeout(Duration.ofSeconds(1)) + .apiCallAttemptTimeout(Duration.ofSeconds(1))) .build(); }