fix/troubleshooting

Author: dreamorosi

packages/tracing/tests/e2e/tracer.test.ts

@@ -154,11 +154,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;
 
@@ -225,11 +239,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;
 
@@ -296,11 +324,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;
 
@@ -387,7 +429,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
@@ -489,7 +531,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
@@ -591,8 +633,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[] = [];