Update kubernetes_cluster_node_pool_resource_test.go

Author: hqhqhqhqhqhqhqhqhqhqhq

internal/services/containers/kubernetes_cluster_node_pool_resource_test.go

@@ -156,18 +156,16 @@ func TestAccKubernetesClusterNodePool_gatewayProfile(t *testing.T) {
 
 	data.ResourceTest(t, r, []acceptance.TestStep{
 		{
-			Config: r.gatewayProfileConfig(data),
+			Config: r.gatewayProfileConfig(data, 30),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("gateway_profile.0.public_ip_prefix_size").HasValue("30"),
 			),
 		},
 		data.ImportStep(),
 		{
-			Config: r.gatewayProfileConfigUpdated(data),
+			Config: r.gatewayProfileConfig(data, 29),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("gateway_profile.0.public_ip_prefix_size").HasValue("29"),
 			),
 		},
 		data.ImportStep(),
@@ -3531,90 +3529,43 @@ resource "azurerm_kubernetes_cluster_node_pool" "pool2" {
 	)
 }
 
-func (KubernetesClusterNodePoolResource) gatewayProfileConfig(data acceptance.TestData) string {
+func (KubernetesClusterNodePoolResource) gatewayProfileConfig(data acceptance.TestData, publicIPPrefixSize int) string {
 	return fmt.Sprintf(`
 provider "azurerm" {
   features {}
 }
 
 resource "azurerm_resource_group" "test" {
-  name     = "acctestRG-aks-%d"
-  location = "%s"
+  name     = "acctestRG-aks-%[1]d"
+  location = "%[2]s"
 }
 
 resource "azurerm_kubernetes_cluster" "test" {
-  name                = "acctestaks%d"
+  name                = "acctestaks%[1]d"
   location            = azurerm_resource_group.test.location
   resource_group_name = azurerm_resource_group.test.name
-  dns_prefix          = "acctestaks%d"
-  sku_tier            = "Standard"
-  default_node_pool {
-    name       = "default"
-    node_count = 1
-    vm_size    = "Standard_DS2_v2"
-  }
-  identity {
-    type = "SystemAssigned"
-  }
-}
-
-resource "azurerm_kubernetes_cluster_node_pool" "test" {
-  name                  = "acctestaks%d"
-  kubernetes_cluster_id = azurerm_kubernetes_cluster.test.id
-  vm_size               = "Standard_DS2_v2"
-  node_count            = 1
-  gateway_profile {
-    public_ip_prefix_size = 30
-  }
-}
-`, data.RandomInteger, // resource_group name
-		data.Locations.Primary, // resource_group location
-		data.RandomInteger,     // kubernetes_cluster name
-		data.RandomInteger,     // kubernetes_cluster dns_prefix
-		data.RandomInteger,     // node_pool name
-	)
-}
-
-func (KubernetesClusterNodePoolResource) gatewayProfileConfigUpdated(data acceptance.TestData) string {
-	return fmt.Sprintf(`
-provider "azurerm" {
-  features {}
-}
-
-resource "azurerm_resource_group" "test" {
-  name     = "acctestRG-aks-%d"
-  location = "%s"
-}
+  dns_prefix          = "acctestaks%[1]d"
 
-resource "azurerm_kubernetes_cluster" "test" {
-  name                = "acctestaks%d"
-  location            = azurerm_resource_group.test.location
-  resource_group_name = azurerm_resource_group.test.name
-  dns_prefix          = "acctestaks%d"
-  sku_tier            = "Standard"
   default_node_pool {
     name       = "default"
     node_count = 1
-    vm_size    = "Standard_DS2_v2"
+    vm_size    = "Standard_D2s_v3"
   }
+
   identity {
     type = "SystemAssigned"
   }
 }
 
 resource "azurerm_kubernetes_cluster_node_pool" "test" {
-  name                  = "acctestaks%d"
+  name                  = "gateway"
   kubernetes_cluster_id = azurerm_kubernetes_cluster.test.id
-  vm_size               = "Standard_DS2_v2"
+  vm_size               = "Standard_D2s_v3"
   node_count            = 1
+
   gateway_profile {
-    public_ip_prefix_size = 29
+    public_ip_prefix_size = %[3]d
   }
 }
-`, data.RandomInteger, // resource_group name
-		data.Locations.Primary, // resource_group location
-		data.RandomInteger,     // kubernetes_cluster name
-		data.RandomInteger,     // kubernetes_cluster dns_prefix
-		data.RandomInteger,     // node_pool name
-	)
+`, data.RandomInteger, data.Locations.Primary, publicIPPrefixSize)
 }