feat(function): add go, python runtimes to defineFunction

Author: MarlonJD

package-lock.json

@@ -21,6 +21,8 @@
     "@aws-amplify/backend-output-schemas": "^1.1.0",
     "@aws-amplify/backend-output-storage": "^1.0.1",
     "@aws-amplify/plugin-types": "^1.0.0",
+    "@aws-cdk/aws-lambda-go-alpha": "^2.141.0-alpha.0",
+    "@aws-cdk/aws-lambda-python-alpha": "^2.144.0-alpha.0",
     "execa": "^8.0.1"
   },
   "devDependencies": {

packages/backend-function/package.json

@@ -13,7 +13,7 @@ import {
 } from '@aws-amplify/backend-platform-test-stubs';
 import { defaultLambda } from './test-assets/default-lambda/resource.js';
 import { Template } from 'aws-cdk-lib/assertions';
-import { NodeVersion, defineFunction } from './factory.js';
+import { RuntimeType, defineFunction } from './factory.js';
 import { lambdaWithDependencies } from './test-assets/lambda-with-dependencies/resource.js';
 import { Runtime } from 'aws-cdk-lib/aws-lambda';
 import { Policy, PolicyStatement } from 'aws-cdk-lib/aws-iam';
@@ -273,7 +273,7 @@ void describe('AmplifyFunctionFactory', () => {
     void it('sets valid runtime', () => {
       const lambda = defineFunction({
         entry: './test-assets/default-lambda/handler.ts',
-        runtime: 16,
+        runtime: 'NODEJS_16_X',
       }).getInstance(getInstanceProps);
       const template = Template.fromStack(Stack.of(lambda.resources.lambda));
 
@@ -282,6 +282,78 @@ void describe('AmplifyFunctionFactory', () => {
       });
     });
 
+    void it('sets valid go runtime', () => {
+      const lambda = defineFunction({
+        entry: './test-assets/golang-lambda/',
+        runtime: 'GO_1_X_PROVIDED_AL2023',
+      }).getInstance(getInstanceProps);
+      const template = Template.fromStack(Stack.of(lambda.resources.lambda));
+
+      template.hasResourceProperties('AWS::Lambda::Function', {
+        Runtime: Runtime.PROVIDED_AL2023.name,
+      });
+    });
+
+    void it('sets not valid go runtime', () => {
+      assert.throws(
+        () =>
+          defineFunction({
+            entry: './test-assets/golang-lambda/',
+            runtime: 'GO_1_X' as RuntimeType,
+          }).getInstance(getInstanceProps),
+        new Error(
+          'runtime must be one of the following: NODEJS_16_X, NODEJS_18_X, NODEJS_20_X, GO_1_X_PROVIDED_AL2023, PYTHON_3_8, PYTHON_3_9, PYTHON_3_10, PYTHON_3_11, PYTHON_3_12'
+        )
+      );
+    });
+
+    void it('sets not valid go runtime, entry empty', () => {
+      assert.throws(
+        () =>
+          defineFunction({
+            runtime: 'GO_1_X_PROVIDED_AL2023',
+          }).getInstance(getInstanceProps),
+        new Error(
+          'entry must be set for custom runtime "GO_1_X_PROVIDED_AL2023"'
+        )
+      );
+    });
+
+    void it('sets valid python runtime', () => {
+      const lambda = defineFunction({
+        entry: './test-assets/python-lambda/',
+        runtime: 'PYTHON_3_8',
+      }).getInstance(getInstanceProps);
+      const template = Template.fromStack(Stack.of(lambda.resources.lambda));
+
+      template.hasResourceProperties('AWS::Lambda::Function', {
+        Runtime: Runtime.PYTHON_3_8.name,
+      });
+    });
+
+    void it('sets not valid python runtime not valid runtime version', () => {
+      assert.throws(
+        () =>
+          defineFunction({
+            entry: './test-assets/python-lambda/',
+            runtime: 'PYTHON_3_7' as RuntimeType,
+          }).getInstance(getInstanceProps),
+        new Error(
+          'runtime must be one of the following: NODEJS_16_X, NODEJS_18_X, NODEJS_20_X, GO_1_X_PROVIDED_AL2023, PYTHON_3_8, PYTHON_3_9, PYTHON_3_10, PYTHON_3_11, PYTHON_3_12'
+        )
+      );
+    });
+
+    void it('sets not valid python runtime empty entry', () => {
+      assert.throws(
+        () =>
+          defineFunction({
+            runtime: 'PYTHON_3_8',
+          }).getInstance(getInstanceProps),
+        new Error('entry must be set for custom runtime "PYTHON_3_8"')
+      );
+    });
+
     void it('defaults to oldest LTS runtime', () => {
       const lambda = defineFunction({
         entry: './test-assets/default-lambda/handler.ts',
@@ -298,9 +370,11 @@ void describe('AmplifyFunctionFactory', () => {
         () =>
           defineFunction({
             entry: './test-assets/default-lambda/handler.ts',
-            runtime: 14 as NodeVersion,
+            runtime: 'NODEJS_14_X' as RuntimeType,
           }).getInstance(getInstanceProps),
-        new Error('runtime must be one of the following: 16, 18, 20')
+        new Error(
+          'runtime must be one of the following: NODEJS_16_X, NODEJS_18_X, NODEJS_20_X, GO_1_X_PROVIDED_AL2023, PYTHON_3_8, PYTHON_3_9, PYTHON_3_10, PYTHON_3_11, PYTHON_3_12'
+        )
       );
     });