@datadog/datadog-ci 0.17.9 → 0.17.13

package/dist/commands/lambda/__tests__/functions/uninstrument.test.js

@@ -49,6 +49,7 @@ describe('uninstrument', () => {
                 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
                     Environment: {
                         Variables: {
+                            [constants_1.API_KEY_SECRET_ARN_ENV_VAR]: 'some-secret:arn:from:aws',
                             [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
                             [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
                             [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
@@ -146,7 +147,7 @@ describe('uninstrument', () => {
       `);
         }));
     });
-    describe('getFunctionConfigs', () => {
+    describe('getUninstrumentedFunctionConfigs', () => {
         const OLD_ENV = process.env;
         beforeEach(() => {
             jest.resetModules();
@@ -178,7 +179,7 @@ describe('uninstrument', () => {
                 },
             });
             const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
-            const result = yield uninstrument_1.getFunctionConfigs(lambda, cloudWatch, ['arn:aws:lambda:us-east-1:000000000000:function:uninstrument'], undefined);
+            const result = yield uninstrument_1.getUninstrumentedFunctionConfigs(lambda, cloudWatch, ['arn:aws:lambda:us-east-1:000000000000:function:uninstrument'], undefined);
             expect(result.length).toEqual(1);
             expect(result[0].updateRequest).toMatchInlineSnapshot(`
         Object {
@@ -218,7 +219,7 @@ describe('uninstrument', () => {
                 },
             });
             const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
-            const result = yield uninstrument_1.getFunctionConfigs(lambda, cloudWatch, [
+            const result = yield uninstrument_1.getUninstrumentedFunctionConfigs(lambda, cloudWatch, [
                 'arn:aws:lambda:us-east-1:000000000000:function:another-func',
                 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument',
             ], undefined);
@@ -235,7 +236,7 @@ describe('uninstrument', () => {
       `);
         }));
     });
-    describe('getFunctionConfig', () => {
+    describe('getUninstrumentedFunctionConfig', () => {
         const OLD_ENV = process.env;
         beforeEach(() => {
             jest.resetModules();
@@ -253,7 +254,7 @@ describe('uninstrument', () => {
             });
             const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
             const config = yield commons_1.getLambdaFunctionConfig(lambda, 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument');
-            yield expect(uninstrument_1.getFunctionConfig(lambda, cloudWatch, config, undefined)).rejects.toThrow();
+            yield expect(uninstrument_1.getUninstrumentedFunctionConfig(lambda, cloudWatch, config, undefined)).rejects.toThrow();
         }));
         test('returns configurations without updateRequest when no changes need to be made', () => __awaiter(void 0, void 0, void 0, function* () {
             const lambda = fixtures_1.makeMockLambda({
@@ -268,7 +269,7 @@ describe('uninstrument', () => {
             });
             const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
             const config = yield commons_1.getLambdaFunctionConfig(lambda, 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument');
-            expect((yield uninstrument_1.getFunctionConfig(lambda, cloudWatch, config, undefined)).updateRequest).toBeUndefined();
+            expect((yield uninstrument_1.getUninstrumentedFunctionConfig(lambda, cloudWatch, config, undefined)).updateRequest).toBeUndefined();
         }));
         test('returns log group configuration subscription delete request when forwarderARN is set', () => __awaiter(void 0, void 0, void 0, function* () {
             const logGroupName = '/aws/lambda/group';
@@ -285,7 +286,7 @@ describe('uninstrument', () => {
             });
             const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
             const config = yield commons_1.getLambdaFunctionConfig(lambda, 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument');
-            const result = yield uninstrument_1.getFunctionConfig(lambda, cloudWatch, config, 'valid-forwarder-arn');
+            const result = yield uninstrument_1.getUninstrumentedFunctionConfig(lambda, cloudWatch, config, 'valid-forwarder-arn');
             expect(result).toBeDefined();
             expect(result.logGroupConfiguration).toMatchInlineSnapshot(`
         Object {
@@ -295,4 +296,76 @@ describe('uninstrument', () => {
       `);
         }));
     });
+    describe('getUninstrumentedFunctionConfigsFromRegEx', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('returns the update request for each function that matches the pattern', () => __awaiter(void 0, void 0, void 0, function* () {
+            const lambda = fixtures_1.makeMockLambda({
+                'autoinstrument-scooby': {
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'index.handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                            [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                            [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                            [constants_1.TRACE_ENABLED_ENV_VAR]: 'true',
+                            [constants_1.VERSION_ENV_VAR]: '0.2',
+                            USER_VARIABLE: 'shouldnt be deleted by uninstrumentation',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:us-east-1:000000000000:function:autoinstrument-scooby',
+                    FunctionName: 'autoinstrument-scooby',
+                    Handler: '/opt/nodejs/node_modules/datadog-lambda-js/handler.handler',
+                    Runtime: 'nodejs12.x',
+                },
+                'autoinstrument-scrapy': {
+                    Environment: {
+                        Variables: {
+                            [constants_1.API_KEY_ENV_VAR]: '1234',
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'index.handler',
+                            [constants_1.LOG_LEVEL_ENV_VAR]: 'debug',
+                            [constants_1.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:us-east-1:000000000000:function:autoinstrument-scrapy',
+                    FunctionName: 'autoinstrument-scrapy',
+                    Handler: '/opt/nodejs/node_modules/datadog-lambda-js/handler.handler',
+                    Runtime: 'nodejs12.x',
+                },
+            });
+            const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
+            const result = yield uninstrument_1.getUninstrumentedFunctionConfigsFromRegEx(lambda, cloudWatch, 'autoinstrument-scr.', undefined);
+            expect(result.length).toEqual(1);
+            expect(result[0].updateRequest).toMatchInlineSnapshot(`
+        Object {
+          "Environment": Object {
+            "Variables": Object {},
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:000000000000:function:autoinstrument-scrapy",
+          "Handler": "index.handler",
+        }
+      `);
+        }));
+        test('fails when retry count is exceeded', () => __awaiter(void 0, void 0, void 0, function* () {
+            const makeMockLambdaListFunctionsError = () => ({
+                listFunctions: jest.fn().mockImplementation((args) => ({
+                    promise: () => Promise.reject(),
+                })),
+            });
+            const lambda = makeMockLambdaListFunctionsError();
+            const cloudWatch = fixtures_1.makeMockCloudWatchLogs({});
+            yield expect(uninstrument_1.getUninstrumentedFunctionConfigsFromRegEx(lambda, cloudWatch, 'fake-pattern', undefined)).rejects.toStrictEqual(new Error('Max retry count exceeded.'));
+        }));
+    });
 });