From 80c72756a824d08c2834368fae758d4171124963 Mon Sep 17 00:00:00 2001 From: Avinash Patnala Date: Wed, 23 Oct 2024 18:46:14 +0000 Subject: [PATCH] update test cases Signed-off-by: Avinash Patnala --- .../constrainttemplate_controller.go | 14 +++++++------- .../constrainttemplate_controller_test.go | 9 +++++---- 2 files changed, 12 insertions(+), 11 deletions(-) diff --git a/pkg/controller/constrainttemplate/constrainttemplate_controller.go b/pkg/controller/constrainttemplate/constrainttemplate_controller.go index 45a530cf66b..53403156257 100644 --- a/pkg/controller/constrainttemplate/constrainttemplate_controller.go +++ b/pkg/controller/constrainttemplate/constrainttemplate_controller.go @@ -22,7 +22,6 @@ import ( "reflect" "time" - "github.com/go-logr/logr" "github.com/open-policy-agent/frameworks/constraint/pkg/apis/templates/v1beta1" constraintclient "github.com/open-policy-agent/frameworks/constraint/pkg/client" "github.com/open-policy-agent/frameworks/constraint/pkg/core/templates" @@ -70,12 +69,13 @@ var ( discoveryErr *apiutil.ErrResourceDiscoveryFailed ) -var g *logr.Logger -var gvkConstraintTemplate = schema.GroupVersionKind{ - Group: v1beta1.SchemeGroupVersion.Group, - Version: v1beta1.SchemeGroupVersion.Version, - Kind: "ConstraintTemplate", -} +var ( + gvkConstraintTemplate = schema.GroupVersionKind{ + Group: v1beta1.SchemeGroupVersion.Group, + Version: v1beta1.SchemeGroupVersion.Version, + Kind: "ConstraintTemplate", + } +) type Adder struct { CFClient *constraintclient.Client diff --git a/pkg/controller/constrainttemplate/constrainttemplate_controller_test.go b/pkg/controller/constrainttemplate/constrainttemplate_controller_test.go index b1b282c13ed..895043b253f 100644 --- a/pkg/controller/constrainttemplate/constrainttemplate_controller_test.go +++ b/pkg/controller/constrainttemplate/constrainttemplate_controller_test.go @@ -32,6 +32,7 @@ import ( "github.com/open-policy-agent/gatekeeper/v3/pkg/controller/constraint" "github.com/open-policy-agent/gatekeeper/v3/pkg/drivers/k8scel" celSchema "github.com/open-policy-agent/gatekeeper/v3/pkg/drivers/k8scel/schema" + "github.com/open-policy-agent/gatekeeper/v3/pkg/drivers/k8scel/transform" "github.com/open-policy-agent/gatekeeper/v3/pkg/fakes" "github.com/open-policy-agent/gatekeeper/v3/pkg/readiness" "github.com/open-policy-agent/gatekeeper/v3/pkg/target" @@ -271,8 +272,8 @@ func TestReconcile(t *testing.T) { ctx := context.Background() testutils.StartManager(ctx, t, mgr) - constraint.VapAPIEnabled = ptr.To[bool](true) - constraint.GroupVersion = &admissionregistrationv1beta1.SchemeGroupVersion + transform.VapAPIEnabled = ptr.To[bool](true) + transform.GroupVersion = &admissionregistrationv1beta1.SchemeGroupVersion t.Run("CRD Gets Created", func(t *testing.T) { suffix := "CRDGetsCreated" @@ -696,7 +697,7 @@ func TestReconcile(t *testing.T) { suffix := "VapShouldBeCreatedV1" logger.Info("Running test: Vap should be created with v1") - constraint.GroupVersion = &admissionregistrationv1.SchemeGroupVersion + transform.GroupVersion = &admissionregistrationv1.SchemeGroupVersion constraintTemplate := makeReconcileConstraintTemplateForVap(suffix, ptr.To[bool](true)) t.Cleanup(testutils.DeleteObjectAndConfirm(ctx, t, c, expectedCRD(suffix))) testutils.CreateThenCleanup(ctx, t, c, constraintTemplate) @@ -721,7 +722,7 @@ func TestReconcile(t *testing.T) { suffix := "VapBindingShouldBeCreatedV1" logger.Info("Running test: VapBinding should be created with v1") constraint.DefaultGenerateVAPB = ptr.To[bool](true) - constraint.GroupVersion = &admissionregistrationv1.SchemeGroupVersion + transform.GroupVersion = &admissionregistrationv1.SchemeGroupVersion constraintTemplate := makeReconcileConstraintTemplateForVap(suffix, ptr.To[bool](true)) cstr := newDenyAllCstr(suffix) t.Cleanup(testutils.DeleteObjectAndConfirm(ctx, t, c, expectedCRD(suffix)))