vertexai: add psc_automation_configs to resource google_vertex_ai_index_endpoint (#15636)

Author: ramonvermeulen

mmv1/products/vertexai/IndexEndpoint.yaml

@@ -131,6 +131,20 @@ properties:
         immutable: true
         item_type:
           type: String
+      - name: 'pscAutomationConfigs'
+        type: Array
+        description: List of projects and networks where the PSC endpoints will be created. This field is used by Online Inference(Prediction) only.
+        item_type:
+          type: NestedObject
+          properties:
+            - name: 'projectId'
+              type: String
+              required: true
+              description: Project id used to create forwarding rule.
+            - name: 'network'
+              type: String
+              required: true
+              description: 'The full name of the Google Compute Engine [network](https://cloud.google.com/compute/docs/networks-and-firewalls#networks). [Format](https://cloud.google.com/compute/docs/reference/rest/v1/networks/get): projects/{project}/global/networks/{network}.'
   - name: 'publicEndpointEnabled'
     type: Boolean
     description: If true, the deployed index will be accessible through public endpoint.

mmv1/templates/terraform/custom_flatten/vertex_ai_index_endpoint_private_service_connect_config.go.tmpl

@@ -28,6 +28,7 @@ func flatten{{$.GetPrefix}}{{$.TitlelizeProperty}}(v interface{}, d *schema.Reso
         flattenVertexAIIndexEndpointPrivateServiceConnectConfigEnablePrivateServiceConnect(original["enablePrivateServiceConnect"], d, config)
     transformed["project_allowlist"] =
         flattenVertexAIIndexEndpointPrivateServiceConnectConfigProjectAllowlist(original["projectAllowlist"], d, config)
+    transformed["psc_automation_configs"] = d.Get("private_service_connect_config.0.psc_automation_configs")
     return []interface{}{transformed}
 }
 

mmv1/third_party/terraform/services/vertexai/resource_vertex_ai_index_endpoint_test.go

@@ -44,6 +44,31 @@ func TestAccVertexAIIndexEndpoint_updated(t *testing.T) {
 	})
 }
 
+func TestAccVertexAIIndexEndpoint_psc_automation_config(t *testing.T) {
+	t.Parallel()
+
+	context := map[string]interface{}{
+		"random_suffix": acctest.RandString(t, 10),
+	}
+
+	acctest.VcrTest(t, resource.TestCase{
+		PreCheck:                 func() { acctest.AccTestPreCheck(t) },
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t),
+		CheckDestroy:             testAccCheckVertexAIIndexEndpointDestroyProducer(t),
+		Steps: []resource.TestStep{
+			{
+				Config: testAccVertexAIIndexEndpoint_psc_automation_config(context),
+			},
+			{
+				ResourceName:            "google_vertex_ai_index_endpoint.index_endpoint",
+				ImportState:             true,
+				ImportStateVerify:       true,
+				ImportStateVerifyIgnore: []string{"etag", "region", "labels", "terraform_labels", "private_service_connect_config.0.psc_automation_configs"},
+			},
+		},
+	})
+}
+
 func testAccVertexAIIndexEndpoint_basic(context map[string]interface{}) string {
 	return acctest.Nprintf(`
 resource "google_vertex_ai_index_endpoint" "index_endpoint" {
@@ -89,3 +114,43 @@ data "google_compute_network" "vertex_network" {
 data "google_project" "project" {}
 `, context)
 }
+
+func testAccVertexAIIndexEndpoint_psc_automation_config(context map[string]interface{}) string {
+	return acctest.Nprintf(`
+resource "google_vertex_ai_index_endpoint" "index_endpoint" {
+  display_name = "tf-test-sample-endpoint-%{random_suffix}"
+  description  = "A sample vertex endpoint"
+  region       = "us-central1"
+  
+  labels = {
+    label-one = "value-one"
+  }
+
+  private_service_connect_config {
+    enable_private_service_connect = true
+    project_allowlist = [
+        data.google_project.project.name,
+    ]
+
+    psc_automation_configs {
+      project_id = data.google_project.project.id
+      network    = google_compute_network.network.id
+    }
+  }
+}
+
+resource "google_compute_subnetwork" "subnetwork" {
+  name          = "tf-test-subnetwork-%{random_suffix}"
+  ip_cidr_range = "192.168.0.0/24"
+  region        = "us-central1"
+  network       = google_compute_network.network.id
+}
+
+resource "google_compute_network" "network" {
+  name                    = "tf-test-network-%{random_suffix}"
+  auto_create_subnetworks = false
+}
+
+data "google_project" "project" {}
+`, context)
+}