Tests for updating the GitOps templates.

Further tests for the GitOps templates.
weaveworks · Mar 13, 2023 · 16ff757 · 16ff757
1 parent 3f69119
commit 16ff757
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 1 deletion.
diff --git a/controllers/capitemplate_controller_test.go b/controllers/capitemplate_controller_test.go
@@ -13,9 +13,11 @@ import (
 	capiv1alpha1 "github.com/weaveworks/templates-controller/apis/capi/v1alpha1"
 	capiv1alpha2 "github.com/weaveworks/templates-controller/apis/capi/v1alpha2"
 	"github.com/weaveworks/templates-controller/apis/core"
+	gitopsv1alpha1 "github.com/weaveworks/templates-controller/apis/gitops/v1alpha1"
+	gitopsv1alpha2 "github.com/weaveworks/templates-controller/apis/gitops/v1alpha2"
 )
 
-func TestCAPITemplateConversion(t *testing.T) {
+func TestCAPITemplateV1Alpha1Conversion(t *testing.T) {
 	ctx := context.TODO()
 
 	ct := &capiv1alpha1.CAPITemplate{
@@ -73,6 +75,64 @@ func TestCAPITemplateConversion(t *testing.T) {
 	}
 }
 
+func TestGitOpsTemplateV1Alpha1Conversion(t *testing.T) {
+	ctx := context.TODO()
+
+	ct := &gitopsv1alpha1.GitOpsTemplate{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-template",
+			Namespace: "default",
+		},
+		TypeMeta: metav1.TypeMeta{
+			Kind:       capiv1alpha1.Kind,
+			APIVersion: "capi.weave.works/v1alpha1",
+		},
+		Spec: core.TemplateSpecV1{
+			Description: "this is test template 1",
+			Params: []core.TemplateParam{
+				{
+					Name:        "CLUSTER_NAME",
+					Description: "This is used for the cluster naming.",
+					Required:    false,
+				},
+			},
+		},
+	}
+
+	assertNoError(t, testEnv.Create(ctx, ct))
+	defer cleanupResource(t, testEnv, ct)
+
+	updated := &gitopsv1alpha2.GitOpsTemplate{}
+	assertNoError(t, testEnv.Get(ctx, client.ObjectKeyFromObject(ct), updated))
+
+	want := &gitopsv1alpha2.GitOpsTemplate{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-template",
+			Namespace: "default",
+		},
+		TypeMeta: metav1.TypeMeta{
+			Kind:       gitopsv1alpha2.Kind,
+			APIVersion: "templates.weave.works/v1alpha2",
+		},
+		Spec: core.TemplateSpec{
+			Description: "this is test template 1",
+			RenderType:  "envsubst",
+			Params: []core.TemplateParam{
+				{
+					Name:        "CLUSTER_NAME",
+					Description: "This is used for the cluster naming.",
+					Required:    false,
+				},
+			},
+			ResourceTemplates: []core.ResourceTemplate{{}},
+		},
+	}
+
+	if diff := cmp.Diff(want, updated, ignoreObjectMeta); diff != "" {
+		t.Fatalf("failed to update the template:\n%s\n", diff)
+	}
+}
+
 var ignoreObjectMeta = cmpopts.IgnoreFields(metav1.ObjectMeta{}, "ResourceVersion", "UID", "Generation", "CreationTimestamp", "ManagedFields")
 
 func cleanupResource(t *testing.T, cl client.Client, obj client.Object) {

diff --git a/controllers/main_test.go b/controllers/main_test.go
@@ -15,6 +15,8 @@ import (
 
 	capiv1alpha1 "github.com/weaveworks/templates-controller/apis/capi/v1alpha1"
 	capiv1alpha2 "github.com/weaveworks/templates-controller/apis/capi/v1alpha2"
+	gitopsv1alpha1 "github.com/weaveworks/templates-controller/apis/gitops/v1alpha1"
+	gitopsv1alpha2 "github.com/weaveworks/templates-controller/apis/gitops/v1alpha2"
 )
 
 var (
@@ -29,6 +31,8 @@ func init() {
 func TestMain(m *testing.M) {
 	utilruntime.Must(capiv1alpha1.AddToScheme(scheme.Scheme))
 	utilruntime.Must(capiv1alpha2.AddToScheme(scheme.Scheme))
+	utilruntime.Must(gitopsv1alpha1.AddToScheme(scheme.Scheme))
+	utilruntime.Must(gitopsv1alpha2.AddToScheme(scheme.Scheme))
 	testEnv = testenv.New(testenv.WithCRDPath(filepath.Join("..", "config", "crd", "bases")))
 
 	if err := (&CAPITemplateReconciler{