@datadog/datadog-ci 2.40.1 → 2.40.3

package/dist/commands/lambda/__tests__/functions/instrument.part2.test.js

@@ -0,0 +1,721 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+jest.mock('../../loggroup');
+const client_lambda_1 = require("@aws-sdk/client-lambda");
+const aws_sdk_client_mock_1 = require("aws-sdk-client-mock");
+const constants_1 = require("../../../../constants");
+const fixtures_1 = require("../../../../helpers/__tests__/fixtures");
+const constants_2 = require("../../constants");
+const instrument_1 = require("../../functions/instrument");
+const fixtures_2 = require("../fixtures");
+describe('instrument', () => {
+    describe('calculateUpdateRequest', () => {
+        const OLD_ENV = process.env;
+        beforeEach(() => {
+            jest.resetModules();
+            process.env = {};
+        });
+        afterAll(() => {
+            process.env = OLD_ENV;
+        });
+        test('calculates an update request with just lambda library layers', () => __awaiter(void 0, void 0, void 0, function* () {
+            const runtime = client_lambda_1.Runtime.nodejs20x;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_FLUSH_TO_LOG": "false",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "/opt/nodejs/node_modules/datadog-lambda-js/handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Node20-x:5",
+          ],
+        }
+      `);
+        }));
+        test.each([
+            [client_lambda_1.Runtime.python310, 'Datadog-Python310'],
+            [client_lambda_1.Runtime.python311, 'Datadog-Python311'],
+            [client_lambda_1.Runtime.python312, 'Datadog-Python312'],
+        ])('calculates an update request for %s', (runtime, layer) => __awaiter(void 0, void 0, void 0, function* () {
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 71,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_FLUSH_TO_LOG": "false",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "datadog_lambda.handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:${layer}:71",
+          ],
+        }
+      `);
+        }));
+        test('calculates an update request with just lambda library layers in arm architecture', () => __awaiter(void 0, void 0, void 0, function* () {
+            const runtime = client_lambda_1.Runtime.python39;
+            const config = {
+                Architectures: [client_lambda_1.Architecture.arm64],
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'handler.hello',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 11,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_FLUSH_TO_LOG": "false",
+              "DD_LAMBDA_HANDLER": "handler.hello",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "datadog_lambda.handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Python39-ARM:11",
+          ],
+        }
+      `);
+        }));
+        test('calculates an update request with a lambda library, extension, and DATADOG_API_KEY', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_1.CI_API_KEY_ENV_VAR] = fixtures_1.MOCK_DATADOG_API_KEY;
+            const runtime = client_lambda_1.Runtime.nodejs20x;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                extensionVersion: 6,
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_API_KEY": "02aeb762fff59ac0d5ad1536cd9633bd",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "/opt/nodejs/node_modules/datadog-lambda-js/handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Extension:6",
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Node20-x:5",
+          ],
+        }
+      `);
+        }));
+        test('calculates an update request with a lambda library, extension, and DD_API_KEY', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_1.API_KEY_ENV_VAR] = 'SOME-DD-API-KEY';
+            const runtime = client_lambda_1.Runtime.nodejs20x;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                extensionVersion: 6,
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_API_KEY": "SOME-DD-API-KEY",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "/opt/nodejs/node_modules/datadog-lambda-js/handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Extension:6",
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Node20-x:5",
+          ],
+        }
+      `);
+        }));
+        test('calculates an update request with a lambda library, extension, and DATADOG_API_KEY_SECRET_ARN', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_2.CI_API_KEY_SECRET_ARN_ENV_VAR] = 'some-secret:arn:from:aws';
+            const runtime = client_lambda_1.Runtime.python39;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                extensionVersion: 11,
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 49,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_API_KEY_SECRET_ARN": "some-secret:arn:from:aws",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "datadog_lambda.handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Extension:11",
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Python39:49",
+          ],
+        }
+      `);
+        }));
+        test('calculates an update request with a lambda library, extension, and DATADOG_KMS_API_KEY', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+            const runtime = client_lambda_1.Runtime.python39;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                extensionVersion: 6,
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_KMS_API_KEY": "5678",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+          "Handler": "datadog_lambda.handler.handler",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Extension:6",
+            "arn:aws:lambda:sa-east-1:123456789012:layer:Datadog-Python39:5",
+          ],
+        }
+      `);
+        }));
+        test('prioritizes the KMS API KEY when all of them are exported', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env = {
+                [constants_1.CI_API_KEY_ENV_VAR]: fixtures_1.MOCK_DATADOG_API_KEY,
+                [constants_2.CI_API_KEY_SECRET_ARN_ENV_VAR]: '5678',
+                [constants_2.CI_KMS_API_KEY_ENV_VAR]: 'should-be-selected',
+            };
+            const config = {
+                FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+            };
+            const runtime = 'python3.9';
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, {}, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_KMS_API_KEY": "should-be-selected",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_SITE": "datadoghq.com",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world",
+        }
+      `);
+        }));
+        test("doesn't set DD_FLUSH_TO_LOGS when extension is being used", () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_1.CI_API_KEY_ENV_VAR] = fixtures_1.MOCK_DATADOG_API_KEY;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+            };
+            const runtime = 'python3.9';
+            const region = 'sa-east-1';
+            const settings = {
+                extensionVersion: 13,
+                flushMetricsToLogs: true,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_API_KEY": "02aeb762fff59ac0d5ad1536cd9633bd",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:sa-east-1:123456789012:function:lambda-hello-world",
+          "Layers": [
+            "arn:aws:lambda:sa-east-1:464622532012:layer:Datadog-Extension:13",
+          ],
+        }
+      `);
+        }));
+        test('by default calculates an update request with DATADOG_SITE being set to datadoghq.com', () => __awaiter(void 0, void 0, void 0, function* () {
+            const runtime = client_lambda_1.Runtime.python39;
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_FLUSH_TO_LOG": "false",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.com",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+        }
+      `);
+        }));
+        test('calculates an update request with DATADOG_SITE being set to datadoghq.eu', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env.DATADOG_SITE = 'datadoghq.eu';
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'sa-east-1';
+            const runtime = 'python3.9';
+            const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            expect(updateRequest).toMatchInlineSnapshot(`
+        {
+          "Environment": {
+            "Variables": {
+              "DD_FLUSH_TO_LOG": "false",
+              "DD_LAMBDA_HANDLER": "index.handler",
+              "DD_MERGE_XRAY_TRACES": "false",
+              "DD_SITE": "datadoghq.eu",
+              "DD_TRACE_ENABLED": "false",
+            },
+          },
+          "FunctionName": "arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world",
+        }
+      `);
+        }));
+        test('throws an error when an invalid DATADOG_SITE url is given', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env.DATADOG_SITE = 'datacathq.eu';
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                interactive: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'us-east-1';
+            const runtime = 'python3.9';
+            let error;
+            try {
+                yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe('Warning: Invalid site URL. Must be either datadoghq.com, datadoghq.eu, us3.datadoghq.com, us5.datadoghq.com, ap1.datadoghq.com, or ddog-gov.com.');
+        }));
+        test('throws an error when neither DATADOG_API_KEY nor DATADOG_KMS_API_KEY are given through the environment while using extensionVersion', () => __awaiter(void 0, void 0, void 0, function* () {
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+            };
+            const settings = {
+                extensionVersion: 6,
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 5,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const region = 'us-east-1';
+            const runtime = 'python3.9';
+            let error;
+            try {
+                yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe("When 'extensionLayer' is set, DATADOG_API_KEY, DATADOG_KMS_API_KEY, or DATADOG_API_KEY_SECRET_ARN must also be set");
+        }));
+        test('throws error when trying to add `DD_API_KEY_SECRET_ARN` while using sync metrics in a node runtime', () => __awaiter(void 0, void 0, void 0, function* () {
+            process.env[constants_2.CI_API_KEY_SECRET_ARN_ENV_VAR] = 'some-secret:arn:from:aws';
+            const runtime = client_lambda_1.Runtime.nodejs20x;
+            const region = 'us-east-1';
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: runtime,
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 13,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            let error;
+            try {
+                yield (0, instrument_1.calculateUpdateRequest)(config, settings, region, runtime);
+            }
+            catch (e) {
+                if (e instanceof Error) {
+                    error = e;
+                }
+            }
+            expect(error === null || error === void 0 ? void 0 : error.message).toBe('`apiKeySecretArn` is not supported for Node runtimes when using Synchronous Metrics. Use either `apiKey` or `apiKmsKey`.');
+        }));
+        describe('test universal instrumentation workflow for Java and .Net', () => {
+            const region = 'us-east-1';
+            const config = {
+                FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                Handler: 'index.handler',
+                Layers: [],
+                Runtime: 'runtime',
+            };
+            const settings = {
+                flushMetricsToLogs: false,
+                layerAWSAccount: fixtures_2.mockAwsAccount,
+                layerVersion: 13,
+                mergeXrayTraces: false,
+                tracingEnabled: false,
+            };
+            const dotnetRuntime = client_lambda_1.Runtime.dotnet6;
+            const javaRuntime = client_lambda_1.Runtime.java11;
+            describe(`test for runtime ${dotnetRuntime}`, () => {
+                const dotNetConfig = Object.assign(Object.assign({}, config), { Runtime: dotnetRuntime });
+                const dotNetConfigOnArm86 = Object.assign(Object.assign({}, config), { Runtime: dotnetRuntime, Architectures: [client_lambda_1.Architecture.arm64] });
+                test('should throw error when the extension version and trace version are not compatible', () => __awaiter(void 0, void 0, void 0, function* () {
+                    process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+                    const badSettings = Object.assign(Object.assign({}, settings), { extensionVersion: 24, layerVersion: 3 });
+                    let error;
+                    try {
+                        yield (0, instrument_1.calculateUpdateRequest)(dotNetConfig, badSettings, region, dotnetRuntime);
+                    }
+                    catch (e) {
+                        if (e instanceof Error) {
+                            error = e;
+                        }
+                    }
+                    expect(error === null || error === void 0 ? void 0 : error.message).toBe(`For the ${dotnetRuntime} runtime, the dd-trace version 3 is not compatible with the dd-extension version 24`);
+                }));
+                test('should throw error if it is running on arm64 with an old dd-extension version', () => __awaiter(void 0, void 0, void 0, function* () {
+                    process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+                    const curSettings = Object.assign(Object.assign({}, settings), { extensionVersion: 23 });
+                    let error;
+                    try {
+                        yield (0, instrument_1.calculateUpdateRequest)(dotNetConfigOnArm86, curSettings, region, dotnetRuntime);
+                    }
+                    catch (e) {
+                        if (e instanceof Error) {
+                            error = e;
+                        }
+                    }
+                    expect(error === null || error === void 0 ? void 0 : error.message).toBe('Instrumenting arm64 architecture is not supported for the given dd-extension version. Please choose the latest dd-extension version or use x86_64 architecture.');
+                }));
+                const baseVariables = {
+                    DD_KMS_API_KEY: '5678',
+                    DD_MERGE_XRAY_TRACES: 'false',
+                    DD_SITE: 'datadoghq.com',
+                    DD_TRACE_ENABLED: 'false',
+                };
+                const compatibleTracerAndExtension = {
+                    Environment: {
+                        Variables: Object.assign(Object.assign({}, baseVariables), { AWS_LAMBDA_EXEC_WRAPPER: '/opt/datadog_wrapper' }),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: [
+                        'arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:25',
+                        'arn:aws:lambda:us-east-1:123456789012:layer:dd-trace-dotnet:4',
+                    ],
+                };
+                const oldExtensionVersion = {
+                    Environment: {
+                        Variables: Object.assign(Object.assign({}, baseVariables), { CORECLR_ENABLE_PROFILING: '1', CORECLR_PROFILER: '{846F5F1C-F9AE-4B07-969E-05C26BC060D8}', CORECLR_PROFILER_PATH: '/opt/datadog/Datadog.Trace.ClrProfiler.Native.so', DD_DOTNET_TRACER_HOME: '/opt/datadog' }),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: [
+                        'arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:23',
+                        'arn:aws:lambda:us-east-1:123456789012:layer:dd-trace-dotnet:2',
+                    ],
+                };
+                const traceUndefined = {
+                    Environment: {
+                        Variables: Object.assign(Object.assign({}, baseVariables), { CORECLR_ENABLE_PROFILING: '1', CORECLR_PROFILER: '{846F5F1C-F9AE-4B07-969E-05C26BC060D8}', CORECLR_PROFILER_PATH: '/opt/datadog/Datadog.Trace.ClrProfiler.Native.so', DD_DOTNET_TRACER_HOME: '/opt/datadog' }),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: ['arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:15'],
+                };
+                test.each `
+          extensionVersion | traceVersion | outputResult
+          ${25}            | ${4}         | ${compatibleTracerAndExtension}
+          ${23}            | ${2}         | ${oldExtensionVersion}
+          ${15}            | ${undefined} | ${traceUndefined}
+        `('should the output match the expected if extensionVersion=$extensionVersion and traceVersion=$traceVersion', ({ extensionVersion, traceVersion, outputResult }) => __awaiter(void 0, void 0, void 0, function* () {
+                    const curSettings = Object.assign(Object.assign({}, settings), { extensionVersion, layerVersion: traceVersion });
+                    process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+                    const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(dotNetConfig, curSettings, region, dotnetRuntime);
+                    expect(updateRequest).toEqual(outputResult);
+                }));
+            });
+            describe(`test for runtime ${javaRuntime}`, () => {
+                const javaConfig = Object.assign(Object.assign({}, config), { Runtime: javaRuntime });
+                test('should throw error when the extension version and trace version are not compatible', () => __awaiter(void 0, void 0, void 0, function* () {
+                    process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+                    const badSettings = Object.assign(Object.assign({}, settings), { extensionVersion: 24, layerVersion: 4 });
+                    let error;
+                    try {
+                        yield (0, instrument_1.calculateUpdateRequest)(javaConfig, badSettings, region, javaRuntime);
+                    }
+                    catch (e) {
+                        if (e instanceof Error) {
+                            error = e;
+                        }
+                    }
+                    expect(error === null || error === void 0 ? void 0 : error.message).toBe(`For the ${javaRuntime} runtime, the dd-trace version 4 is not compatible with the dd-extension version 24`);
+                }));
+                const baseVariables = {
+                    DD_KMS_API_KEY: '5678',
+                    DD_MERGE_XRAY_TRACES: 'false',
+                    DD_SITE: 'datadoghq.com',
+                    DD_TRACE_ENABLED: 'false',
+                };
+                const compatibleTracerAndExtension = {
+                    Environment: {
+                        Variables: Object.assign(Object.assign({}, baseVariables), { AWS_LAMBDA_EXEC_WRAPPER: '/opt/datadog_wrapper' }),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: [
+                        'arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:25',
+                        'arn:aws:lambda:us-east-1:123456789012:layer:dd-trace-java:5',
+                    ],
+                };
+                const oldExtensionVersion = {
+                    Environment: {
+                        Variables: Object.assign({}, baseVariables),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: [
+                        'arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:23',
+                        'arn:aws:lambda:us-east-1:123456789012:layer:dd-trace-java:2',
+                    ],
+                };
+                const traceUndefined = {
+                    Environment: {
+                        Variables: Object.assign({}, baseVariables),
+                    },
+                    FunctionName: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                    Layers: ['arn:aws:lambda:us-east-1:123456789012:layer:Datadog-Extension:15'],
+                };
+                test.each `
+          extensionVersion | traceVersion | outputResult
+          ${25}            | ${5}         | ${compatibleTracerAndExtension}
+          ${23}            | ${2}         | ${oldExtensionVersion}
+          ${15}            | ${undefined} | ${traceUndefined}
+        `('should the output match the expected if extensionVersion=$extensionVersion and traceVersion=$traceVersion', ({ extensionVersion, traceVersion, outputResult }) => __awaiter(void 0, void 0, void 0, function* () {
+                    const curSettings = Object.assign(Object.assign({}, settings), { extensionVersion, layerVersion: traceVersion });
+                    process.env[constants_2.CI_KMS_API_KEY_ENV_VAR] = '5678';
+                    const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(javaConfig, curSettings, region, javaRuntime);
+                    expect(updateRequest).toEqual(outputResult);
+                }));
+            });
+        });
+        describe('sets handlers correctly', () => {
+            const lambdaClientMock = (0, aws_sdk_client_mock_1.mockClient)(client_lambda_1.LambdaClient);
+            describe('interactive mode', () => {
+                const extensionLayer = `arn:aws:lambda:us-east-1:${constants_2.DEFAULT_LAYER_AWS_ACCOUNT}:layer:Datadog-Extension`;
+                beforeAll(() => {
+                    lambdaClientMock.reset();
+                    process.env = {};
+                    (0, fixtures_2.mockLambdaClientCommands)(lambdaClientMock);
+                });
+                afterAll(() => {
+                    process.env = OLD_ENV;
+                });
+                it.each([
+                    [
+                        'python',
+                        {
+                            runtime: client_lambda_1.Runtime.python311,
+                            layer: 'Datadog-Python311',
+                        },
+                        'datadog_lambda.handler.handler',
+                    ],
+                    [
+                        'node',
+                        {
+                            runtime: client_lambda_1.Runtime.nodejs16x,
+                            layer: 'Datadog-Node16-x',
+                        },
+                        '/opt/nodejs/node_modules/datadog-lambda-js/handler.handler',
+                    ],
+                ])('%s', (_, test, expected) => __awaiter(void 0, void 0, void 0, function* () {
+                    process.env[constants_1.CI_API_KEY_ENV_VAR] = fixtures_1.MOCK_DATADOG_API_KEY;
+                    const libraryLayer = `arn:aws:lambda:us-east-1:${constants_2.DEFAULT_LAYER_AWS_ACCOUNT}:layer:${test.layer}`;
+                    // Mock Layers for interactive mode.
+                    (0, fixtures_2.mockLambdaLayers)(lambdaClientMock, {
+                        [`${libraryLayer}:1`]: {
+                            LayerName: `${libraryLayer}`,
+                            VersionNumber: 1,
+                        },
+                        [`${extensionLayer}:1`]: {
+                            LayerName: `${extensionLayer}`,
+                            VersionNumber: 1,
+                        },
+                    });
+                    const config = {
+                        FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:lambda-hello-world',
+                        Handler: 'index.handler',
+                        Layers: [],
+                        Runtime: test.runtime,
+                    };
+                    const settings = {
+                        flushMetricsToLogs: false,
+                        interactive: true,
+                        mergeXrayTraces: false,
+                        tracingEnabled: false,
+                    };
+                    const updateRequest = yield (0, instrument_1.calculateUpdateRequest)(config, settings, 'us-east-1', test.runtime);
+                    expect(updateRequest === null || updateRequest === void 0 ? void 0 : updateRequest.Handler).toBe(expected);
+                }));
+            });
+        });
+    });
+});
+//# sourceMappingURL=instrument.part2.test.js.map