Add sqs handler test for promise style async handler

Author: wpessers

plugins/node/opentelemetry-instrumentation-aws-lambda/test/integrations/lambda-handler.test.ts

@@ -828,7 +828,7 @@ describe('lambda handler', () => {
     });
   });
 
-  describe('sqs test', () => {
+  describe('sync handler sqs propagation', () => {
     it('creates process span for sqs record, with lambda invocation span as parent and span link to the producer traceId and spanId', async () => {
       initializeHandler('lambda-test/sync.sqshandler');
       const producerTraceId = '1df415edd0ad7f83e573f6504381dcec';
@@ -867,4 +867,42 @@ describe('lambda handler', () => {
       assert.equal(spans[0].links[0].context.spanId, producerSpanId);
     });
   });
+
+  describe('async handler sqs propagation', () => {
+    it('creates process span for sqs record, with lambda invocation span as parent and span link to the producer traceId and spanId', async () => {
+      initializeHandler('lambda-test/async.sqshandler');
+      const producerTraceId = '1df415edd0ad7f83e573f6504381dcec';
+      const producerSpanId = '83b7424a259945cb';
+      const event = {
+        Records: [
+          {
+            messageAttributes: {
+              traceparent: {
+                stringValue: `00-${producerTraceId}-${producerSpanId}-01`,
+                dataType: 'String',
+              },
+            },
+            eventSource: 'aws:sqs',
+            eventSourceARN:
+              'arn:aws:sqs:eu-central-1:783764587482:launch-queue',
+          },
+        ],
+      };
+
+      await lambdaRequire('lambda-test/async').sqshandler(event, ctx);
+      const spans = memoryExporter.getFinishedSpans();
+
+      assert.strictEqual(spans.length, 2);
+      assert.equal(
+        spans[0].parentSpanContext?.traceId,
+        spans[1].spanContext().traceId
+      );
+      assert.equal(
+        spans[0].parentSpanContext?.spanId,
+        spans[1].spanContext().spanId
+      );
+      assert.equal(spans[0].links[0]?.context.traceId, producerTraceId);
+      assert.equal(spans[0].links[0].context.spanId, producerSpanId);
+    });
+  });
 });

plugins/node/opentelemetry-instrumentation-aws-lambda/test/lambda-test/async.js

@@ -19,19 +19,26 @@ exports.handler = async function (event, context) {
   return 'ok';
 };
 
+exports.sqshandler = async function (event, context) {
+  event.Records.forEach(r => {});
+  return 'ok';
+};
+
 exports.error = async function (event, context) {
   throw new Error('handler error');
-}
+};
 
 exports.stringerror = async function (event, context) {
   throw 'handler error';
-}
+};
 
 exports.context = async function (event, context) {
   return api.trace.getSpan(api.context.active()).spanContext().traceId;
 };
 
 exports.handler_return_baggage = async function (event, context) {
-  const [baggageEntryKey, baggageEntryValue] =  api.propagation.getBaggage(api.context.active()).getAllEntries()[0];
+  const [baggageEntryKey, baggageEntryValue] = api.propagation
+    .getBaggage(api.context.active())
+    .getAllEntries()[0];
   return `${baggageEntryKey}=${baggageEntryValue.value}`;
-}
+};