@datadog/datadog-ci 2.3.2 → 2.4.1

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

@@ -38,577 +38,641 @@ const constants_1 = require("../constants");
 const prompt_1 = require("../prompt");
 const uninstrument_1 = require("../uninstrument");
 const fixtures_1 = require("./fixtures");
-describe('uninstrument', () => {
-    describe('execute', () => {
-        const OLD_ENV = process.env;
-        beforeEach(() => {
-            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' }));
-            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',
-                            [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',
+describe('lambda', () => {
+    describe('uninstrument', () => {
+        describe('execute', () => {
+            const OLD_ENV = process.env;
+            beforeEach(() => {
+                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' }));
+                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',
+                                [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: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',
                     },
-                    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 = fixtures_1.makeCli();
+                const context = 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\\": \\"shouldnt be deleted by uninstrumentation\\"
+              }
+            },
+            \\"Layers\\": []
+          }
+          "
+        `);
             }));
-            const cli = fixtures_1.makeCli();
-            const context = 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(`
-"\n[Dry Run] 🐶 Uninstrumenting Lambda function
-\n[!] Functions to be updated:
-\t- arn:aws:lambda:us-east-1:000000000000:function:uninstrument\n
-[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* () {
-            ;
-            fs.readFile.mockImplementation((a, b, callback) => callback({ code: 'ENOENT' }));
-            const lambda = fixtures_1.makeMockLambda({
-                'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
-                    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',
+            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' }));
+                const lambda = fixtures_1.makeMockLambda({
+                    'arn:aws:lambda:us-east-1:000000000000:function:uninstrument': {
+                        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: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',
                     },
-                    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',
-                },
-            });
-            aws_sdk_1.Lambda.mockImplementation(() => lambda);
-            const cli = fixtures_1.makeCli();
-            const context = 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(lambda.updateFunctionConfiguration).toHaveBeenCalled();
-        }));
-        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' }));
-            aws_sdk_1.Lambda.mockImplementation(() => ({
-                getFunction: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('Lambda failed') })),
+                });
+                aws_sdk_1.Lambda.mockImplementation(() => lambda);
+                const cli = fixtures_1.makeCli();
+                const context = 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(lambda.updateFunctionConfiguration).toHaveBeenCalled();
+            }));
+            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' }));
+                aws_sdk_1.Lambda.mockImplementation(() => ({
+                    getFunction: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('Lambda failed') })),
+                }));
+                process.env = {};
+                const command = 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).toMatch("[Error] Couldn't fetch Lambda functions. Lambda failed\n");
             }));
-            process.env = {};
-            const command = 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).toMatch("[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* () {
-            ;
-            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', '--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' }));
-            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'], context);
-            const output = context.stdout.toString();
-            expect(code).toBe(1);
-            expect(output).toMatchInlineSnapshot(`
-"\n🐶 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({}));
-            process.env = {};
-            const command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
-            command['config']['region'] = 'ap-southeast-1';
-            yield command['execute']();
-            const output = command.context.stdout.toString();
-            expect(output).toMatchInlineSnapshot(`
-"\n🐶 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({}));
-            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(() => ({
-                listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('ListFunctionsError') })),
+            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' }));
+                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', '--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');
             }));
-            process.env = {};
-            aws_sdk_1.Lambda.mockImplementation(() => ({
-                listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('ListFunctionsError') })),
+            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' }));
+                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'], context);
+                const output = context.stdout.toString();
+                expect(code).toBe(1);
+                expect(output).toMatchInlineSnapshot(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] No functions specified to remove instrumentation.
+          "
+        `);
             }));
-            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("\n[Error] Couldn't fetch Lambda functions. Error: Max retry count exceeded. ListFunctionsError\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',
+            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({}));
+                process.env = {};
+                const command = 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({}));
+                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(() => ({
+                    listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('ListFunctionsError') })),
+                }));
+                process.env = {};
+                aws_sdk_1.Lambda.mockImplementation(() => ({
+                    listFunctions: jest.fn().mockImplementation(() => ({ promise: () => Promise.reject('ListFunctionsError') })),
+                }));
+                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("\n[Error] Couldn't fetch Lambda functions. Error: Max retry count exceeded. ListFunctionsError\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',
                     },
-                    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',
+                    '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',
                     },
-                    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).toMatchSnapshot();
             }));
-            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(`
-"\n🐶 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.
-\n[!] Functions to be updated:
-\t- arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
-\t- 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\\": []
-}
-[!] Confirmation needed.
-[!] 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',
+            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',
                     },
-                    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',
+                    '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',
                     },
-                    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).toMatchSnapshot();
             }));
