@datadog/datadog-ci 0.17.9 → 0.17.13

package/dist/commands/lambda/__tests__/prompt.test.js

@@ -0,0 +1,216 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+jest.mock('inquirer');
+const chalk_1 = require("chalk");
+const inquirer_1 = require("inquirer");
+const constants_1 = require("../constants");
+const prompt_1 = require("../prompt");
+const fixtures_1 = require("./fixtures");
+describe('prompt', () => {
+    describe('confirmationQuestion', () => {
+        test('returns question with provided message', () => {
+            const message = 'Do you want to continue?';
+            const question = prompt_1.confirmationQuestion(message);
+            expect(question.message).toBe(message);
+        });
+    });
+    describe('datadogApiKeyTypeQuestion', () => {
+        test('returns question with message pointing to the correct given site', () => {
+            const site = 'datadoghq.com';
+            const question = prompt_1.datadogApiKeyTypeQuestion(site);
+            expect(question.message).toBe(`Which type of Datadog API Key you want to set? \nLearn more at ${chalk_1.blueBright(`https://app.${site}/organization-settings/api-keys`)}`);
+        });
+    });
+    describe('datadogEnvVarsQuestions', () => {
+        test('returns correct message when user selects DATADOG_API_KEY', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_API_KEY_ENV_VAR,
+                message: 'API Key:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('API Key:');
+            expect(question.name).toBe(constants_1.CI_API_KEY_ENV_VAR);
+        });
+        test('returns correct message when user selects DATADOG_KMS_API_KEY', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_KMS_API_KEY_ENV_VAR,
+                message: 'KMS API Key:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('KMS API Key:');
+            expect(question.name).toBe(constants_1.CI_KMS_API_KEY_ENV_VAR);
+        });
+        test('returns correct message when user selects DATADOG_API_KEY_SECRET_ARN', () => {
+            const datadogApiKeyType = {
+                envVar: constants_1.CI_API_KEY_SECRET_ARN_ENV_VAR,
+                message: 'API Key Secret ARN:',
+            };
+            const question = prompt_1.datadogEnvVarsQuestions(datadogApiKeyType);
+            expect(question.message).toBe('API Key Secret ARN:');
+            expect(question.name).toBe(constants_1.CI_API_KEY_SECRET_ARN_ENV_VAR);
+        });
+    });
+    describe('functionSelectionQuestion', () => {
+        test('returns question with the provided function names being its choices', () => {
+            const functionNames = ['my-func', 'my-func-2', 'my-third-func'];
+            const question = prompt_1.functionSelectionQuestion(functionNames);
+            expect(question.choices).toBe(functionNames);
+        });
+    });
+    describe('requestAWSCrendentials', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('sets the AWS credentials as environment variables', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            }));
+            yield prompt_1.requestAWSCredentials();
+            expect(process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]).toBe(fixtures_1.mockAwsAccessKeyId);
+            expect(process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]).toBe(fixtures_1.mockAwsSecretAccessKey);
+            expect(process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR]).toBe('sa-east-1');
+        }));
+        test('sets the AWS credentials with session token as environment variables', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+                [constants_1.AWS_SESSION_TOKEN_ENV_VAR]: 'some-session-token',
+            }));
+            yield prompt_1.requestAWSCredentials();
+            expect(process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]).toBe(fixtures_1.mockAwsAccessKeyId);
+            expect(process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]).toBe(fixtures_1.mockAwsSecretAccessKey);
+            expect(process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR]).toBe('sa-east-1');
+            expect(process.env[constants_1.AWS_SESSION_TOKEN_ENV_VAR]).toBe('some-session-token');
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestAWSCredentials();
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't set AWS Credentials. Unexpected error");
+        }));
+    });
+    describe('requestChangesConfirmation', () => {
+        test('returns boolean when users responds to confirmation question', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({
+                confirmation: true,
+            }));
+            const confirmation = yield prompt_1.requestChangesConfirmation('Do you want to continue?');
+            expect(confirmation).toBe(true);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestChangesConfirmation('Do you wanna continue?');
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't receive confirmation. Unexpected error");
+        }));
+    });
+    describe('requestDatadogEnvVars', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('sets the Datadog Environment Variables as provided/selected by user', () => __awaiter(void 0, void 0, void 0, function* () {
+            const site = 'datadoghq.com';
+            inquirer_1.prompt.mockImplementation((question) => {
+                switch (question.name) {
+                    case constants_1.CI_API_KEY_ENV_VAR:
+                        return Promise.resolve({
+                            [constants_1.CI_API_KEY_ENV_VAR]: fixtures_1.mockDatadogApiKey,
+                        });
+                    case constants_1.CI_SITE_ENV_VAR:
+                        return Promise.resolve({
+                            [constants_1.CI_SITE_ENV_VAR]: 'datadoghq.com',
+                        });
+                    case 'type':
+                        return Promise.resolve({
+                            type: {
+                                envVar: constants_1.CI_API_KEY_ENV_VAR,
+                                message: 'API Key:',
+                            },
+                        });
+                    default:
+                }
+            });
+            yield prompt_1.requestDatadogEnvVars();
+            expect(process.env[constants_1.CI_SITE_ENV_VAR]).toBe(site);
+            expect(process.env[constants_1.CI_API_KEY_ENV_VAR]).toBe(fixtures_1.mockDatadogApiKey);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestDatadogEnvVars();
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't set Datadog Environment Variables. Unexpected error");
+        }));
+    });
+    describe('requestFunctionSelection', () => {
+        const selectedFunctions = ['my-func', 'my-func-2', 'my-third-func'];
+        test('returns the selected functions', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.resolve({ functions: selectedFunctions }));
+            const functions = yield prompt_1.requestFunctionSelection(selectedFunctions);
+            expect(functions).toBe(selectedFunctions);
+        }));
+        test('throws error when something unexpected happens while prompting', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            inquirer_1.prompt.mockImplementation(() => Promise.reject(new Error('Unexpected error')));
+            let error;
+            try {
+                yield prompt_1.requestFunctionSelection(selectedFunctions);
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("Couldn't receive selected functions. Unexpected error");
+        }));
+    });
+});

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

