Enhance integration tests by adding retry functionality. Updated test annotations to use @RetryingTest and @RetryingParameterizedTest, and included TestRetryExtension for improved error handling during test execution.

Author: YunchuWang

client/src/test/java/com/microsoft/durabletask/ErrorHandlingIntegrationTests.java

@@ -7,6 +7,7 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.ValueSource;
+import org.junit.jupiter.api.extension.ExtendWith;
 
 import java.time.Duration;
 import java.util.concurrent.TimeoutException;
@@ -24,14 +25,15 @@
  * client operations and sends invocation instructions to the DurableTaskWorker).
  */
 @Tag("integration")
+@ExtendWith(TestRetryExtension.class)
 public class ErrorHandlingIntegrationTests extends IntegrationTestBase {
     @BeforeEach
     private void startUp() {
         DurableTaskClient client = new DurableTaskGrpcClientBuilder().build();
         client.deleteTaskHub();
     }
 
-    @Test
+    @RetryingTest
     void orchestratorException() throws TimeoutException {
         final String orchestratorName = "OrchestratorWithException";
         final String errorMessage = "Kah-BOOOOOM!!!";
@@ -57,7 +59,7 @@ void orchestratorException() throws TimeoutException {
         }
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(booleans = {true, false})
     void activityException(boolean handleException) throws TimeoutException {
         final String orchestratorName = "OrchestratorWithActivityException";
@@ -109,7 +111,7 @@ void activityException(boolean handleException) throws TimeoutException {
         }
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(ints = {1, 2, 10})
     public void retryActivityFailures(int maxNumberOfAttempts) throws TimeoutException {
         // There is one task for each activity call and one task between each retry
@@ -123,7 +125,7 @@ public void retryActivityFailures(int maxNumberOfAttempts) throws TimeoutExcepti
         });
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(ints = {1, 2, 10})
     public void retryActivityFailuresWithCustomLogic(int maxNumberOfAttempts) throws TimeoutException {
         // This gets incremented every time the retry handler is invoked
@@ -140,7 +142,7 @@ public void retryActivityFailuresWithCustomLogic(int maxNumberOfAttempts) throws
         assertEquals(maxNumberOfAttempts, retryHandlerCalls.get());
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(booleans = {true, false})
     void subOrchestrationException(boolean handleException) throws TimeoutException {
         final String orchestratorName = "OrchestrationWithBustedSubOrchestrator";
@@ -190,7 +192,7 @@ void subOrchestrationException(boolean handleException) throws TimeoutException
         }
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(ints = {1, 2, 10})
     public void retrySubOrchestratorFailures(int maxNumberOfAttempts) throws TimeoutException {
         // There is one task for each sub-orchestrator call and one task between each retry
@@ -205,7 +207,7 @@ public void retrySubOrchestratorFailures(int maxNumberOfAttempts) throws Timeout
             });
     }
 
-    @ParameterizedTest
+    @RetryingParameterizedTest
     @ValueSource(ints = {1, 2, 10})
     public void retrySubOrchestrationFailuresWithCustomLogic(int maxNumberOfAttempts) throws TimeoutException {
         // This gets incremented every time the retry handler is invoked