Testing stability changes. This is a partial commit from #6348. (#6493)

Taken from 442cd70

Signed-off-by: David Venable <dlv@amazon.com>
Co-authored-by: Santhosh Gandhe <1909520+san81@users.noreply.github.com>

Author: dlvenable

data-prepper-plugins/dynamodb-source-coordination-store/build.gradle

@@ -17,6 +17,7 @@ dependencies {
     implementation 'software.amazon.awssdk:sts'
     implementation 'javax.inject:javax.inject:1'
     testImplementation 'com.amazonaws:DynamoDBLocal:2.2.1'
+    testImplementation 'org.awaitility:awaitility:4.2.0'
 }
 
 configurations {

data-prepper-plugins/dynamodb-source-coordination-store/src/test/java/org/opensearch/dataprepper/plugins/sourcecoordinator/dynamodb/DynamoDbSourceCoordinationStoreIT.java

@@ -36,7 +36,9 @@
 import java.util.Optional;
 import java.util.Random;
 import java.util.UUID;
+import java.util.concurrent.TimeUnit;
 
+import static org.awaitility.Awaitility.await;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.equalTo;
 import static org.hamcrest.Matchers.greaterThanOrEqualTo;
@@ -265,8 +267,16 @@ void tryAcquireAvailablePartition_gets_first_unassigned_partition() {
         objectUnderTest.tryCreatePartitionItem(sourceIdentifier,
                 unassignedPartitionKey3, SourcePartitionStatus.UNASSIGNED, 1L, partitionProgressState, false);
 
-        final Optional<SourcePartitionStoreItem> maybeAcquired =
-                objectUnderTest.tryAcquireAvailablePartition(sourceIdentifier, ownerId, Duration.ofSeconds(20));
+        // Wait for partition to be available in DynamoDB Local before attempting to acquire
+        final Optional<SourcePartitionStoreItem>[] maybeAcquiredHolder = new Optional[]{Optional.empty()};
+        await().atMost(5, TimeUnit.SECONDS)
+                .pollInterval(100, TimeUnit.MILLISECONDS)
+                .untilAsserted(() -> {
+                    maybeAcquiredHolder[0] = objectUnderTest.tryAcquireAvailablePartition(sourceIdentifier, ownerId, Duration.ofSeconds(20));
+                    assertThat(maybeAcquiredHolder[0].isPresent(), equalTo(true));
+                });
+
+        final Optional<SourcePartitionStoreItem> maybeAcquired = maybeAcquiredHolder[0];
 
         assertThat(maybeAcquired, notNullValue());
         assertThat(maybeAcquired.isPresent(), equalTo(true));

data-prepper-plugins/saas-source-plugins/confluence-source/src/test/java/org/opensearch/dataprepper/plugins/source/confluence/ConfluenceConfigHelperTest.java

@@ -118,7 +118,7 @@ void testValidateConfig() {
 
     @Test
     void testValidateConfigBasic() {
-        when(confluenceSourceConfig.getAccountUrl()).thenReturn("https://test.com");
+        when(confluenceSourceConfig.getAccountUrl()).thenReturn("https://somedomain.atlassian.net");
         when(confluenceSourceConfig.getAuthType()).thenReturn(BASIC);
         when(confluenceSourceConfig.getAuthenticationConfig()).thenReturn(authenticationConfig);
         when(authenticationConfig.getBasicConfig()).thenReturn(basicConfig);
@@ -137,7 +137,7 @@ void testValidateConfigBasic() {
 
     @Test
     void testValidateConfigOauth2() {
-        when(confluenceSourceConfig.getAccountUrl()).thenReturn("https://test.com");
+        when(confluenceSourceConfig.getAccountUrl()).thenReturn("https://somedomain.atlassian.net");
         when(confluenceSourceConfig.getAuthType()).thenReturn(OAUTH2);
         when(confluenceSourceConfig.getAuthenticationConfig()).thenReturn(authenticationConfig);
         when(authenticationConfig.getOauth2Config()).thenReturn(oauth2Config);

data-prepper-plugins/saas-source-plugins/jira-source/src/test/java/org/opensearch/dataprepper/plugins/source/jira/JiraConfigHelperTest.java

@@ -126,7 +126,7 @@ void testGetProjectNameFilter() {
     void testValidateConfig() {
         assertThrows(RuntimeException.class, () -> JiraConfigHelper.validateConfig(jiraSourceConfig));
 
-        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://test.com");
+        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://somedomain.atlassian.net");
         assertThrows(RuntimeException.class, () -> JiraConfigHelper.validateConfig(jiraSourceConfig));
 
         when(jiraSourceConfig.getAuthType()).thenReturn("fakeType");
@@ -135,7 +135,7 @@ void testValidateConfig() {
 
     @Test
     void testValidateConfigBasic() {
-        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://test.com");
+        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://somedomain.atlassian.net");
         when(jiraSourceConfig.getAuthType()).thenReturn(BASIC);
         when(jiraSourceConfig.getAuthenticationConfig()).thenReturn(authenticationConfig);
         when(authenticationConfig.getBasicConfig()).thenReturn(basicConfig);
@@ -154,7 +154,7 @@ void testValidateConfigBasic() {
 
     @Test
     void testValidateConfigOauth2() {
-        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://test.com");
+        when(jiraSourceConfig.getAccountUrl()).thenReturn("https://somedomain.atlassian.net");
         when(jiraSourceConfig.getAuthType()).thenReturn(OAUTH2);
         when(jiraSourceConfig.getAuthenticationConfig()).thenReturn(authenticationConfig);
         when(authenticationConfig.getOauth2Config()).thenReturn(oauth2Config);