-            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(`
-"\n🐶 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.\n
-[!] Functions to be updated:
-\t- arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world
-\t- 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\\": []
-}
-[!] Confirmation needed.
-[!] 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(`
-"\n🐶 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* () {
-            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(`
-"\n🐶 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* () {
-            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('ListFunctionsError') })),
+            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(`
+          "
+          🐶 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* () {
+                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(`
+          "
+          🐶 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* () {
+                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('ListFunctionsError') })),
+                }));
+                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(`
+          "
+          🐶 Uninstrumenting Lambda function
+          [Error] Couldn't fetch Lambda functions. Error: Max retry count exceeded. ListFunctionsError
+          "
+        `);
+            }));
+            test('prints error when updating aws profile credentials fails', () => __awaiter(void 0, void 0, void 0, function* () {
+                ;
+                aws_sdk_1.SharedIniFileCredentials.mockImplementation(() => {
+                    throw Error('Update failed!');
+                });
+                const cli = fixtures_1.makeCli();
+                const context = 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',
+                ];
+                aws_sdk_1.Lambda.mockImplementation(() => (Object.assign(Object.assign({}, fixtures_1.makeMockLambda({
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1': {
+                        FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                        FunctionName: 'lambda-1-us-east-1',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs12.x',
+                    },
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1': {
+                        FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-2-us-east-1',
+                        FunctionName: 'lambda-2-us-east-1',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs12.x',
+                    },
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-3-us-east-1': {
+                        FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-3-us-east-1',
+                        FunctionName: 'lambda-3-us-east-1',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs12.x',
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2': {
+                        FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                        FunctionName: 'lambda-1-us-east-2',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs14.x',
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-2-us-east-2': {
+                        FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-2-us-east-2',
+                        FunctionName: 'lambda-2-us-east-2',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs16.x',
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-3-us-east-2': {
+                        FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-3-us-east-2',
+                        FunctionName: 'lambda-3-us-east-2',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs18.x',
+                    },
+                })), { updateFunctionConfiguration: jest.fn().mockImplementation((updateRequest) => {
+                        if (failingLambdas.includes(updateRequest['FunctionName'])) {
+                            return { promise: () => Promise.reject(Error('Unexpected error updating request')) };
+                        }
+                        return { promise: () => Promise.resolve() };
+                    }) })));
+                const cli = fixtures_1.makeCli();
+                const context = 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();
             }));
-            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(`
-"\n🐶 Uninstrumenting Lambda function
-[Error] Couldn't fetch Lambda functions. Error: Max retry count exceeded. ListFunctionsError
-"
-`);
-        }));
-        test('prints error when updating aws profile credentials fails', () => __awaiter(void 0, void 0, void 0, function* () {
-            ;
-            aws_sdk_1.SharedIniFileCredentials.mockImplementation(() => {
-                throw Error('Update failed!');
+            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',
+                ];
+                aws_sdk_1.Lambda.mockImplementation(() => (Object.assign(Object.assign({}, fixtures_1.makeMockLambda({
+                    'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1': {
+                        FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-1-us-east-1',
+                        FunctionName: 'lambda-1-us-east-1',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs12.x',
+                    },
+                    'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2': {
+                        FunctionArn: 'arn:aws:lambda:us-east-2:123456789012:function:lambda-1-us-east-2',
+                        FunctionName: 'lambda-1-us-east-2',
+                        Handler: 'index.handler',
+                        Runtime: 'nodejs14.x',
+                    },
+                })), { updateFunctionConfiguration: jest.fn().mockImplementation((updateRequest) => {
+                        if (failingLambdas.includes(updateRequest['FunctionName'])) {
+                            return { promise: () => Promise.reject(Error('Unexpected error updating request')) };
+                        }
+                        return { promise: () => Promise.resolve() };
+                    }) })));
+                const cli = fixtures_1.makeCli();
+                const context = 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 = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
+                command['printPlannedActions']([]);
+                const output = command.context.stdout.toString();
+                expect(output).toMatchInlineSnapshot(`
+                   "
+                   No updates will be applied.
+                   "
+                `);
             });
-            const cli = fixtures_1.makeCli();
-            const context = 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(`
-"\n🐶 Uninstrumenting Lambda function
-[Error] Error: Couldn't set AWS profile credentials. Update failed!
-"
-`);
-        }));
-    });
-    describe('printPlannedActions', () => {
-        test('prints no output when list is empty', () => {
-            process.env = {};
-            const command = fixtures_1.createCommand(uninstrument_1.UninstrumentCommand);
-            command['printPlannedActions']([]);
-            const output = command.context.stdout.toString();
-            expect(output).toMatchInlineSnapshot(`
-       "
-       No updates will be applied.
-       "
-      `);
         });
     });
 });