Adding retryable annotation to the rest of the flaky tets (#6020)

Co-authored-by: Ran Vaknin <[email protected]>

Author: RanVaknin

test/protocol-tests/src/test/java/software/amazon/awssdk/protocol/tests/timeout/BaseApiCallAttemptTimeoutTest.java

@@ -28,19 +28,19 @@ public abstract class BaseApiCallAttemptTimeoutTest extends BaseTimeoutTest {
     protected static final Duration DELAY_BEFORE_API_CALL_ATTEMPT_TIMEOUT = Duration.ofMillis(50);
     protected static final Duration DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT = Duration.ofMillis(500);
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void nonstreamingOperation200_finishedWithinTime_shouldSucceed() throws Exception {
         stubSuccessResponse(DELAY_BEFORE_API_CALL_ATTEMPT_TIMEOUT);
         verifySuccessResponseNotTimedOut();
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void nonstreamingOperation200_notFinishedWithinTime_shouldTimeout() {
         stubSuccessResponse(DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT);
         verifyTimedOut();
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void nonstreamingOperation500_finishedWithinTime_shouldNotTimeout() throws Exception {
         stubErrorResponse(DELAY_BEFORE_API_CALL_ATTEMPT_TIMEOUT);
         verifyFailedResponseNotTimedOut();
@@ -58,13 +58,13 @@ public void streamingOperation_finishedWithinTime_shouldSucceed() throws Excepti
         verifySuccessResponseNotTimedOut();
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void streamingOperation_notFinishedWithinTime_shouldTimeout() {
         stubSuccessResponse(DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT);
         verifyTimedOut();
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void firstAttemptTimeout_retryFinishWithInTime_shouldNotTimeout() throws Exception {
         mockHttpClient().stubResponses(Pair.of(mockResponse(200), DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT),
                                        Pair.of(mockResponse(200), DELAY_BEFORE_API_CALL_ATTEMPT_TIMEOUT));
@@ -73,15 +73,15 @@ public void firstAttemptTimeout_retryFinishWithInTime_shouldNotTimeout() throws
         verifyRequestCount(2);
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void firstAttemptTimeout_retryFinishWithInTime500_shouldNotTimeout() {
         mockHttpClient().stubResponses(Pair.of(mockResponse(200), DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT),
                                        Pair.of(mockResponse(500), DELAY_BEFORE_API_CALL_ATTEMPT_TIMEOUT));
         verifyRetraybleFailedResponseNotTimedOut();
         verifyRequestCount(2);
     }
 
-    @Test
+    @RetryableTest(maxRetries = 3)
     public void allAttemptsNotFinishedWithinTime_shouldTimeout() {
         stubSuccessResponse(DELAY_AFTER_API_CALL_ATTEMPT_TIMEOUT);
         verifyRetryableTimeout();