Multitenancy provider tests

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

Author: ksamoray

nsxt/data_source_nsxt_policy_context_profile_test.go

@@ -65,7 +65,30 @@ func TestAccDataSourceNsxtPolicyContextProfile_multitenancy(t *testing.T) {
 		Providers: testAccProviders,
 		Steps: []resource.TestStep{
 			{
-				Config: testAccNsxtPolicyContextProfileMultitenancyTemplate(name),
+				Config: testAccNsxtPolicyContextProfileMultitenancyTemplate(name, true),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(testResourceName, "display_name", name),
+					resource.TestCheckResourceAttrSet(testResourceName, "description"),
+					resource.TestCheckResourceAttrSet(testResourceName, "path"),
+				),
+			},
+		},
+	})
+}
+
+func TestAccDataSourceNsxtPolicyContextProfile_multitenancyProvider(t *testing.T) {
+	name := getAccTestResourceName()
+	testResourceName := "data.nsxt_policy_context_profile.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			testAccPreCheck(t)
+			testAccOnlyMultitenancy(t)
+		},
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccNsxtPolicyContextProfileMultitenancyTemplate(name, false),
 				Check: resource.ComposeTestCheckFunc(
 					resource.TestCheckResourceAttr(testResourceName, "display_name", name),
 					resource.TestCheckResourceAttrSet(testResourceName, "description"),
@@ -83,8 +106,11 @@ data "nsxt_policy_context_profile" "test" {
 }`, name)
 }
 
-func testAccNsxtPolicyContextProfileMultitenancyTemplate(name string) string {
-	context := testAccNsxtPolicyMultitenancyContext()
+func testAccNsxtPolicyContextProfileMultitenancyTemplate(name string, withContext bool) string {
+	context := ""
+	if withContext {
+		context = testAccNsxtPolicyMultitenancyContext()
+	}
 	return fmt.Sprintf(`
 resource "nsxt_policy_context_profile" "test" {
 %s

nsxt/utils.go

@@ -681,6 +681,10 @@ func handlePagination(lister func(*paginationInfo) error) (int64, error) {
 }
 
 func getContextSchema(isRequired, isComputed, isVPC bool) *schema.Schema {
+	return getContextExtendedSchema(isRequired, isComputed, isVPC, false)
+}
+
+func getContextExtendedSchema(isRequired, isComputed, isVPC, isProvider bool) *schema.Schema {
 	elemSchema := map[string]*schema.Schema{
 		"project_id": {
 			Type:         schema.TypeString,
@@ -699,6 +703,10 @@ func getContextSchema(isRequired, isComputed, isVPC bool) *schema.Schema {
 			ValidateFunc: validation.StringIsNotWhiteSpace,
 		}
 	}
+	if isProvider {
+		elemSchema["project_id"].DefaultFunc = schema.EnvDefaultFunc("NSXT_PROVIDER_PROJECT_ID", false)
+		elemSchema["vpc_id"].DefaultFunc = schema.EnvDefaultFunc("NSXT_PROVIDER_VPC_ID", false)
+	}
 	return &schema.Schema{
 		Type:        schema.TypeList,
 		Description: "Resource context",

nsxt/utils_test.go

@@ -243,6 +243,10 @@ func testAccIsMultitenancy() bool {
 	return os.Getenv("NSXT_PROJECT_ID") != ""
 }
 
+func testAccIsMultitenancyProvider() bool {
+	return os.Getenv("NSXT_PROVIDER_PROJECT_ID") != ""
+}
+
 func testAccIsVPC() bool {
 	return os.Getenv("NSXT_VPC_PROJECT_ID") != "" && os.Getenv("NSXT_VPC_ID") != ""
 }
@@ -322,6 +326,13 @@ func testAccOnlyMultitenancy(t *testing.T) {
 	}
 }
 
+func testAccOnlyMultitenancyProvider(t *testing.T) {
+	testAccNSXVersion(t, "4.1.0")
+	if !testAccIsMultitenancyProvider() {
+		t.Skipf("This test requires a multitenancy environment with provider configuration")
+	}
+}
+
 func testAccNSXGlobalManagerSitePrecheck(t *testing.T) {
 	if testAccIsGlobalManager() && getTestSiteName() == "" {
 		str := fmt.Sprintf("%s must be set for this acceptance test", "NSXT_TEST_SITE_NAME")