Support constraint resource

Signed-off-by: Anna Khmelnitsky <[email protected]>

Author: annakhm

Diff for: api/infra/constraint.go

@@ -0,0 +1,137 @@
+//nolint:revive
+package infra
+
+// The following file has been autogenerated. Please avoid any changes!
+import (
+	"errors"
+
+	vapiProtocolClient_ "github.com/vmware/vsphere-automation-sdk-go/runtime/protocol/client"
+	client0 "github.com/vmware/vsphere-automation-sdk-go/services/nsxt/infra"
+	model0 "github.com/vmware/vsphere-automation-sdk-go/services/nsxt/model"
+	client1 "github.com/vmware/vsphere-automation-sdk-go/services/nsxt/orgs/projects/infra"
+
+	utl "github.com/vmware/terraform-provider-nsxt/api/utl"
+)
+
+type ConstraintClientContext utl.ClientContext
+
+func NewConstraintsClient(sessionContext utl.SessionContext, connector vapiProtocolClient_.Connector) *ConstraintClientContext {
+	var client interface{}
+
+	switch sessionContext.ClientType {
+
+	case utl.Local:
+		client = client0.NewConstraintsClient(connector)
+
+	case utl.Multitenancy:
+		client = client1.NewConstraintsClient(connector)
+
+	default:
+		return nil
+	}
+	return &ConstraintClientContext{Client: client, ClientType: sessionContext.ClientType, ProjectID: sessionContext.ProjectID, VPCID: sessionContext.VPCID}
+}
+
+func (c ConstraintClientContext) Get(constraintIdParam string) (model0.Constraint, error) {
+	var obj model0.Constraint
+	var err error
+
+	switch c.ClientType {
+
+	case utl.Local:
+		client := c.Client.(client0.ConstraintsClient)
+		obj, err = client.Get(constraintIdParam)
+		if err != nil {
+			return obj, err
+		}
+
+	case utl.Multitenancy:
+		client := c.Client.(client1.ConstraintsClient)
+		obj, err = client.Get(utl.DefaultOrgID, c.ProjectID, constraintIdParam)
+		if err != nil {
+			return obj, err
+		}
+
+	default:
+		return obj, errors.New("invalid infrastructure for model")
+	}
+	return obj, err
+}
+
+func (c ConstraintClientContext) Patch(constraintIdParam string, constraintParam model0.Constraint) error {
+	var err error
+
+	switch c.ClientType {
+
+	case utl.Local:
+		client := c.Client.(client0.ConstraintsClient)
+		err = client.Patch(constraintIdParam, constraintParam)
+
+	case utl.Multitenancy:
+		client := c.Client.(client1.ConstraintsClient)
+		err = client.Patch(utl.DefaultOrgID, c.ProjectID, constraintIdParam, constraintParam)
+
+	default:
+		err = errors.New("invalid infrastructure for model")
+	}
+	return err
+}
+
+func (c ConstraintClientContext) Update(constraintIdParam string, constraintParam model0.Constraint) (model0.Constraint, error) {
+	var err error
+	var obj model0.Constraint
+
+	switch c.ClientType {
+
+	case utl.Local:
+		client := c.Client.(client0.ConstraintsClient)
+		obj, err = client.Update(constraintIdParam, constraintParam)
+
+	case utl.Multitenancy:
+		client := c.Client.(client1.ConstraintsClient)
+		obj, err = client.Update(utl.DefaultOrgID, c.ProjectID, constraintIdParam, constraintParam)
+
+	default:
+		err = errors.New("invalid infrastructure for model")
+	}
+	return obj, err
+}
+
+func (c ConstraintClientContext) Delete(constraintIdParam string) error {
+	var err error
+
+	switch c.ClientType {
+
+	case utl.Local:
+		client := c.Client.(client0.ConstraintsClient)
+		err = client.Delete(constraintIdParam)
+
+	case utl.Multitenancy:
+		client := c.Client.(client1.ConstraintsClient)
+		err = client.Delete(utl.DefaultOrgID, c.ProjectID, constraintIdParam)
+
+	default:
+		err = errors.New("invalid infrastructure for model")
+	}
+	return err
+}
+
+func (c ConstraintClientContext) List(cursorParam *string, includeMarkForDeleteObjectsParam *bool, includedFieldsParam *string, pageSizeParam *int64, sortAscendingParam *bool, sortByParam *string) (model0.ConstraintListResult, error) {
+	var err error
+	var obj model0.ConstraintListResult
+
+	switch c.ClientType {
+
+	case utl.Local:
+		client := c.Client.(client0.ConstraintsClient)
+		obj, err = client.List(cursorParam, includeMarkForDeleteObjectsParam, includedFieldsParam, pageSizeParam, sortAscendingParam, sortByParam)
+
+	case utl.Multitenancy:
+		client := c.Client.(client1.ConstraintsClient)
+		obj, err = client.List(utl.DefaultOrgID, c.ProjectID, cursorParam, includeMarkForDeleteObjectsParam, includedFieldsParam, pageSizeParam, sortAscendingParam, sortByParam)
+
+	default:
+		err = errors.New("invalid infrastructure for model")
+	}
+	return obj, err
+}

Diff for: nsxt/data_source_nsxt_vpc_test.go

@@ -56,7 +56,7 @@ func testAccDataSourceNsxtVPCCreate(name string) error {
 	}
 
 	ipBlockID := newUUID()
-	err = testAccDataSourceNsxtPolicyIPBlockCreate(testAccGetProjectContext(), name, ipBlockID, "192.168.240.0/24", true)
+	err = testAccDataSourceNsxtPolicyIPBlockCreate(testAccGetMultitenancyContext(), name, ipBlockID, "192.168.240.0/24", true)
 	if err != nil {
 		return err
 	}
@@ -108,7 +108,7 @@ func testAccDataSourceNsxtVPCDeleteByName(name string) error {
 			if err != nil {
 				return handleDeleteError("VPC", *objInList.Id, err)
 			}
-			return testAccDataSourceNsxtPolicyIPBlockDeleteByName(testAccGetProjectContext(), name)
+			return testAccDataSourceNsxtPolicyIPBlockDeleteByName(testAccGetMultitenancyContext(), name)
 		}
 	}
 	return fmt.Errorf("error while deleting VPC '%s': resource not found", name)

Diff for: nsxt/metadata/metadata.go

@@ -26,6 +26,11 @@ var StandardTypeIdentifier = TypeIdentifier{
 	APIFieldName: "type",
 }
 
+var ResourceTypeTypeIdentifier = TypeIdentifier{
+	SdkName:      "ResourceType",
+	APIFieldName: "resource_type",
+}
+
 const (
 	PolymorphicTypeFlatten = "flatten"
 	PolymorphicTypeNested  = "nested"

Diff for: nsxt/provider.go

@@ -542,6 +542,7 @@ func Provider() *schema.Provider {
 			"nsxt_policy_edge_transport_node":            resourceNsxtPolicyEdgeTransportNode(),
 			"nsxt_policy_edge_high_availability_profile": resourceNsxtPolicyEdgeHighAvailabilityProfile(),
 			"nsxt_policy_edge_cluster":                   resourceNsxtPolicyEdgeCluster(),
+			"nsxt_policy_constraint":                     resourceNsxtPolicyConstraint(),
 		},
 
 		ConfigureFunc: providerConfigure,