@@ -31,10 +31,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 // tslint:disable: no-string-literal
 jest.mock('fs');
 jest.mock('aws-sdk');
+jest.mock('../prompt');
 const aws_sdk_1 = require("aws-sdk");
 const chalk_1 = require("chalk");
 const fs = __importStar(require("fs"));
 const constants_1 = require("../constants");
+const prompt_1 = require("../prompt");
 const uninstrument_1 = require("../uninstrument");
 const fixtures_1 = require("./fixtures");
 describe('uninstrument', () => {
@@ -52,6 +54,7 @@ describe('uninstrument', () => {
             fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
             aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({
                 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
+                    Architectures: ['x86_64'],
                     Environment: {
                         Variables: {
                             [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
@@ -93,20 +96,22 @@ describe('uninstrument', () => {
             const output = context.stdout.toString();
             expect(code).toBe(0);
             expect(output).toMatchInlineSnapshot(`
-        "${chalk_1.bold(chalk_1.cyan('[Dry Run] '))}Will apply the following updates:
-        UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:uninstrument
-        {
-          \\"FunctionName\\": \\"arn:aws:lambda:us-east-1:000000000000:function:uninstrument\\",
-          \\"Handler\\": \\"lambda_function.lambda_handler\\",
-          \\"Environment\\": {
-            \\"Variables\\": {
-              \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
-            }
-          },
-          \\"Layers\\": []
-        }
-        "
-      `);
+"\n${chalk_1.bold(chalk_1.yellow('[!]'))} Functions to be updated:
+\t- ${chalk_1.bold('arn:aws:lambda:us-east-1:000000000000:function:uninstrument')}\n
+${chalk_1.bold(chalk_1.cyan('[Dry Run] '))}Will apply the following updates:
+UpdateFunctionConfiguration -> arn:aws:lambda:us-east-1:000000000000:function:uninstrument
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:us-east-1:000000000000:function:uninstrument\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {
+      \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
+    }
+  },
+  \\"Layers\\": []
+}
+"
+`);
         }));
         test('runs function update command for valid uninstrumentation', () => __awaiter(void 0, void 0, void 0, function* () {
             ;
@@ -167,7 +172,7 @@ describe('uninstrument', () => {
             const code = yield command['execute']();
             const output = command.context.stdout.toString();
             expect(code).toBe(1);
-            expect(output).toMatch(`${chalk_1.red('[Error]')} Couldn't fetch lambda functions. Lambda failed\n`);
+            expect(output).toMatch(`${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Lambda failed\n`);
         }));
         test("aborts early when function regions can't be found", () => __awaiter(void 0, void 0, void 0, function* () {
             ;
@@ -190,7 +195,7 @@ describe('uninstrument', () => {
             const output = context.stdout.toString();
             expect(code).toBe(1);
             expect(output).toMatchInlineSnapshot(`
-        "No functions specified for un-instrumentation.
+        "${chalk_1.red('[Error]')} No functions specified for un-instrumentation.
         "
       `);
         }));
@@ -203,10 +208,369 @@ describe('uninstrument', () => {
             yield command['execute']();
             const output = command.context.stdout.toString();
             expect(output).toMatchInlineSnapshot(`
-        "No functions specified for un-instrumentation.
+        "${chalk_1.red('[Error]')} No functions specified for un-instrumentation.
         "
       `);
         }));
+        test('aborts if functions and a pattern are set at the same time', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({}));
+            process.env = {};
+            let command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+            command['config']['region'] = 'ap-southeast-1';
+            command['config']['functions'] = ['arn:aws:lambda:ap-southeast-1:123456789012:function:lambda-hello-world'];
+            command['regExPattern'] = 'valid-pattern';
+            yield command['execute']();
+            let output = command.context.stdout.toString();
+            expect(output).toMatch('Functions in config file and "--functions-regex" should not be used at the same time.\n');
+            command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+            command['region'] = 'ap-southeast-1';
+            command['functions'] = ['arn:aws:lambda:ap-southeast-1:123456789012:function:lambda-hello-world'];
+            command['regExPattern'] = 'valid-pattern';
+            yield command['execute']();
+            output = command.context.stdout.toString();
+            expect(output).toMatch('"--functions" and "--functions-regex" should not be used at the same time.\n');
+        }));
+        test('aborts if the regEx pattern is an ARN', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({}));
+            process.env = {};
+            const command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+            command['region'] = 'ap-southeast-1';
+            command['regExPattern'] = 'arn:aws:lambda:ap-southeast-1:123456789012:function:*';
+            const code = yield command['execute']();
+            const output = command.context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatch(`"--functions-regex" isn't meant to be used with ARNs.\n`);
+        }));
+        test('aborts if the regEx pattern is set but no region is specified', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({}));
+            process.env = {};
+            const command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+            command['regExPattern'] = 'my-function';
+            const code = yield command['execute']();
+            const output = command.context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatch('No default region specified. Use `-r`, `--region`.');
+        }));
+        test('aborts if the the aws-sdk fails', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({}));
+            aws_sdk_1.Lambda.mockImplementation(() => ({ promise: Promise.reject() }));
+            process.env = {};
+            const command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+            command['region'] = 'ap-southeast-1';
+            command['regExPattern'] = 'my-function';
+            const code = yield command['execute']();
+            const output = command.context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatch(`Fetching Lambda functions, this might take a while.\n${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Error: Max retry count exceeded.\n`);
+        }));
+        test('uninstrument multiple functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_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:sa-east-1:123456789012:function:lambda-hello-world',
+                    FunctionName: 'lambda-hello-world',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python38:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.8',
+                },
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_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',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+                    FunctionName: 'lambda-hello-world-2',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python39:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.9',
+                },
+            }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => {
+                process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_1.mockAwsAccessKeyId;
+                process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_1.mockAwsSecretAccessKey;
+                process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+            });
+            prompt_1.requestFunctionSelection.mockImplementation(() => [
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ]);
+            prompt_1.requestChangesConfirmation.mockImplementation(() => true);
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(0);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!
+Fetching Lambda functions, this might take a while.\n
+${chalk_1.bold(chalk_1.yellow('[!]'))} Functions to be updated:
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world')}
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2')}\n
+Will apply the following updates:
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {
+      \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
+    }
+  },
+  \\"Layers\\": []
+}
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {}
+  },
+  \\"Layers\\": []
+}
+${chalk_1.yellow('[!]')} Confirmation needed.
+${chalk_1.yellow('[!]')} Uninstrumenting functions.
+"
+`);
+        }));
+        test('uninstrument multiple specified functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_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:sa-east-1:123456789012:function:lambda-hello-world',
+                    FunctionName: 'lambda-hello-world',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python38:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.8',
+                },
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2': {
+                    Architectures: ['x86_64'],
+                    Environment: {
+                        Variables: {
+                            [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                            [constants_1.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                            [constants_1.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_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',
+                        },
+                    },
+                    FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+                    FunctionName: 'lambda-hello-world-2',
+                    Handler: 'datadog_lambda.handler.handler',
+                    Layers: [
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Extension:11',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                        {
+                            Arn: 'arn:aws:lambda:sa-east-1:000000000000:layer:Datadog-Python39:49',
+                            CodeSize: 0,
+                            SigningJobArn: 'some-signing-job-arn',
+                            SigningProfileVersionArn: 'some-signing-profile',
+                        },
+                    ],
+                    Runtime: 'python3.9',
+                },
+            }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => {
+                process.env[constants_1.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_1.mockAwsAccessKeyId;
+                process.env[constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_1.mockAwsSecretAccessKey;
+                process.env[constants_1.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+            });
+            prompt_1.requestFunctionSelection.mockImplementation(() => [
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ]);
+            prompt_1.requestChangesConfirmation.mockImplementation(() => true);
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run([
+                'lambda',
+                'uninstrument',
+                '-i',
+                '-f',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                '-f',
+                'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2',
+            ], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(0);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!\n
+${chalk_1.bold(chalk_1.yellow('[!]'))} Functions to be updated:
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world')}
+\t- ${chalk_1.bold('arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2')}\n
+Will apply the following updates:
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {
+      \\"USER_VARIABLE\\": \\"shouldnt be deleted by uninstrumentation\\"
+    }
+  },
+  \\"Layers\\": []
+}
+UpdateFunctionConfiguration -> arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2
+{
+  \\"FunctionName\\": \\"arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world-2\\",
+  \\"Handler\\": \\"lambda_function.lambda_handler\\",
+  \\"Environment\\": {
+    \\"Variables\\": {}
+  },
+  \\"Layers\\": []
+}
+${chalk_1.yellow('[!]')} Confirmation needed.
+${chalk_1.yellow('[!]')} Uninstrumenting functions.
+"
+`);
+        }));
+        test('aborts if a problem occurs while setting the AWS credentials interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            ;
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            prompt_1.requestAWSCredentials.mockImplementation(() => Promise.reject('Unexpected error'));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"${chalk_1.bold(chalk_1.yellow('[!]'))} No existing AWS credentials found, let's set them up!
+${chalk_1.red('[Error]')} Unexpected error
+"
+`);
+        }));
+        test('aborts if there are no functions to uninstrument in the user AWS account', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env = {
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            };
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => fixtures_1.makeMockLambda({}));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"Fetching Lambda functions, this might take a while.
+${chalk_1.red('[Error]')} Couldn't find any Lambda functions in the specified region.
+"
+`);
+        }));
+        test('aborts early when the aws-sdk throws an error while uninstrumenting interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env = {
+                [constants_1.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_1.mockAwsAccessKeyId,
+                [constants_1.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_1.mockAwsSecretAccessKey,
+                [constants_1.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+            };
+            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+            aws_sdk_1.Lambda.mockImplementation(() => ({
+                listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('Lambda failed') })),
+            }));
+            const cli = fixtures_1.makeCli();
+            const context = fixtures_1.createMockContext();
+            const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+            const output = context.stdout.toString();
+            expect(code).toBe(1);
+            expect(output).toMatchInlineSnapshot(`
+"Fetching Lambda functions, this might take a while.
+${chalk_1.red('[Error]')} Couldn't fetch Lambda functions. Error: Max retry count exceeded.
+"
+`);
+        }));
     });
     describe('printPlannedActions', () => {
         test('prints no output when list is empty', () => {