From 907f3612a6d60f8c4b3f1db239dff38572fc826f Mon Sep 17 00:00:00 2001
From: willie-yao <williamyao@microsoft.com>
Date: Mon, 27 Feb 2023 22:38:33 +0000
Subject: [PATCH] Add validate create test

---
 ...anagedcontrolplanetemplate_webhook_test.go | 88 ++++++++++++++++++-
 1 file changed, 87 insertions(+), 1 deletion(-)

diff --git a/api/v1beta1/azuremanagedcontrolplanetemplate_webhook_test.go b/api/v1beta1/azuremanagedcontrolplanetemplate_webhook_test.go
index c7979ac8e40..b258bb19c85 100644
--- a/api/v1beta1/azuremanagedcontrolplanetemplate_webhook_test.go
+++ b/api/v1beta1/azuremanagedcontrolplanetemplate_webhook_test.go
@@ -20,10 +20,63 @@ import (
 	"testing"
 
 	. "github.com/onsi/gomega"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	utilfeature "k8s.io/component-base/featuregate/testing"
 	"k8s.io/utils/pointer"
+	"sigs.k8s.io/cluster-api-provider-azure/feature"
+	capifeature "sigs.k8s.io/cluster-api/feature"
 )
 
-func TestAzureManagedControlPlaneTemplateTemplateValidateUpdate(t *testing.T) {
+func TestAzureManagedControlPlaneTemplate_ValidateCreate(t *testing.T) {
+	// NOTE: AzureManageControlPlaneTemplate is behind AKS feature gate flag; the webhook
+	// must prevent creating new objects in case the feature flag is disabled.
+	defer utilfeature.SetFeatureGateDuringTest(t, feature.Gates, capifeature.MachinePool, true)()
+	g := NewWithT(t)
+
+	tests := []struct {
+		name    string
+		amcpt   *AzureManagedControlPlaneTemplate
+		wantErr bool
+	}{
+		{
+			name:    "all valid",
+			amcpt:   getKnownValidAzureManagedControlPlaneTemplate(),
+			wantErr: false,
+		},
+		{
+			name:    "invalid DNSServiceIP",
+			amcpt:   createAzureManagedControlPlaneTemplate(metav1.ObjectMeta{}, "192.168.0.0.3", "v1.18.0"),
+			wantErr: true,
+		},
+		{
+			name:    "invalid version",
+			amcpt:   createAzureManagedControlPlaneTemplate(metav1.ObjectMeta{}, "192.168.0.0", "honk.version"),
+			wantErr: true,
+		},
+		{
+			name:    "invalid name with microsoft",
+			amcpt:   createAzureManagedControlPlaneTemplate(metav1.ObjectMeta{Name: "microsoft-cluster"}, "192.168.0.0", "v1.18.0"),
+			wantErr: true,
+		},
+		{
+			name:    "invalid name with windows",
+			amcpt:   createAzureManagedControlPlaneTemplate(metav1.ObjectMeta{Name: "a-windows-cluster"}, "192.168.0.0", "v1.18.0"),
+			wantErr: true,
+		},
+	}
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			err := tc.amcpt.ValidateCreate(nil)
+			if tc.wantErr {
+				g.Expect(err).To(HaveOccurred())
+			} else {
+				g.Expect(err).NotTo(HaveOccurred())
+			}
+		})
+	}
+}
+
+func TestAzureManagedControlPlaneTemplate_ValidateUpdate(t *testing.T) {
 
 	oldManagedControlPlaneTemplate := &AzureManagedControlPlaneTemplate{
 		Spec: AzureManagedControlPlaneTemplateSpec{
@@ -50,3 +103,36 @@ func TestAzureManagedControlPlaneTemplateTemplateValidateUpdate(t *testing.T) {
 		g.Expect(newManagedControlPlaneTemplate.ValidateUpdate(oldManagedControlPlaneTemplate, nil)).NotTo(Succeed())
 	})
 }
+
+func createAzureManagedControlPlaneTemplate(objectMeta metav1.ObjectMeta, serviceIP, version string) *AzureManagedControlPlaneTemplate {
+	return &AzureManagedControlPlaneTemplate{
+		ObjectMeta: objectMeta,
+		Spec: AzureManagedControlPlaneTemplateSpec{
+			Template: AzureManagedControlPlaneTemplateResource{
+				Spec: AzureManagedControlPlaneTemplateResourceSpec{
+					DNSServiceIP: pointer.String(serviceIP),
+					Version:      version,
+				},
+			},
+		},
+	}
+}
+
+func getKnownValidAzureManagedControlPlaneTemplate() *AzureManagedControlPlaneTemplate {
+	return &AzureManagedControlPlaneTemplate{
+		Spec: AzureManagedControlPlaneTemplateSpec{
+			Template: AzureManagedControlPlaneTemplateResource{
+				Spec: AzureManagedControlPlaneTemplateResourceSpec{
+					DNSServiceIP: pointer.String("192.168.0.0"),
+					Version:      "v1.18.0",
+					AADProfile: &AADProfile{
+						Managed: true,
+						AdminGroupObjectIDs: []string{
+							"616077a8-5db7-4c98-b856-b34619afg75h",
+						},
+					},
+				},
+			},
+		},
+	}
+}