new resource azurerm_arc_kubernetes_provisioned_cluster_instance

internal/services/arckubernetes/arc_kubernetes_provisioned_cluster_instance_resource_test.go

@@ -0,0 +1,341 @@
+package arckubernetes_test
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/hashicorp/go-azure-helpers/lang/pointer"
+	"github.com/hashicorp/go-azure-helpers/resourcemanager/commonids"
+	"github.com/hashicorp/go-azure-sdk/resource-manager/hybridkubernetes/2024-01-01/connectedclusters"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance/check"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/clients"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/services/arckubernetes/parse"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
+)
+
+// https://learn.microsoft.com/en-us/azure/aks/hybrid/aks-create-clusters-cli?toc=%2Fazure-stack%2Fhci%2Ftoc.json&bc=%2Fazure-stack%2Fbreadcrumb%2Ftoc.json#before-you-begin
+// The resource can only be created on the customlocation generated after HCI deployment
+const (
+	customLocationIdEnv = "ARM_TEST_STACK_HCI_CUSTOM_LOCATION_ID"
+)
+
+type ArcKubernetesProvisionedClusterInstanceResource struct{}
+
+func TestAccArcKubernetesProvisionedClusterInstance(t *testing.T) {
+	// NOTE: this is a combined test rather than separate split out tests due to
+	// the test environment network limitation
+	// (which our test suite can't easily work around)
+
+	testCases := map[string]func(t *testing.T){
+		"basic":          testAccArcKubernetesProvisionedClusterInstance_basic,
+		"complete":       testAccArcKubernetesProvisionedClusterInstance_complete,
+		"update":         testAccArcKubernetesProvisionedClusterInstance_update,
+		"requiresImport": testAccArcKubernetesProvisionedClusterInstance_requiresImport,
+	}
+	for name, m := range testCases {
+		t.Run(name, func(t *testing.T) {
+			m(t)
+		})
+	}
+}
+
+func testAccArcKubernetesProvisionedClusterInstance_basic(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_arc_kubernetes_provisioned_cluster_instance", "test")
+	r := ArcKubernetesProvisionedClusterInstanceResource{}
+	data.ResourceSequentialTest(t, r, []acceptance.TestStep{
+		{
+			Config: r.basic(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
+func testAccArcKubernetesProvisionedClusterInstance_complete(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_arc_kubernetes_provisioned_cluster_instance", "test")
+	r := ArcKubernetesProvisionedClusterInstanceResource{}
+	data.ResourceSequentialTest(t, r, []acceptance.TestStep{
+		{
+			Config: r.complete(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
+func testAccArcKubernetesProvisionedClusterInstance_update(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_arc_kubernetes_provisioned_cluster_instance", "test")
+	r := ArcKubernetesProvisionedClusterInstanceResource{}
+	data.ResourceSequentialTest(t, r, []acceptance.TestStep{
+		{
+			Config: r.basic(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+		{
+			Config: r.complete(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+		{
+			Config: r.basic(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
+func testAccArcKubernetesProvisionedClusterInstance_requiresImport(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_arc_kubernetes_provisioned_cluster_instance", "test")
+	r := ArcKubernetesProvisionedClusterInstanceResource{}
+	data.ResourceSequentialTest(t, r, []acceptance.TestStep{
+		{
+			Config: r.basic(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		{
+			Config:      r.requiresImport(data),
+			ExpectError: acceptance.RequiresImportError(data.ResourceType),
+		},
+	})
+}
+
+func (r ArcKubernetesProvisionedClusterInstanceResource) Exists(ctx context.Context, clients *clients.Client, state *pluginsdk.InstanceState) (*bool, error) {
+	id, err := parse.ArcKubernetesProvisionedClusterInstanceID(state.ID)
+	if err != nil {
+		return nil, err
+	}
+
+	connectedClusterId := connectedclusters.NewConnectedClusterID(id.SubscriptionId, id.ResourceGroup, id.ConnectedClusterName)
+	scopeId := commonids.NewScopeID(connectedClusterId.ID())
+
+	resp, err := clients.ArcKubernetes.ProvisionedClusterInstancesClient.ProvisionedClusterInstancesGet(ctx, scopeId)
+	if err != nil {
+		return nil, fmt.Errorf("retrieving %s: %+v", *id, err)
+	}
+
+	return pointer.To(resp.Model != nil), nil
+}
+
+func (r ArcKubernetesProvisionedClusterInstanceResource) basic(data acceptance.TestData) string {
+	template := r.template(data)
+	return fmt.Sprintf(`
+provider "azurerm" {
+  features {}
+}
+
+%[1]s
+
+resource "azurerm_arc_kubernetes_provisioned_cluster_instance" "test" {
+  cluster_id         = azurerm_arc_kubernetes_provisioned_cluster.test.id
+  custom_location_id = "%[3]s"
+  kubernetes_version = "1.28.5"
+
+  agent_pool_profile {
+    name    = "nodepool1"
+    os_sku  = "CBLMariner"
+    os_type = "Linux"
+    vm_size = "Standard_A4_v2"
+  }
+
+  cloud_provider_profile {
+    infra_network_profile {
+      vnet_subnet_ids = [azurerm_stack_hci_logical_network.test.id]
+    }
+  }
+
+  control_plane_profile {
+    host_ip = "192.168.1.190"
+    vm_size = "Standard_A4_v2"
+  }
+
+  linux_profile {
+    ssh_key = [tls_private_key.rsaKey.public_key_openssh]
+  }
+
+  network_profile {
+    network_policy = "calico"
+    pod_cidr       = "10.244.0.0/16"
+  }
+
+  storage_profile {
+    smb_csi_driver_enabled = false
+    nfs_csi_driver_enabled = false
+  }
+}
+`, template, data.RandomInteger, os.Getenv(customLocationIdEnv))
+}
+
+func (r ArcKubernetesProvisionedClusterInstanceResource) requiresImport(data acceptance.TestData) string {
+	config := r.basic(data)
+	return fmt.Sprintf(`
+%s
+
+resource "azurerm_arc_kubernetes_provisioned_cluster_instance" "import" {
+  cluster_id         = azurerm_arc_kubernetes_provisioned_cluster_instance.test.cluster_id
+  custom_location_id = azurerm_arc_kubernetes_provisioned_cluster_instance.test.custom_location_id
+  kubernetes_version = "1.28.5"
+
+  agent_pool_profile {
+    name    = "nodepool1"
+    os_sku  = "CBLMariner"
+    os_type = "Linux"
+    vm_size = "Standard_A4_v2"
+  }
+
+  cloud_provider_profile {
+    infra_network_profile {
+      vnet_subnet_ids = [azurerm_stack_hci_logical_network.test.id]
+    }
+  }
+
+  control_plane_profile {
+    host_ip = "192.168.1.190"
+    vm_size = "Standard_A4_v2"
+  }
+
+  linux_profile {
+    ssh_key = [tls_private_key.rsaKey.public_key_openssh]
+  }
+
+  network_profile {
+    network_policy = "calico"
+    pod_cidr       = "10.244.0.0/16"
+  }
+
+  storage_profile {
+    smb_csi_driver_enabled = false
+    nfs_csi_driver_enabled = false
+  }
+}
+`, config)
+}
+
+func (r ArcKubernetesProvisionedClusterInstanceResource) complete(data acceptance.TestData) string {
+	template := r.template(data)
+	return fmt.Sprintf(`
+provider "azurerm" {
+  features {}
+}
+
+%[1]s
+
+resource "azurerm_arc_kubernetes_provisioned_cluster_instance" "test" {
+  cluster_id         = azurerm_arc_kubernetes_provisioned_cluster.test.id
+  custom_location_id = "%[3]s"
+  kubernetes_version = "1.28.5"
+
+  agent_pool_profile {
+    auto_scaling_enabled = true
+    count                = 1
+    max_count            = 2
+    min_count            = 1
+    max_pods             = 20
+    name                 = "nodepool1"
+    os_sku               = "CBLMariner"
+    os_type              = "Linux"
+    vm_size              = "Standard_A4_v2"
+    node_taints          = ["env=prod:NoSchedule"]
+
+    node_labels = {
+      foo = "bar"
+      env = "dev"
+    }
+  }
+
+  cloud_provider_profile {
+    infra_network_profile {
+      vnet_subnet_ids = [azurerm_stack_hci_logical_network.test.id]
+    }
+  }
+
+  cluster_vm_access_profile {
+    authorized_ip_ranges = "192.168.0.1,192.168.0.2"
+  }
+
+  control_plane_profile {
+    count   = 3
+    vm_size = "Standard_A4_v2"
+    host_ip = "192.168.1.190"
+  }
+
+  license_profile {
+    azure_hybrid_benefit = "False"
+  }
+
+  linux_profile {
+    ssh_key = [tls_private_key.rsaKey.public_key_openssh]
+  }
+
+  network_profile {
+    network_policy = "calico"
+    pod_cidr       = "10.244.0.0/16"
+  }
+
+  storage_profile {
+    smb_csi_driver_enabled = true
+    nfs_csi_driver_enabled = true
+  }
+}
+`, template, data.RandomInteger, os.Getenv(customLocationIdEnv))
+}
+
+func (r ArcKubernetesProvisionedClusterInstanceResource) template(data acceptance.TestData) string {
+	return fmt.Sprintf(`
+resource "tls_private_key" "rsaKey" {
+  algorithm = "RSA"
+  rsa_bits  = 4096
+}
+
+resource "azurerm_resource_group" "test" {
+  name     = "acctest-akpci-%[1]d"
+  location = "%[2]s"
+}
+
+resource "azurerm_stack_hci_logical_network" "test" {
+  name                = "acctest-ln-%[1]d"
+  resource_group_name = azurerm_resource_group.test.name
+  location            = azurerm_resource_group.test.location
+  custom_location_id  = "%[3]s"
+  virtual_switch_name = "ConvergedSwitch(managementcompute)"
+  dns_servers         = ["192.168.1.254"]
+
+  subnet {
+    ip_allocation_method = "Static"
+    address_prefix       = "192.168.1.0/24"
+    ip_pool {
+      start = "192.168.1.171"
+      end   = "192.168.1.190"
+    }
+    route {
+      address_prefix      = "0.0.0.0/0"
+      next_hop_ip_address = "192.168.1.1"
+    }
+  }
+}
+
+resource "azurerm_arc_kubernetes_provisioned_cluster" "test" {
+  name                = "acctest-akcc-%[1]d"
+  resource_group_name = azurerm_resource_group.test.name
+  location            = azurerm_resource_group.test.location
+  identity {
+    type = "SystemAssigned"
+  }
+}
+`, data.RandomInteger, data.Locations.Primary, os.Getenv(customLocationIdEnv))
+}

internal/services/arckubernetes/client/client.go

@@ -6,16 +6,18 @@ package client
 import (
 	"fmt"
 
+	"github.com/hashicorp/go-azure-sdk/resource-manager/hybridazurekubernetesservice/2024-01-01/provisionedclusterinstances"
 	"github.com/hashicorp/go-azure-sdk/resource-manager/hybridkubernetes/2024-01-01/connectedclusters"
 	"github.com/hashicorp/go-azure-sdk/resource-manager/kubernetesconfiguration/2022-11-01/extensions"
 	"github.com/hashicorp/go-azure-sdk/resource-manager/kubernetesconfiguration/2023-05-01/fluxconfiguration"
 	"github.com/hashicorp/terraform-provider-azurerm/internal/common"
 )
 
 type Client struct {
-	ArcKubernetesClient     *connectedclusters.ConnectedClustersClient
-	ExtensionsClient        *extensions.ExtensionsClient
-	FluxConfigurationClient *fluxconfiguration.FluxConfigurationClient
+	ArcKubernetesClient               *connectedclusters.ConnectedClustersClient
+	ExtensionsClient                  *extensions.ExtensionsClient
+	FluxConfigurationClient           *fluxconfiguration.FluxConfigurationClient
+	ProvisionedClusterInstancesClient *provisionedclusterinstances.ProvisionedClusterInstancesClient
 }
 
 func NewClient(o *common.ClientOptions) (*Client, error) {
@@ -37,9 +39,16 @@ func NewClient(o *common.ClientOptions) (*Client, error) {
 	}
 	o.Configure(fluxConfigurationClient.Client, o.Authorizers.ResourceManager)
 
+	provisionedClusterInstancesClientClient, err := provisionedclusterinstances.NewProvisionedClusterInstancesClientWithBaseURI(o.Environment.ResourceManager)
+	if err != nil {
+		return nil, fmt.Errorf("building ProvisionedClusterInstancesClient client: %+v", err)
+	}
+	o.Configure(provisionedClusterInstancesClientClient.Client, o.Authorizers.ResourceManager)
+
 	return &Client{
-		ArcKubernetesClient:     arcKubernetesClient,
-		ExtensionsClient:        extensionsClient,
-		FluxConfigurationClient: fluxConfigurationClient,
+		ArcKubernetesClient:               arcKubernetesClient,
+		ExtensionsClient:                  extensionsClient,
+		FluxConfigurationClient:           fluxConfigurationClient,
+		ProvisionedClusterInstancesClient: provisionedClusterInstancesClientClient,
 	}, nil
 }