fix/troubleshooting

Author: dreamorosi

Diff for: packages/tracing/tests/e2e/tracer.test.ts

@@ -151,11 +151,25 @@ describe('Tracer integration tests', () => {
           // Assert that there are three subsegments
           expect(handlerSubsegment?.subsegments?.length).toBe(3);
 
-          const [ AWSSDKSubsegment1, AWSSDKSubsegment2, HTTPSegment ] = handlerSubsegment?.subsegments;
-          // Assert that the subsegment names are the expected ones
-          expect(AWSSDKSubsegment1.name).toBe('DynamoDB');
-          expect(AWSSDKSubsegment2.name).toBe('DynamoDB');
-          expect(HTTPSegment.name).toBe('httpbin.org');
+          // Sort the subsegments by name
+          const dynamoDBSubsegments: ParsedDocument[] = [];
+          const httpSubsegment: ParsedDocument[] = [];
+          const otherSegments: ParsedDocument[] = [];
+          handlerSubsegment?.subsegments.forEach(subsegment => {
+            if (subsegment.name === 'DynamoDB') {
+              dynamoDBSubsegments.push(subsegment);
+            } else if (subsegment.name === 'httpbin.org') {
+              httpSubsegment.push(subsegment);
+            } else {
+              otherSegments.push(subsegment);
+            }
+          });
+          // Assert that there are exactly two subsegment with the name 'DynamoDB'
+          expect(dynamoDBSubsegments.length).toBe(2);
+          // Assert that there is exactly one subsegment with the name 'httpbin.org'
+          expect(httpSubsegment.length).toBe(1);
+          // Assert that there are exactly zero other subsegments
+          expect(otherSegments.length).toBe(0);
 
           const { annotations, metadata } = handlerSubsegment;
 
@@ -222,11 +236,25 @@ describe('Tracer integration tests', () => {
           // Assert that there're two subsegments
           expect(handlerSubsegment?.subsegments?.length).toBe(3);
 
-          const [ AWSSDKSubsegment1, AWSSDKSubsegment2, HTTPSegment ] = handlerSubsegment?.subsegments;
-          // Assert that the subsegment names is the expected ones
-          expect(AWSSDKSubsegment1.name).toBe('DynamoDB');
-          expect(AWSSDKSubsegment2.name).toBe('DynamoDB');
-          expect(HTTPSegment.name).toBe('httpbin.org');
+          // Sort the subsegments by name
+          const dynamoDBSubsegments: ParsedDocument[] = [];
+          const httpSubsegment: ParsedDocument[] = [];
+          const otherSegments: ParsedDocument[] = [];
+          handlerSubsegment?.subsegments.forEach(subsegment => {
+            if (subsegment.name === 'DynamoDB') {
+              dynamoDBSubsegments.push(subsegment);
+            } else if (subsegment.name === 'httpbin.org') {
+              httpSubsegment.push(subsegment);
+            } else {
+              otherSegments.push(subsegment);
+            }
+          });
+          // Assert that there are exactly two subsegment with the name 'DynamoDB'
+          expect(dynamoDBSubsegments.length).toBe(2);
+          // Assert that there is exactly one subsegment with the name 'httpbin.org'
+          expect(httpSubsegment.length).toBe(1);
+          // Assert that there are exactly zero other subsegments
+          expect(otherSegments.length).toBe(0);
 
           const { annotations, metadata } = handlerSubsegment;
 
@@ -293,11 +321,25 @@ describe('Tracer integration tests', () => {
           // Assert that there're two subsegments
           expect(handlerSubsegment?.subsegments?.length).toBe(3);
 
-          const [ AWSSDKSubsegment1, AWSSDKSubsegment2, HTTPSegment ] = handlerSubsegment?.subsegments;
-          // Assert that the subsegment names are the expected ones
-          expect(AWSSDKSubsegment1.name).toBe('DynamoDB');
-          expect(AWSSDKSubsegment2.name).toBe('DynamoDB');
-          expect(HTTPSegment.name).toBe('httpbin.org');
+          // Sort the subsegments by name
+          const dynamoDBSubsegments: ParsedDocument[] = [];
+          const httpSubsegment: ParsedDocument[] = [];
+          const otherSegments: ParsedDocument[] = [];
+          handlerSubsegment?.subsegments.forEach(subsegment => {
+            if (subsegment.name === 'DynamoDB') {
+              dynamoDBSubsegments.push(subsegment);
+            } else if (subsegment.name === 'httpbin.org') {
+              httpSubsegment.push(subsegment);
+            } else {
+              otherSegments.push(subsegment);
+            }
+          });
+          // Assert that there are exactly two subsegment with the name 'DynamoDB'
+          expect(dynamoDBSubsegments.length).toBe(2);
+          // Assert that there is exactly one subsegment with the name 'httpbin.org'
+          expect(httpSubsegment.length).toBe(1);
+          // Assert that there are exactly zero other subsegments
+          expect(otherSegments.length).toBe(0);
 
           const { annotations, metadata } = handlerSubsegment;
 
@@ -384,7 +426,7 @@ describe('Tracer integration tests', () => {
         // Assert that the subsegment name is the expected one
         expect(handlerSubsegment.name).toBe('## index.handler');
         if (handlerSubsegment?.subsegments !== undefined) {
-          // Assert that there're three subsegments
+          // Assert that there are four subsegments
           expect(handlerSubsegment?.subsegments?.length).toBe(4);
 
           // Sort the subsegments by name
@@ -486,7 +528,7 @@ describe('Tracer integration tests', () => {
         // Assert that the subsegment name is the expected one
         expect(handlerSubsegment.name).toBe('## index.handler');
         if (handlerSubsegment?.subsegments !== undefined) {
-          // Assert that there're three subsegments
+          // Assert that there are four subsegments
           expect(handlerSubsegment?.subsegments?.length).toBe(4);
 
           // Sort the subsegments by name
@@ -588,8 +630,8 @@ describe('Tracer integration tests', () => {
         // Assert that the subsegment name is the expected one
         expect(handlerSubsegment.name).toBe('## index.handler');
         if (handlerSubsegment?.subsegments !== undefined) {
-          // Assert that there're three subsegments
-          expect(handlerSubsegment?.subsegments?.length).toBe(5);
+          // Assert that there are four subsegments
+          expect(handlerSubsegment?.subsegments?.length).toBe(4);
 
           // Sort the subsegments by name
           const dynamoDBSubsegments: ParsedDocument[] = [];