Added lambda invoke action changes

Author: sasidhar-aws

.changelog/45170.txt

@@ -0,0 +1,19 @@
+```release-note:enhancement
+resource/aws_lambda_function: Add `tenancy_config` argument
+```
+
+```release-note:enhancement
+data-source/aws_lambda_function: Add `tenancy_config` attribute
+```
+
+```release-note:enhancement
+resource/aws_lambda_invocation: Add `tenant_id` argument
+```
+
+```release-note:enhancement
+data-source/aws_lambda_invocation: Add `tenant_id` argument
+```
+
+```release-note:enhancement
+action/aws_lambda_invoke: Add `tenant_id` argument
+```

internal/service/lambda/invoke_action.go

@@ -43,6 +43,7 @@ type invokeActionModel struct {
 	InvocationType fwtypes.StringEnum[awstypes.InvocationType] `tfsdk:"invocation_type"`
 	LogType        fwtypes.StringEnum[awstypes.LogType]        `tfsdk:"log_type"`
 	ClientContext  types.String                                `tfsdk:"client_context"`
+	TenantId       types.String                                `tfsdk:"tenant_id"`
 }
 
 func (a *invokeAction) Schema(ctx context.Context, req action.SchemaRequest, resp *action.SchemaResponse) {
@@ -78,6 +79,10 @@ func (a *invokeAction) Schema(ctx context.Context, req action.SchemaRequest, res
 				Description: "Up to 3,583 bytes of base64-encoded data about the invoking client to pass to the function in the context object. This is only used for mobile applications.",
 				Optional:    true,
 			},
+			"tenant_id": schema.StringAttribute{
+				Description: "The Tenant Id for lambda function invocation. This is mandatory, if tenancy_config is enabled in lambda function",
+				Optional:    true,
+			},
 		},
 	}
 }
@@ -134,6 +139,10 @@ func (a *invokeAction) Invoke(ctx context.Context, req action.InvokeRequest, res
 	if !config.Qualifier.IsNull() {
 		input.Qualifier = config.Qualifier.ValueStringPointer()
 	}
+	// Set optional parameters
+	if !config.TenantId.IsNull() {
+		input.TenantId = config.TenantId.ValueStringPointer()
+	}
 
 	if !config.ClientContext.IsNull() {
 		clientContext := config.ClientContext.ValueString()

internal/service/lambda/invoke_action_test.go

@@ -211,6 +211,35 @@ func TestAccLambdaInvokeAction_complexPayload(t *testing.T) {
 	})
 }
 
+func TestAccLambdaInvokeAction_tenantId(t *testing.T) {
+	ctx := acctest.Context(t)
+	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+	testData := "value3"
+	inputJSON := `{"key1":"value1","key2":"value2"}`
+	expectedResult := fmt.Sprintf(`{"key1":"value1","key2":"value2","key3":%q}`, testData)
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			acctest.PreCheck(ctx, t)
+			acctest.PreCheckPartitionHasService(t, names.LambdaEndpointID)
+		},
+		ErrorCheck:               acctest.ErrorCheck(t, names.LambdaServiceID),
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+		TerraformVersionChecks: []tfversion.TerraformVersionCheck{
+			tfversion.SkipBelow(tfversion.Version1_14_0),
+		},
+		CheckDestroy: acctest.CheckDestroyNoop,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccInvokeActionConfig_tenantId(rName, testData, inputJSON),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckInvokeAction(ctx, rName, inputJSON, expectedResult),
+				),
+			},
+		},
+	})
+}
+
 // Test helper functions
 
 // testAccCheckInvokeAction verifies that the action can successfully invoke a Lambda function
@@ -559,3 +588,52 @@ resource "terraform_data" "trigger" {
 }
 `, inputJSON, clientContext))
 }
+
+func testAccInvokeActionConfig_function_tenant(rName, testData string) string {
+	return acctest.ConfigCompose(
+		testAccInvokeActionConfig_base(rName),
+		fmt.Sprintf(`
+resource "aws_lambda_function" "test" {
+  depends_on = [aws_iam_role_policy_attachment.test]
+
+  filename      = "test-fixtures/lambda_invocation.zip"
+  function_name = %[1]q
+  role          = aws_iam_role.test.arn
+  handler       = "lambda_invocation.handler"
+  runtime       = "nodejs18.x"
+
+  tenancy_config {
+    tenant_isolation_mode = "PER_TENANT"
+  }
+  environment {
+    variables = {
+      TEST_DATA = %[2]q
+    }
+  }
+}
+`, rName, testData))
+}
+
+func testAccInvokeActionConfig_tenantId(rName, testData, inputJSON string) string {
+	return acctest.ConfigCompose(
+		testAccInvokeActionConfig_function_tenant(rName, testData),
+		fmt.Sprintf(`
+action "aws_lambda_invoke" "test" {
+  config {
+    function_name = aws_lambda_function.test.function_name
+    payload       = %[1]q
+	tenant_id     = "tenant-1"
+  }
+}
+
+resource "terraform_data" "trigger" {
+  input = "trigger"
+  lifecycle {
+    action_trigger {
+      events  = [before_create, before_update]
+      actions = [action.aws_lambda_invoke.test]
+    }
+  }
+}
+`, inputJSON))
+}

website/docs/actions/lambda_invoke.html.markdown

@@ -220,3 +220,4 @@ This action supports the following arguments:
 * `payload` - (Required) JSON payload to send to the Lambda function. This should be a valid JSON string that represents the event data for your function. The payload size limit is 6 MB for synchronous invocations and 256 KB for asynchronous invocations.
 * `region` - (Optional) Region where this action should be [run](https://docs.aws.amazon.com/general/latest/gr/rande.html#regional-endpoints). Defaults to the Region set in the [provider configuration](https://registry.terraform.io/providers/hashicorp/aws/latest/docs#aws-configuration-reference).
 * `qualifier` - (Optional) Version or alias of the Lambda function to invoke. If not specified, the `$LATEST` version will be invoked. Can be a version number (e.g., `1`) or an alias (e.g., `PROD`).
+* `tenant_id` - (Optional)  Tenant Id to serve invocations from specified tenant.