Test sqsContextGetter keys method

Author: wpessers

packages/instrumentation-aws-lambda/src/instrumentation.ts

@@ -75,7 +75,7 @@ const headerGetter: TextMapGetter<APIGatewayProxyEventHeaders> = {
   },
 };
 
-const sqsContextGetter: TextMapGetter = {
+export const sqsContextGetter: TextMapGetter = {
   keys(carrier): string[] {
     if (carrier == null) {
       return [];

packages/instrumentation-aws-lambda/test/integrations/lambda-handler.test.ts

@@ -54,6 +54,7 @@ import {
 import { AWSXRayPropagator } from '@opentelemetry/propagator-aws-xray';
 import { W3CTraceContextPropagator } from '@opentelemetry/core';
 import { AWSXRayLambdaPropagator } from '@opentelemetry/propagator-aws-xray-lambda';
+import { sqsContextGetter } from '../../src/instrumentation';
 
 const memoryExporter = new InMemorySpanExporter();
 
@@ -907,4 +908,33 @@ describe('lambda handler', () => {
       assert.equal(spans[0].links[0].context.spanId, producerSpanId);
     });
   });
+
+  describe('sqsContextGetter', () => {
+    it('returns the keys for a given message attributes carrier', () => {
+      const carrier = {
+        'x-amzn-trace-id': {
+          stringValue: 'dummy',
+          stringListValues: [],
+          binaryListValues: [],
+          dataType: 'String',
+        },
+        traceparent: {
+          stringValue: 'dummy',
+          stringListValues: [],
+          binaryListValues: [],
+          dataType: 'String',
+        },
+      };
+
+      const keys = sqsContextGetter.keys(carrier);
+      assert.deepEqual(keys, ['x-amzn-trace-id', 'traceparent']);
+    });
+
+    it('returns empty array for null or undefined carrier', () => {
+      const keysNull = sqsContextGetter.keys(null);
+      const keysUndefined = sqsContextGetter.keys(undefined);
+      assert.deepEqual(keysNull, []);
+      assert.deepEqual(keysUndefined, []);
+    });
+  });
 });