@datadog/datadog-ci 2.40.0 → 2.40.2

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

@@ -0,0 +1,722 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+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('fs');
+jest.mock('@aws-sdk/credential-providers', () => (Object.assign(Object.assign({}, jest.requireActual('@aws-sdk/credential-providers')), { fromIni: jest.fn() })));
+jest.mock('../prompt');
+jest.mock('../renderers/instrument-uninstrument-renderer');
+jest.mock('../../../helpers/prompt');
+jest.mock('../../../../package.json', () => ({ version: 'XXXX' }));
+const fs = __importStar(require("fs"));
+const client_lambda_1 = require("@aws-sdk/client-lambda");
+const credential_providers_1 = require("@aws-sdk/credential-providers");
+const aws_sdk_client_mock_1 = require("aws-sdk-client-mock");
+const constants_1 = require("../../../constants");
+const fixtures_1 = require("../../../helpers/__tests__/fixtures");
+const prompt_1 = require("../../../helpers/prompt");
+require("aws-sdk-client-mock-jest");
+const constants_2 = require("../constants");
+const prompt_2 = require("../prompt");
+const uninstrument_1 = require("../uninstrument");
+const fixtures_2 = require("./fixtures");
+describe('lambda', () => {
+    const lambdaClientMock = (0, aws_sdk_client_mock_1.mockClient)(client_lambda_1.LambdaClient);
+    describe('uninstrument', () => {
+        describe('execute', () => {
+            const OLD_ENV = process.env;
+            beforeEach(() => {
+                lambdaClientMock.reset();
+                (0, fixtures_2.mockLambdaClientCommands)(lambdaClientMock);
+                jest.resetModules();
+                process.env = {};
+            });
+            afterAll(() => {
+                process.env = OLD_ENV;
+            });
+            test('prints dry run data for a valid uninstrumentation', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
+                        config: {
+                            Architectures: ['x86_64'],
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.TRACE_ENABLED_ENV_VAR]: 'true',
+                                    [constants_1.VERSION_ENV_VAR]: '0.2',
+                                    [constants_2.APM_FLUSH_DEADLINE_MILLISECONDS_ENV_VAR]: '42',
+                                    USER_VARIABLE: 'shouldnt be deleted by uninstrumentation',
+                                },
+                            },
+                            FunctionArn: 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument',
+                            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',
+                        },
+                    },
+                });
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const functionARN = 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument';
+                process.env.DATADOG_API_KEY = '1234';
+                const code = yield cli.run(['lambda', 'uninstrument', '-f', functionARN, '-r', 'us-east-1', '-d'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(0);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          [Dry Run] 🐶 Uninstrumenting Lambda function
+
+          [!] Functions to be updated:
+          	- arn:aws:lambda:us-east-1:000000000000:function:uninstrument
+
+          [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": "sh**********tion"
+              }
+            },
+            "Layers": []
+          }
+          "
+        `);
+            }));
+            test('runs function update command for valid uninstrumentation', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
+                        config: {
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.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:uninstrument',
+                            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',
+                        },
+                    },
+                });
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const functionARN = 'arn:aws:lambda:us-east-1:000000000000:function:uninstrument';
+                process.env.DATADOG_API_KEY = '1234';
+                yield cli.run(['lambda', 'uninstrument', '-f', functionARN, '-r', 'us-east-1'], context);
+                expect(lambdaClientMock).toHaveReceivedCommand(client_lambda_1.UpdateFunctionConfigurationCommand);
+            }));
+            test('aborts early when the aws-sdk throws an error', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                lambdaClientMock.on(client_lambda_1.GetFunctionCommand).rejects('Lambda Failed');
+                process.env = {};
+                const command = (0, fixtures_1.createCommand)(uninstrument_1.UninstrumentCommand);
+                command['functions'] = ['my-func'];
+                command['region'] = 'us-east-1';
+                const code = yield command['execute']();
+                const output = command.context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] Couldn't fetch Lambda functions. Error: Lambda Failed
+          "
+        `);
+            }));
+            test("aborts early when function regions can't be found", () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument', '--function', 'my-func'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatch('No default region specified for ["my-func"]. Use -r, --region, or use a full functionARN');
+            }));
+            test('aborts early when no functions are specified', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] No functions specified to remove instrumentation.
+          "
+        `);
+            }));
+            test('aborts early when no functions are specified while using config file', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                process.env = {};
+                const command = (0, fixtures_1.createCommand)(uninstrument_1.UninstrumentCommand);
+                command['config']['region'] = 'ap-southeast-1';
+                yield command['execute']();
+                const output = command.context.stdout.toString();
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] No functions specified to remove 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({ code: 'ENOENT' }));
+                process.env = {};
+                let command = (0, 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 = (0, 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({ code: 'ENOENT' }));
+                process.env = {};
+                const command = (0, 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({ code: 'ENOENT' }));
+                process.env = {};
+                const command = (0, 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. [-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({ code: 'ENOENT' }));
+                process.env = {};
+                lambdaClientMock.on(client_lambda_1.ListFunctionsCommand).rejects('ListFunctionsError');
+                const command = (0, 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).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] Couldn't fetch Lambda functions. Error: ListFunctionsError
+          "
+        `);
+            }));
+            test('uninstrument multiple functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                        config: {
+                            Architectures: ['x86_64'],
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.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': {
+                        config: {
+                            Architectures: ['x86_64'],
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.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_2.requestAWSCredentials.mockImplementation(() => {
+                    process.env[constants_2.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_2.mockAwsAccessKeyId;
+                    process.env[constants_2.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_2.mockAwsSecretAccessKey;
+                    process.env[constants_2.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+                });
+                prompt_2.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.requestConfirmation.mockImplementation(() => true);
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(0);
+                expect(output).toMatchSnapshot();
+            }));
+            test('uninstrument multiple specified functions interactively', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world': {
+                        config: {
+                            Architectures: ['x86_64'],
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.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': {
+                        config: {
+                            Architectures: ['x86_64'],
+                            Environment: {
+                                Variables: {
+                                    [constants_1.ENVIRONMENT_ENV_VAR]: 'staging',
+                                    [constants_2.FLUSH_TO_LOG_ENV_VAR]: 'true',
+                                    [constants_2.LAMBDA_HANDLER_ENV_VAR]: 'lambda_function.lambda_handler',
+                                    [constants_2.LOG_LEVEL_ENV_VAR]: 'debug',
+                                    [constants_2.MERGE_XRAY_TRACES_ENV_VAR]: 'false',
+                                    [constants_1.SERVICE_ENV_VAR]: 'middletier',
+                                    [constants_1.SITE_ENV_VAR]: 'datadoghq.com',
+                                    [constants_2.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_2.requestAWSCredentials.mockImplementation(() => {
+                    process.env[constants_2.AWS_ACCESS_KEY_ID_ENV_VAR] = fixtures_2.mockAwsAccessKeyId;
+                    process.env[constants_2.AWS_SECRET_ACCESS_KEY_ENV_VAR] = fixtures_2.mockAwsSecretAccessKey;
+                    process.env[constants_2.AWS_DEFAULT_REGION_ENV_VAR] = 'sa-east-1';
+                });
+                prompt_2.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.requestConfirmation.mockImplementation(() => true);
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, 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).toMatchSnapshot();
+            }));
+            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_2.requestAWSCredentials.mockImplementation(() => Promise.reject('Unexpected error'));
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [!] No AWS credentials found, let's set them up! Or you can re-run the command and supply the AWS credentials in the same way when you invoke the AWS CLI.
+          [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* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                process.env = {
+                    [constants_2.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_2.mockAwsAccessKeyId,
+                    [constants_2.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_2.mockAwsSecretAccessKey,
+                    [constants_2.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+                };
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [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* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                process.env = {
+                    [constants_2.AWS_ACCESS_KEY_ID_ENV_VAR]: fixtures_2.mockAwsAccessKeyId,
+                    [constants_2.AWS_SECRET_ACCESS_KEY_ENV_VAR]: fixtures_2.mockAwsSecretAccessKey,
+                    [constants_2.AWS_DEFAULT_REGION_ENV_VAR]: 'sa-east-1',
+                };
+                lambdaClientMock.on(client_lambda_1.ListFunctionsCommand).rejects('ListFunctionsError');
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run(['lambda', 'uninstrument', '-i'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] Couldn't fetch Lambda functions. Error: ListFunctionsError
+          "
+        `);
+            }));
+            test('prints error when updating aws profile credentials fails', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                credential_providers_1.fromIni.mockImplementation(() => {
+                    throw Error('Update failed!');
+                });
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const functionARN = 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world';
+                const code = yield cli.run(['lambda', 'uninstrument', '-f', functionARN, '--profile', 'SOME-AWS-PROFILE'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] Error: Couldn't set AWS profile credentials. Update failed!
+          "
+        `);
+            }));
+            test('prints which functions failed to uninstrument without aborting when at least one function was uninstrumented correctly', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                const failingLambdas = [
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                ];
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                            FunctionName: 'lambda-1-us-east-1',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs16.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1',
+                            FunctionName: 'lambda-2-us-east-1',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs16.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-3-us-east-1': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-3-us-east-1',
+                            FunctionName: 'lambda-3-us-east-1',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs16.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                            FunctionName: 'lambda-1-us-east-2',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs18.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-2-us-east-2': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-2-us-east-2',
+                            FunctionName: 'lambda-2-us-east-2',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs18.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-3-us-east-2': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-3-us-east-2',
+                            FunctionName: 'lambda-3-us-east-2',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs20.x',
+                        },
+                    },
+                });
+                for (const failingLambda of failingLambdas) {
+                    lambdaClientMock
+                        .on(client_lambda_1.UpdateFunctionConfigurationCommand, { FunctionName: failingLambda })
+                        .rejects('Unexpected error updating request');
+                }
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run([
+                    'lambda',
+                    'instrument',
+                    '-f',
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                    '-f',
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1',
+                    '-f',
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-3-us-east-1',
+                    '-f',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                    '-f',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-2-us-east-2',
+                    '-f',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-3-us-east-2',
+                ], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(0);
+                expect(output).toMatchSnapshot();
+            }));
+            test('aborts when every lambda function fails to update on uninstrument', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
+                const failingLambdas = [
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                ];
+                (0, fixtures_2.mockLambdaConfigurations)(lambdaClientMock, {
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                            FunctionName: 'lambda-1-us-east-1',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs16.x',
+                        },
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2': {
+                        config: {
+                            FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                            FunctionName: 'lambda-1-us-east-2',
+                            Handler: 'index.handler',
+                            Runtime: 'nodejs18.x',
+                        },
+                    },
+                });
+                for (const failingLambda of failingLambdas) {
+                    lambdaClientMock
+                        .on(client_lambda_1.UpdateFunctionConfigurationCommand, { FunctionName: failingLambda })
+                        .rejects('Unexpected error updating request');
+                }
+                const cli = (0, fixtures_2.makeCli)();
+                const context = (0, fixtures_1.createMockContext)();
+                const code = yield cli.run([
+                    'lambda',
+                    'instrument',
+                    '-f',
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                    '-f',
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                ], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchSnapshot();
+            }));
+        });
+        describe('printPlannedActions', () => {
+            test('prints no output when list is empty', () => {
+                process.env = {};
+                const command = (0, fixtures_1.createCommand)(uninstrument_1.UninstrumentCommand);
+                command['printPlannedActions']([]);
+                const output = command.context.stdout.toString();
+                expect(output).toMatchInlineSnapshot(`
+                   "
+                   No updates will be applied.
+                   "
+                `);
+            });
+        });
+    });
+});
+//# sourceMappingURL=uninstrument.test.js.map