Added CloudQuotas service and Create QuotaInfo Datasource (GoogleClou…

…dPlatform#10071) * add quota info data source * add test and documention for quota_info data source * fix lint error * clean up * update test * manually import cloudquotas to terraform provider * Update mmv1/third_party/terraform/provider/provider_mmv1_resources.go.erb Co-authored-by: Riley Karson <[email protected]> * Update mmv1/third_party/terraform/provider/provider_mmv1_resources.go.erb Co-authored-by: Riley Karson <[email protected]> --------- Co-authored-by: Riley Karson <[email protected]>
lola98 · Mar 13, 2024 · c1d5720 · c1d5720
1 parent 3ef7532
commit c1d5720
Show file tree

Hide file tree

Showing 5 changed files with 375 additions and 2 deletions.
diff --git a/mmv1/third_party/terraform/.teamcity/components/inputs/services_beta.kt b/mmv1/third_party/terraform/.teamcity/components/inputs/services_beta.kt
@@ -181,7 +181,7 @@ var ServicesListBeta = mapOf(
     "cloudquotas" to mapOf(
         "name" to "cloudquotas",
         "displayName" to "Cloudquotas",
-        "path" to "./google/services/cloudquotas"
+        "path" to "./google-beta/services/cloudquotas"
     ),
     "cloudrun" to mapOf(
         "name" to "cloudrun",
@@ -698,4 +698,4 @@ var ServicesListBeta = mapOf(
         "displayName" to "Workstations",
         "path" to "./google-beta/services/workstations"
     )
-)
+)
diff --git a/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.erb b/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.erb
@@ -15,6 +15,9 @@ import (
 	"github.com/hashicorp/terraform-provider-google/google/services/dataflow"
 	"github.com/hashicorp/terraform-provider-google/google/services/servicenetworking"
 	"github.com/hashicorp/terraform-provider-google/google/tpgiamresource"
+	<% if version == 'ga' -%> // https://github.com/hashicorp/terraform-provider-google/issues/15633 for details
+	"github.com/hashicorp/terraform-provider-google/google/services/cloudquotas"
+	<% end -%>
 )
 
 // Datasources
@@ -49,6 +52,7 @@ var handwrittenDatasources = map[string]*schema.Resource{
 	"google_cloud_identity_groups":                     cloudidentity.DataSourceGoogleCloudIdentityGroups(),
 	"google_cloud_identity_group_memberships":          cloudidentity.DataSourceGoogleCloudIdentityGroupMemberships(),
 	"google_cloud_identity_group_lookup":               cloudidentity.DataSourceGoogleCloudIdentityGroupLookup(),
+	"google_cloud_quotas_quota_info":                   cloudquotas.DataSourceGoogleCloudQuotasQuotaInfo(),
 	"google_cloud_run_locations":                       cloudrun.DataSourceGoogleCloudRunLocations(),
 	"google_cloud_run_service":                         cloudrun.DataSourceGoogleCloudRunService(),
 	"google_cloud_run_v2_job":                          cloudrunv2.DataSourceGoogleCloudRunV2Job(),

diff --git a/.../third_party/terraform/services/cloudquotas/data_source_google_cloud_quotas_quota_info.go b/.../third_party/terraform/services/cloudquotas/data_source_google_cloud_quotas_quota_info.go
@@ -0,0 +1,247 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+package cloudquotas
+
+import (
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-provider-google/google/tpgresource"
+	transport_tpg "github.com/hashicorp/terraform-provider-google/google/transport"
+)
+
+func DataSourceGoogleCloudQuotasQuotaInfo() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGoogleCloudQuotasQuotaInfoRead,
+
+		Schema: map[string]*schema.Schema{
+			"parent": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"service": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"quota_id": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"name": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"metric": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"is_precise": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
+			"refresh_interval": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"container_type": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"dimensions": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem:     &schema.Schema{Type: schema.TypeString},
+			},
+			"metric_display_name": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"quota_display_name": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"metric_unit": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"quota_increase_eligibility": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"is_eligible": {
+							Type:     schema.TypeBool,
+							Computed: true,
+						},
+						"ineligibility_reason": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+			"is_fixed": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
+			"dimensions_infos": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"dimensions": {
+							Type:     schema.TypeMap,
+							Computed: true,
+						},
+						"details": {
+							Type:     schema.TypeList,
+							Computed: true,
+							Elem: &schema.Resource{
+								Schema: map[string]*schema.Schema{
+									"value": {
+										Type:     schema.TypeString,
+										Computed: true,
+									},
+								},
+							},
+						},
+						"applicable_locations": {
+							Type:     schema.TypeList,
+							Computed: true,
+							Elem:     &schema.Schema{Type: schema.TypeString},
+						},
+					},
+				},
+			},
+			"is_concurrent": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
+			"service_request_quota_uri": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+		},
+		UseJSONNumber: true,
+	}
+}
+
+func dataSourceGoogleCloudQuotasQuotaInfoRead(d *schema.ResourceData, meta interface{}) error {
+	config := meta.(*transport_tpg.Config)
+	userAgent, err := tpgresource.GenerateUserAgentString(d, config.UserAgent)
+	if err != nil {
+		return err
+	}
+
+	url, err := tpgresource.ReplaceVars(d, config, "{{CloudQuotasBasePath}}{{parent}}/locations/global/services/{{service}}/quotaInfos/{{quota_id}}")
+	if err != nil {
+		return fmt.Errorf("error setting api endpoint")
+	}
+
+	res, err := transport_tpg.SendRequest(transport_tpg.SendRequestOptions{
+		Config:    config,
+		Method:    "GET",
+		RawURL:    url,
+		UserAgent: userAgent,
+	})
+
+	if err != nil {
+		return transport_tpg.HandleNotFoundError(err, d, fmt.Sprintf("CloudQuotasQuotaInfo %q", d.Id()))
+	}
+
+	if err := d.Set("name", res["name"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo name: %s", err)
+	}
+	if err := d.Set("quota_id", res["quotaId"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo quota_id: %s", err)
+	}
+	if err := d.Set("metric", res["metric"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo metric: %s", err)
+	}
+	if err := d.Set("service", res["service"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo service: %s", err)
+	}
+	if err := d.Set("is_precise", res["isPrecise"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo is_precise: %s", err)
+	}
+	if err := d.Set("refresh_interval", res["refreshInterval"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo refresh_interval: %s", err)
+	}
+	if err := d.Set("container_type", res["containerType"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo container_type: %s", err)
+	}
+	if err := d.Set("dimensions", res["dimensions"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo dimensions: %s", err)
+	}
+	if err := d.Set("metric_display_name", res["metricDisplayName"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo metric_display_name: %s", err)
+	}
+	if err := d.Set("quota_display_name", res["quotaDisplayName"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo quota_display_name: %s", err)
+	}
+	if err := d.Set("metric_unit", res["metricUnit"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo metric_unit: %s", err)
+	}
+	if err := d.Set("quota_increase_eligibility", flattenCloudQuotasQuotaInfoQuotaIncreaseEligibility(res["quotaIncreaseEligibility"], d, config)); err != nil {
+		return fmt.Errorf("error reading QuotaInfo quota_increase_eligibility: %s", err)
+	}
+	if err := d.Set("is_fixed", res["isFixed"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo is_fixed: %s", err)
+	}
+	if err := d.Set("dimensions_infos", flattenCloudQuotasQuotaInfoDimensionsInfos(res["dimensionsInfos"], d, config)); err != nil {
+		return fmt.Errorf("error reading QuotaInfo dimensions_infos: %s", err)
+	}
+	if err := d.Set("is_concurrent", res["isConcurrent"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo is_concurrent: %s", err)
+	}
+	if err := d.Set("service_request_quota_uri", res["serviceRequestQuotaUri"]); err != nil {
+		return fmt.Errorf("error reading QuotaInfo service_request_quota_uri: %s", err)
+	}
+
+	d.SetId(res["name"].(string))
+	return nil
+}
+
+func flattenCloudQuotasQuotaInfoQuotaIncreaseEligibility(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
+	if v == nil {
+		return nil
+	}
+	original := v.(map[string]interface{})
+	if len(original) == 0 {
+		return nil
+	}
+	transformed := make(map[string]interface{})
+	transformed["is_eligible"] = original["is_eligible"]
+	transformed["ineligibility_reason"] = original["ineligibility_reason"]
+	return []interface{}{transformed}
+}
+
+func flattenCloudQuotasQuotaInfoDimensionsInfos(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) []interface{} {
+	if v == nil {
+		return make([]interface{}, 0)
+	}
+
+	original := v.([]interface{})
+	dimensionsInfos := make([]interface{}, 0, len(original))
+
+	for _, raw := range original {
+		data := make(map[string]interface{})
+		data["details"] = flattenCloudQuotasQuotaInfoDetails(raw.(map[string]interface{})["details"], d, config)
+		data["applicable_locations"] = raw.(map[string]interface{})["applicableLocations"]
+		data["dimensions"] = raw.(map[string]interface{})["dimensions"]
+
+		dimensionsInfos = append(dimensionsInfos, data)
+	}
+	return dimensionsInfos
+}
+
+func flattenCloudQuotasQuotaInfoDetails(v interface{}, d *schema.ResourceData, config *transport_tpg.Config) interface{} {
+	original, ok := v.(map[string]interface{})
+	if !ok || len(original) == 0 {
+		return nil
+	}
+
+	return []interface{}{
+		map[string]interface{}{"value": original["value"]},
+	}
+}
diff --git a/...d_party/terraform/services/cloudquotas/data_source_google_cloud_quotas_quota_info_test.go b/...d_party/terraform/services/cloudquotas/data_source_google_cloud_quotas_quota_info_test.go
@@ -0,0 +1,61 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+package cloudquotas_test
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-provider-google/google/acctest"
+	"github.com/hashicorp/terraform-provider-google/google/envvar"
+)
+
+func TestAccDataSourceGoogleQuotaInfo_basic(t *testing.T) {
+	t.Parallel()
+
+	resourceName := "data.google_cloud_quotas_quota_info.my_quota_info"
+	service := "compute.googleapis.com"
+	quotaId := "CPUS-per-project-region"
+
+	context := map[string]interface{}{
+		"project":  envvar.GetTestProjectFromEnv(),
+		"service":  service,
+		"quota_id": quotaId,
+	}
+
+	acctest.VcrTest(t, resource.TestCase{
+		PreCheck:                 func() { acctest.AccTestPreCheck(t) },
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories(t),
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceGoogleQuotaInfo_basic(context),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet(resourceName, "name"),
+					resource.TestCheckResourceAttr(resourceName, "quota_id", quotaId),
+					resource.TestCheckResourceAttr(resourceName, "metric", "compute.googleapis.com/cpus"),
+					resource.TestCheckResourceAttr(resourceName, "service", service),
+					resource.TestCheckResourceAttrSet(resourceName, "is_precise"),
+					resource.TestCheckResourceAttr(resourceName, "container_type", "PROJECT"),
+					resource.TestCheckResourceAttr(resourceName, "dimensions.0", "region"),
+					resource.TestCheckResourceAttr(resourceName, "metric_display_name", "CPUs"),
+					resource.TestCheckResourceAttr(resourceName, "quota_display_name", "CPUs"),
+					resource.TestCheckResourceAttrSet(resourceName, "metric_unit"),
+					resource.TestCheckResourceAttrSet(resourceName, "quota_increase_eligibility.0.is_eligible"),
+					resource.TestCheckResourceAttrSet(resourceName, "dimensions_infos.0.dimensions.region"),
+					resource.TestCheckResourceAttrSet(resourceName, "dimensions_infos.0.details.0.value"),
+					resource.TestCheckResourceAttrSet(resourceName, "dimensions_infos.0.applicable_locations.0"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceGoogleQuotaInfo_basic(context map[string]interface{}) string {
+	return acctest.Nprintf(`
+		data "google_cloud_quotas_quota_info" "my_quota_info" {
+			parent      = "projects/%{project}"	
+			quota_id    = "%{quota_id}"
+			service 	= "%{service}"
+		}
+	`, context)
+}