Add Edge high availability profile data source

For v9.0 and above.

Signed-off-by: Kobi Samoray <[email protected]>

Author: ksamoray

docs/data-sources/policy_edge_high_availability_profile.md

@@ -0,0 +1,33 @@
+---
+subcategory: "Fabric"
+page_title: "NSXT: nsxt_policy_edge_high_availability_profile"
+description: A policy Edge high availability profile data source.
+---
+
+# nsxt_policy_edge_high_availability_profile
+
+This data source provides information about policy Edge high availability profile configured on NSX.
+This data source is applicable to NSX Policy Manager.
+
+## Example Usage
+
+```hcl
+data "nsxt_policy_edge_high_availability_profile" "edge_ha_profile" {
+  display_name = "edge_ha_profile1"
+  site_path    = data.nsxt_policy_site.paris.path
+}
+```
+
+## Argument Reference
+
+* `id` - (Optional) The ID of policy Edge high availability profile to retrieve.
+* `display_name` - (Optional) The Display Name prefix of the Edge high availability profile to retrieve.
+* `site_path` - (Optional) The path of the site which the Edge high availability profile belongs to.
+
+## Attributes Reference
+
+In addition to arguments listed above, the following attributes are exported:
+
+* `description` - The description of the resource.
+* `path` - The NSX path of the policy resource.
+* `unique_id` - A unique identifier assigned by the system.

nsxt/data_source_nsxt_policy_edge_high_availability_profile.go

@@ -0,0 +1,101 @@
+// © Broadcom. All Rights Reserved.
+// The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
+// SPDX-License-Identifier: MPL-2.0
+
+package nsxt
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/vmware/vsphere-automation-sdk-go/services/nsxt/infra/sites/enforcement_points"
+	"github.com/vmware/vsphere-automation-sdk-go/services/nsxt/model"
+)
+
+func dataSourceNsxtPolicyEdgeHighAvailabilityProfile() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceNsxtPolicyEdgeHighAvailabilityProfileRead,
+
+		Schema: map[string]*schema.Schema{
+			"id":           getDataSourceIDSchema(),
+			"display_name": getDataSourceDisplayNameSchema(),
+			"description":  getDataSourceDescriptionSchema(),
+			"path":         getPathSchema(),
+			"site_path": {
+				Type:         schema.TypeString,
+				Description:  "Path of the site this Transport Node Collection belongs to",
+				Optional:     true,
+				ValidateFunc: validatePolicyPath(),
+			},
+			"unique_id": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "A unique identifier assigned by the system",
+			},
+		},
+	}
+}
+
+func dataSourceNsxtPolicyEdgeHighAvailabilityProfileRead(d *schema.ResourceData, m interface{}) error {
+	connector := getPolicyConnector(m)
+	client := enforcement_points.NewEdgeClusterHighAvailabilityProfilesClient(connector)
+
+	objID := d.Get("id").(string)
+	objName := d.Get("display_name").(string)
+	objSitePath := d.Get("site_path").(string)
+	objSiteID := defaultSite
+	// For local manager, if site path is not provided, use default site
+	if objSitePath != "" {
+		objSiteID = getPolicyIDFromPath(objSitePath)
+	}
+	var obj model.PolicyEdgeHighAvailabilityProfile
+	if objID != "" {
+		// Get by id
+		objGet, err := client.Get(objSiteID, getPolicyEnforcementPoint(m), objID)
+		if err != nil {
+			return handleDataSourceReadError(d, "EdgeHighAvailabilityProfile", objID, err)
+		}
+		obj = objGet
+	} else if objName == "" {
+		return fmt.Errorf("Error obtaining EdgeHighAvailabilityProfile ID or name during read")
+	} else {
+		// Get by full name/prefix
+		objList, err := client.List(objSiteID, getPolicyEnforcementPoint(m), nil, nil, nil, nil, nil, nil, nil)
+		if err != nil {
+			return handleListError("EdgeHighAvailabilityProfile", err)
+		}
+		// go over the list to find the correct one (prefer a perfect match. If not - prefix match)
+		var perfectMatch []model.PolicyEdgeHighAvailabilityProfile
+		var prefixMatch []model.PolicyEdgeHighAvailabilityProfile
+		for _, objInList := range objList.Results {
+			if strings.HasPrefix(*objInList.DisplayName, objName) {
+				prefixMatch = append(prefixMatch, objInList)
+			}
+			if *objInList.DisplayName == objName {
+				perfectMatch = append(perfectMatch, objInList)
+			}
+		}
+		if len(perfectMatch) > 0 {
+			if len(perfectMatch) > 1 {
+				return fmt.Errorf("Found multiple EdgeHighAvailabilityProfile with name '%s'", objName)
+			}
+			obj = perfectMatch[0]
+		} else if len(prefixMatch) > 0 {
+			if len(prefixMatch) > 1 {
+				return fmt.Errorf("Found multiple EdgeHighAvailabilityProfiles with name starting with '%s'", objName)
+			}
+			obj = prefixMatch[0]
+		} else {
+			return fmt.Errorf("EdgeHighAvailabilityProfile with name '%s' was not found", objName)
+		}
+	}
+
+	d.SetId(*obj.Id)
+	d.Set("display_name", obj.DisplayName)
+	d.Set("description", obj.Description)
+	d.Set("path", obj.Path)
+	d.Set("unique_id", obj.UniqueId)
+
+	return nil
+}

nsxt/data_source_nsxt_policy_edge_high_availability_profile_test.go

@@ -0,0 +1,43 @@
+// © Broadcom. All Rights Reserved.
+// The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
+// SPDX-License-Identifier: MPL-2.0
+
+package nsxt
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccDataSourceNsxtPolicyEdgeHighAvailabilityProfile_basic(t *testing.T) {
+	testResourceName := "data.nsxt_policy_edge_high_availability_profile.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			testAccOnlyLocalManager(t)
+			testAccPreCheck(t)
+			testAccNSXVersion(t, "9.0.0")
+		},
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccNSXEdgeHighAvailabilityProfileReadTemplate(),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(testResourceName, "display_name", accTestPolicyEdgeHighAvailabilityProfileUpdateAttributes["display_name"]),
+					resource.TestCheckResourceAttrSet(testResourceName, "id"),
+					resource.TestCheckResourceAttrSet(testResourceName, "unique_id"),
+				),
+			},
+		},
+	})
+}
+
+func testAccNSXEdgeHighAvailabilityProfileReadTemplate() string {
+	return testAccNsxtPolicyEdgeHighAvailabilityProfileMinimalistic() + `
+data "nsxt_policy_edge_high_availability_profile" "test" {
+  display_name = nsxt_policy_edge_high_availability_profile.test.display_name 
+
+  depends_on = [nsxt_policy_edge_high_availability_profile.test]
+}`
+}

nsxt/provider.go

@@ -345,6 +345,7 @@ func Provider() *schema.Provider {
 			"nsxt_policy_services":                                   dataSourceNsxtPolicyServices(),
 			"nsxt_policy_groups":                                     dataSourceNsxtPolicyGroups(),
 			"nsxt_policy_edge_transport_node":                        dataSourceNsxtPolicyEdgeTransportNode(),
+			"nsxt_policy_edge_high_availability_profile":             dataSourceNsxtPolicyEdgeHighAvailabilityProfile(),
 		},
 
 		ResourcesMap: map[string]*schema.Resource{