chore(testing): use retry options in xray trace retrieval

Author: dreamorosi

packages/testing/src/xray-traces-utils.ts

@@ -49,8 +49,7 @@ const getTraceIds = async (
     );
   }
 
-  const ids = [];
-
+  const ids: string[] = [];
   for (const summary of summaries) {
     if (summary.Id === undefined) {
       throw new Error(
@@ -83,10 +82,14 @@ const retriableGetTraceIds = (options: GetXRayTraceIdsOptions) =>
             endTime.getTime() / 1000
           )} --filter-expression 'resource.arn ENDSWITH ":function:${options.resourceName}"'`
         );
+
+        throw new Error(
+          `Failed to get trace IDs after ${retryOptions.retries} attempts`
+        );
       }
       retry(error);
     }
-  });
+  }, retryOptions);
 
 /**
  * Parse and sort the trace segments by start time
@@ -168,13 +171,24 @@ const getTraceDetails = async (
  * @param options - The options to get trace details, including the trace IDs and expected segments count
  */
 const retriableGetTraceDetails = (options: GetXRayTraceDetailsOptions) =>
-  promiseRetry(async (retry) => {
+  promiseRetry(async (retry, attempt) => {
     try {
       return await getTraceDetails(options);
     } catch (error) {
+      if (attempt === retryOptions.retries) {
+        console.log(
+          `Manual query: aws xray batch-get-traces --trace-ids ${
+            options.traceIds
+          }`
+        );
+
+        throw new Error(
+          `Failed to get trace details after ${retryOptions.retries} attempts`
+        );
+      }
       retry(error);
     }
-  });
+  }, retryOptions);
 
 /**
  * Find the main function segment in the trace identified by the `## index.` suffix
@@ -292,7 +306,7 @@ const getTraces = async (
 
   const { resourceName } = options;
 
-  const mainSubsegments = [];
+  const mainSubsegments: EnrichedXRayTraceDocumentParsed[] = [];
   for (const trace of traces) {
     const mainSubsegment = findPowertoolsFunctionSegment(trace, resourceName);
     const enrichedMainSubsegment = {
@@ -319,7 +333,7 @@ const getTracesWithoutMainSubsegments = async (
 
   const { resourceName } = options;
 
-  const lambdaFunctionSegments = [];
+  const lambdaFunctionSegments: EnrichedXRayTraceDocumentParsed[] = [];
   for (const trace of traces) {
     const functionSegment = trace.Segments.find(
       (segment) => segment.Document.origin === 'AWS::Lambda::Function'