Skip to content

Commit 2d26a05

Browse files
committed
Refactor and format lambda handler tests
1 parent 39a8908 commit 2d26a05

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

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

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ import {
3333
ReadableSpan,
3434
} from '@opentelemetry/sdk-trace-base';
3535
import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
36-
import { APIGatewayProxyEvent, Context, SQSEvent } from 'aws-lambda';
36+
import { Context } from 'aws-lambda';
3737
import * as assert from 'assert';
3838
import {
3939
ATTR_URL_FULL,
@@ -1242,7 +1242,7 @@ describe('lambda handler', () => {
12421242
initializeHandler('lambda-test/sync.sqshandler');
12431243
const producerTraceId = '1df415edd0ad7f83e573f6504381dcec';
12441244
const producerSpanId = '83b7424a259945cb';
1245-
const event = {
1245+
const sqsEvent = {
12461246
Records: [
12471247
{
12481248
messageAttributes: {
@@ -1258,11 +1258,14 @@ describe('lambda handler', () => {
12581258
],
12591259
};
12601260

1261-
await lambdaRequire('lambda-test/sync').sqshandler(event, ctx, () => {});
1261+
await lambdaRequire('lambda-test/sync').sqshandler(
1262+
sqsEvent,
1263+
ctx,
1264+
() => {}
1265+
);
12621266
const spans = memoryExporter.getFinishedSpans();
12631267

12641268
assert.strictEqual(spans.length, 2);
1265-
12661269
assert.equal(
12671270
spans[0].parentSpanContext?.traceId,
12681271
spans[1].spanContext().traceId
@@ -1271,7 +1274,6 @@ describe('lambda handler', () => {
12711274
spans[0].parentSpanContext?.spanId,
12721275
spans[1].spanContext().spanId
12731276
);
1274-
12751277
assert.equal(spans[0].links[0]?.context.traceId, producerTraceId);
12761278
assert.equal(spans[0].links[0].context.spanId, producerSpanId);
12771279
});
@@ -1282,7 +1284,7 @@ describe('lambda handler', () => {
12821284
initializeHandler('lambda-test/async.sqshandler');
12831285
const producerTraceId = '1df415edd0ad7f83e573f6504381dcec';
12841286
const producerSpanId = '83b7424a259945cb';
1285-
const event = {
1287+
const sqsEvent = {
12861288
Records: [
12871289
{
12881290
messageAttributes: {
@@ -1298,7 +1300,7 @@ describe('lambda handler', () => {
12981300
],
12991301
};
13001302

1301-
await lambdaRequire('lambda-test/async').sqshandler(event, ctx);
1303+
await lambdaRequire('lambda-test/async').sqshandler(sqsEvent, ctx);
13021304
const spans = memoryExporter.getFinishedSpans();
13031305

13041306
assert.strictEqual(spans.length, 2);

0 commit comments

Comments
 (0)