Reverted to local context/event usage in tests

Author: dreamorosi

packages/tracing/tests/unit/Tracer.test.ts

@@ -1,5 +1,4 @@
 import { LambdaInterface } from '../../examples/utils/lambda';
-import { Events, ContextExamples } from '@aws-lambda-powertools/commons';
 import { Tracer } from '../../src';
 import { Callback, Context } from 'aws-lambda/handler';
 import { Segment, setContextMissingStrategy, Subsegment } from 'aws-xray-sdk-core';
@@ -10,8 +9,25 @@ jest.spyOn(console, 'error').mockImplementation(() => null);
 
 describe('Class: Tracer', () => {
   const ENVIRONMENT_VARIABLES = process.env;
-  const event = Events.Custom.CustomEvent;
-  const context = ContextExamples.helloworldContext;
+  const event = {
+    key1: 'value1',
+    key2: 'value2',
+    key3: 'value3',
+  };
+  const context = {
+    callbackWaitsForEmptyEventLoop: true,
+    functionVersion: '$LATEST',
+    functionName: 'foo-bar-function',
+    memoryLimitInMB: '128',
+    logGroupName: '/aws/lambda/foo-bar-function-123456abcdef',
+    logStreamName: '2021/03/09/[$LATEST]abcdef123456abcdef123456abcdef123456',
+    invokedFunctionArn: 'arn:aws:lambda:eu-central-1:123456789012:function:Example',
+    awsRequestId: 'c6af9ac6-7b61-11e6-9a41-93e8deadbeef',
+    getRemainingTimeInMillis: () => 1234,
+    done: () => console.log('Done!'),
+    fail: () => console.log('Failed!'),
+    succeed: () => console.log('Succeeded!'),
+  };
 
   beforeEach(() => {
     Tracer.coldStart = true;
@@ -556,7 +572,7 @@ describe('Class: Tracer', () => {
       }
 
       // Act
-      await new Lambda().handler(Events.Custom.CustomEvent, context, () => console.log('Lambda invoked!'));
+      await new Lambda().handler(event, context, () => console.log('Lambda invoked!'));
 
       // Assess
       expect(captureAsyncFuncSpy).toHaveBeenCalledTimes(0);

packages/tracing/tests/unit/middy.test.ts

@@ -3,14 +3,27 @@ import middy from '@middy/core';
 import { Tracer } from './../../src';
 import type { Context, Handler } from 'aws-lambda/handler';
 import { Segment, setContextMissingStrategy, Subsegment } from 'aws-xray-sdk-core';
-import { ContextExamples } from '@aws-lambda-powertools/commons';
 
 jest.spyOn(console, 'debug').mockImplementation(() => null);
 jest.spyOn(console, 'warn').mockImplementation(() => null);
 jest.spyOn(console, 'error').mockImplementation(() => null);
 
 describe('Middy middlewares', () => {
   const ENVIRONMENT_VARIABLES = process.env;
+  const context = {
+    callbackWaitsForEmptyEventLoop: true,
+    functionVersion: '$LATEST',
+    functionName: 'foo-bar-function',
+    memoryLimitInMB: '128',
+    logGroupName: '/aws/lambda/foo-bar-function-123456abcdef',
+    logStreamName: '2021/03/09/[$LATEST]abcdef123456abcdef123456abcdef123456',
+    invokedFunctionArn: 'arn:aws:lambda:eu-central-1:123456789012:function:Example',
+    awsRequestId: 'c6af9ac6-7b61-11e6-9a41-93e8deadbeef',
+    getRemainingTimeInMillis: () => 1234,
+    done: () => console.log('Done!'),
+    fail: () => console.log('Failed!'),
+    succeed: () => console.log('Succeeded!'),
+  };
 
   beforeEach(() => {
     Tracer.coldStart = true;
@@ -36,7 +49,6 @@ describe('Middy middlewares', () => {
         foo: 'bar'
       });
       const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-      const context = Object.assign({}, ContextExamples.helloworldContext);
 
       // Act
       await handler({}, context, () => console.log('Lambda invoked!'));
@@ -59,7 +71,6 @@ describe('Middy middlewares', () => {
         throw new Error('Exception thrown!');
       };
       const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-      const context = Object.assign({}, ContextExamples.helloworldContext);
 
       // Act & Assess
       await expect(handler({}, context, () => console.log('Lambda invoked!'))).rejects.toThrowError(Error);
@@ -82,7 +93,6 @@ describe('Middy middlewares', () => {
         foo: 'bar'
       });
       const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-      const context = Object.assign({}, ContextExamples.helloworldContext);
 
       // Act
       await handler({}, context, () => console.log('Lambda invoked!'));
@@ -106,7 +116,6 @@ describe('Middy middlewares', () => {
         foo: 'bar'
       });
       const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-      const context = Object.assign({}, ContextExamples.helloworldContext);
 
       // Act
       await handler({}, context, () => console.log('Lambda invoked!'));
@@ -141,7 +150,6 @@ describe('Middy middlewares', () => {
         throw new Error('Exception thrown!');
       };
       const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-      const context = Object.assign({}, ContextExamples.helloworldContext);
 
       // Act & Assess
       await expect(handler({}, context, () => console.log('Lambda invoked!'))).rejects.toThrowError(Error);
@@ -170,7 +178,6 @@ describe('Middy middlewares', () => {
       throw new Error('Exception thrown!');
     };
     const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-    const context = Object.assign({}, ContextExamples.helloworldContext);
 
     // Act & Assess
     await expect(handler({}, context, () => console.log('Lambda invoked!'))).rejects.toThrowError(Error);
@@ -201,7 +208,6 @@ describe('Middy middlewares', () => {
       foo: 'bar'
     });
     const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-    const context = Object.assign({}, ContextExamples.helloworldContext);
 
     // Act
     await handler({}, context, () => console.log('Lambda invoked!'));
@@ -245,7 +251,6 @@ describe('Middy middlewares', () => {
       foo: 'bar'
     });
     const handler = middy(lambdaHandler).use(captureLambdaHandler(tracer));
-    const context = Object.assign({}, ContextExamples.helloworldContext);
 
     // Act
     await handler({}, context, () => console.log('Lambda invoked!'));