Fix tests

thundra-io · Jul 4, 2019 · dc263d7 · dc263d7
1 parent b9a812c
commit dc263d7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/test/plugins/trace.test.js b/test/plugins/trace.test.js
@@ -571,7 +571,7 @@ describe('Trace', () => {
         it('should set trigger tags for APIGatewayProxy to root span', () => {
             expect(tracer.rootSpan.tags['trigger.domainName']).toBe('API');
             expect(tracer.rootSpan.tags['trigger.className']).toBe('AWS-APIGateway');
-            expect(tracer.rootSpan.tags['trigger.operationNames']).toEqual([ '1234567890.execute-api.us-west-2.amazonaws.com/prod/path/to/resource' ]);
+            expect(tracer.rootSpan.tags['trigger.operationNames']).toEqual([ '/{proxy+}' ]);
             expect(tracer.rootSpan.tags['topology.vertex']).toBe(true);
         });
 

diff --git a/test/thundra-wrapper.test.js b/test/thundra-wrapper.test.js
@@ -142,7 +142,8 @@ describe('ThundraWrapper', () => {
             it('should extract error from response with valid error response', () => {
                 const expectedAfterInvocationData = {
                     error: new HttpError('Lambda returned with error response.'),
-                    response: response   
+                    originalEvent,
+                    response,
                 };
                 expect(thundraWrapper.executeHook).toBeCalledWith('after-invocation', expectedAfterInvocationData, true);
             });
@@ -159,6 +160,7 @@ describe('ThundraWrapper', () => {
             it('should extract error from response with invalid body', () => {   
                 const expectedAfterInvocationData = {
                     error: new HttpError('Lambda returned with error response.'),
+                    originalEvent,
                     response: response   
                 };
                 expect(thundraWrapper.executeHook).toBeCalledWith('after-invocation', expectedAfterInvocationData, true);
@@ -174,6 +176,7 @@ describe('ThundraWrapper', () => {
             it('should extract error from response with success status', () => {   
                 const expectedAfterInvocationData = {
                     error: null,
+                    originalEvent,
                     response: {statusCode: 200, body:'{\'message\':\'I have failed\'}'}  
                 };
                 expect(thundraWrapper.executeHook).toBeCalledWith('after-invocation', expectedAfterInvocationData, true);
@@ -282,6 +285,7 @@ describe('ThundraWrapper', () => {
         pc.timeoutMargin = null;
         let afterInvocationData = {
             error: new TimeoutError('Lambda is timed out.'),
+            originalEvent,
             response: null,
         };