From 6f342c66088d7ab8e0167cdf5ed9bc841c75fc85 Mon Sep 17 00:00:00 2001 From: Michael Lopez Date: Tue, 17 Dec 2024 19:55:39 +0100 Subject: [PATCH] Supporting gke hub feature data source (#12459) --- .../provider/provider_mmv1_resources.go.tmpl | 1 + .../data_source_google_gke_hub_feature.go | 50 +++++++++++ ...data_source_google_gke_hub_feature_test.go | 88 +++++++++++++++++++ 3 files changed, 139 insertions(+) create mode 100644 mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature.go create mode 100644 mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature_test.go diff --git a/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.tmpl b/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.tmpl index 6f1643252c7e..b198fbd0c807 100644 --- a/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.tmpl +++ b/mmv1/third_party/terraform/provider/provider_mmv1_resources.go.tmpl @@ -130,6 +130,7 @@ var handwrittenDatasources = map[string]*schema.Resource{ "google_dns_managed_zones": dns.DataSourceDnsManagedZones(), "google_dns_record_set": dns.DataSourceDnsRecordSet(), "google_gke_hub_membership_binding": gkehub2.DataSourceGoogleGkeHubMembershipBinding(), + "google_gke_hub_feature": gkehub2.DataSourceGoogleGkeHubFeature(), "google_filestore_instance": filestore.DataSourceGoogleFilestoreInstance(), "google_iam_policy": resourcemanager.DataSourceGoogleIamPolicy(), "google_iam_role": resourcemanager.DataSourceGoogleIamRole(), diff --git a/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature.go b/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature.go new file mode 100644 index 000000000000..79e001b5bc58 --- /dev/null +++ b/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature.go @@ -0,0 +1,50 @@ +package gkehub2 + +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 DataSourceGoogleGkeHubFeature() *schema.Resource { + dsSchema := tpgresource.DatasourceSchemaFromResourceSchema(ResourceGKEHub2Feature().Schema) + tpgresource.AddRequiredFieldsToSchema(dsSchema, "location") + tpgresource.AddOptionalFieldsToSchema(dsSchema, "project") + tpgresource.AddRequiredFieldsToSchema(dsSchema, "name") + + return &schema.Resource{ + Read: dataSourceGoogleGkeHubFeatureRead, + Schema: dsSchema, + } +} + +func dataSourceGoogleGkeHubFeatureRead(d *schema.ResourceData, meta interface{}) error { + config := meta.(*transport_tpg.Config) + + id, err := tpgresource.ReplaceVars(d, config, "projects/{{project}}/locations/{{location}}/features/{{name}}") + if err != nil { + return fmt.Errorf("Error constructing id: %s", err) + } + + d.SetId(id) + + err = resourceGKEHub2FeatureRead(d, meta) + if err != nil { + return err + } + + if err := tpgresource.SetDataSourceLabels(d); err != nil { + return err + } + + if err := tpgresource.SetDataSourceAnnotations(d); err != nil { + return err + } + + if d.Id() == "" { + return fmt.Errorf("%s not found", id) + } + return nil +} diff --git a/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature_test.go b/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature_test.go new file mode 100644 index 000000000000..da4cf8be7dfa --- /dev/null +++ b/mmv1/third_party/terraform/services/gkehub2/data_source_google_gke_hub_feature_test.go @@ -0,0 +1,88 @@ +package gkehub2_test + +import ( + "fmt" + "strings" + "testing" + + "github.com/hashicorp/terraform-plugin-testing/helper/resource" + "github.com/hashicorp/terraform-plugin-testing/terraform" + "github.com/hashicorp/terraform-provider-google/google/acctest" + "github.com/hashicorp/terraform-provider-google/google/tpgresource" + transport_tpg "github.com/hashicorp/terraform-provider-google/google/transport" +) + +func TestAccDataSourceGoogleGkeHubFeature_basic(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) }, + Providers: acctest.TestAccProviders, + CheckDestroy: testAccCheckGoogleGkeHubFeatureDestroyProducer(t), + Steps: []resource.TestStep{ + { + Config: testAccDataSourceGoogleGkeHubFeature_basic(context), + Check: resource.ComposeTestCheckFunc( + acctest.CheckDataSourceStateMatchesResourceState("data.google_gke_hub_feature.example", "google_gke_hub_feature.example"), + ), + }, + }, + }) +} + +func testAccDataSourceGoogleGkeHubFeature_basic(context map[string]interface{}) string { + return acctest.Nprintf(` +resource "google_gke_hub_feature" "example" { + location = "global" + name = "servicemesh" +} + +data "google_gke_hub_feature" "example" { + location = google_gke_hub_feature.example.location + name = google_gke_hub_feature.example.name +} +`, context) +} + +func testAccCheckGoogleGkeHubFeatureDestroyProducer(t *testing.T) func(s *terraform.State) error { + return func(s *terraform.State) error { + for name, rs := range s.RootModule().Resources { + if rs.Type != "google_gke_hub_feature" { + continue + } + if strings.HasPrefix(name, "data.") { + continue + } + + config := acctest.GoogleProviderConfig(t) + + url, err := tpgresource.ReplaceVarsForTest(config, rs, "{{GKEHub2BasePath}}projects/{{project}}/locations/{{location}}/features/{{name}}") + if err != nil { + return err + } + + billingProject := "" + + if config.BillingProject != "" { + billingProject = config.BillingProject + } + + _, err = transport_tpg.SendRequest(transport_tpg.SendRequestOptions{ + Config: config, + Method: "GET", + Project: billingProject, + RawURL: url, + UserAgent: config.UserAgent, + }) + if err == nil { + return fmt.Errorf("GKEHub2Feature still exists at %s", url) + } + } + + return nil + } +}