fix: add retry for statemachine integ test (#3630)

Author: godwingrs22

integration/helpers/base_test.py

@@ -567,6 +567,13 @@ def verify_options_request(self, url, expected_status_code, headers=None):
             )
         return response
 
+    @retry(
+        stop=stop_after_attempt(5),
+        wait=wait_exponential(multiplier=1, min=16, max=64) + wait_random(0, 1),
+        retry=retry_if_exception_type(StatusCodeError),
+        after=after_log(LOG, logging.WARNING),
+        reraise=True,
+    )
     def verify_post_request(self, url: str, body_obj, expected_status_code: int, headers=None):
         """Return response to POST request and verify matches expected status code."""
         response = self.do_post_request_with_logging(url, body_obj, headers)

integration/single/test_basic_api.py

@@ -1,5 +1,6 @@
 import json
 import logging
+import time
 from unittest.case import skipIf
 
 from tenacity import after_log, retry_if_exception_type, stop_after_attempt, wait_exponential, wait_random
@@ -129,8 +130,14 @@ def test_state_machine_with_api_single_quotes_input(self):
         api_endpoint = stack_output.get("ApiEndpoint")
 
         input_json = {"f'oo": {"hello": "'wor'l'd'''"}}
+
+        # This will be the wait time before triggering the APIGW request
+        time.sleep(10)
+
         response = self.verify_post_request(api_endpoint, input_json, 200)
 
+        LOG.log(msg=f"retry times {self.verify_get_request_response.retry.statistics}", level=logging.WARNING)
+
         execution_arn = response.json()["executionArn"]
         execution = self.client_provider.sfn_client.describe_execution(executionArn=execution_arn)
         execution_input = json.loads(execution["input"])