@@ -37,7 +37,6 @@ import (
3737 logf "sigs.k8s.io/controller-runtime/pkg/log"
3838 "sigs.k8s.io/controller-runtime/pkg/log/zap"
3939 "sigs.k8s.io/controller-runtime/pkg/manager"
40- "sigs.k8s.io/controller-runtime/pkg/scheme"
4140 "sigs.k8s.io/controller-runtime/pkg/webhook/admission"
4241)
4342
@@ -90,10 +89,7 @@ func runTests(admissionReviewVersion string) {
9089 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
9190
9291 By ("registering the type in the Scheme" )
93- builder := scheme.Builder {GroupVersion : testDefaulterGVK .GroupVersion ()}
94- builder .Register (& TestDefaulterObject {}, & TestDefaulterList {})
95- err = builder .AddToScheme (m .GetScheme ())
96- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
92+ addToScheme (m .GetScheme ())
9793
9894 webhookBuilder := WebhookManagedBy (m , & TestDefaulterObject {})
9995 build (webhookBuilder )
@@ -174,10 +170,7 @@ func runTests(admissionReviewVersion string) {
174170 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
175171
176172 By ("registering the type in the Scheme" )
177- builder := scheme.Builder {GroupVersion : testDefaulterGVK .GroupVersion ()}
178- builder .Register (& TestDefaulterObject {}, & TestDefaulterList {})
179- err = builder .AddToScheme (m .GetScheme ())
180- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
173+ addToScheme (m .GetScheme ())
181174
182175 customPath := "/custom-defaulting-path"
183176 webhookBuilder := WebhookManagedBy (m , & TestDefaulterObject {})
@@ -261,10 +254,7 @@ func runTests(admissionReviewVersion string) {
261254 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
262255
263256 By ("registering the type in the Scheme" )
264- builder := scheme.Builder {GroupVersion : testDefaulterGVK .GroupVersion ()}
265- builder .Register (& TestDefaulterObject {}, & TestDefaulterList {})
266- err = builder .AddToScheme (m .GetScheme ())
267- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
257+ addToScheme (m .GetScheme ())
268258
269259 webhookBuilder := WebhookManagedBy (m , & TestDefaulterObject {})
270260 build (webhookBuilder )
@@ -332,10 +322,7 @@ func runTests(admissionReviewVersion string) {
332322 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
333323
334324 By ("registering the type in the Scheme" )
335- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
336- builder .Register (& TestValidatorObject {}, & TestValidatorList {})
337- err = builder .AddToScheme (m .GetScheme ())
338- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
325+ addToScheme (m .GetScheme ())
339326
340327 webhook := WebhookManagedBy (m , & TestValidatorObject {})
341328 build (webhook )
@@ -457,10 +444,7 @@ func runTests(admissionReviewVersion string) {
457444 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
458445
459446 By ("registering the type in the Scheme" )
460- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
461- builder .Register (& TestValidatorObject {}, & TestValidatorList {})
462- err = builder .AddToScheme (m .GetScheme ())
463- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
447+ addToScheme (m .GetScheme ())
464448
465449 customPath := "/custom-validating-path"
466450 webhookBuilder := WebhookManagedBy (m , & TestValidatorObject {})
@@ -544,10 +528,7 @@ func runTests(admissionReviewVersion string) {
544528 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
545529
546530 By ("registering the type in the Scheme" )
547- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
548- builder .Register (& TestValidatorObject {}, & TestValidatorList {})
549- err = builder .AddToScheme (m .GetScheme ())
550- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
531+ addToScheme (m .GetScheme ())
551532
552533 webhookBuilder := WebhookManagedBy (m , & TestValidatorObject {})
553534 build (webhookBuilder )
@@ -617,10 +598,7 @@ func runTests(admissionReviewVersion string) {
617598 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
618599
619600 By ("registering the type in the Scheme" )
620- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
621- builder .Register (& TestValidatorObject {}, & TestValidatorList {})
622- err = builder .AddToScheme (m .GetScheme ())
623- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
601+ addToScheme (m .GetScheme ())
624602
625603 webhookBuilder := WebhookManagedBy (m , & TestValidatorObject {})
626604 build (webhookBuilder )
@@ -715,10 +693,7 @@ func runTests(admissionReviewVersion string) {
715693 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
716694
717695 By ("registering the type in the Scheme" )
718- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
719- builder .Register (& TestDefaultValidator {}, & TestDefaultValidatorList {})
720- err = builder .AddToScheme (m .GetScheme ())
721- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
696+ addToScheme (m .GetScheme ())
722697
723698 webhookBuilder := WebhookManagedBy (m , & TestDefaultValidator {})
724699 build (webhookBuilder )
@@ -815,10 +790,7 @@ func runTests(admissionReviewVersion string) {
815790 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
816791
817792 By ("registering the type in the Scheme" )
818- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
819- builder .Register (& TestDefaultValidator {}, & TestDefaultValidatorList {})
820- err = builder .AddToScheme (m .GetScheme ())
821- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
793+ addToScheme (m .GetScheme ())
822794
823795 validatingCustomPath := "/custom-validating-path"
824796 defaultingCustomPath := "/custom-defaulting-path"
@@ -936,10 +908,7 @@ func runTests(admissionReviewVersion string) {
936908 m , err := manager .New (cfg , manager.Options {})
937909 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
938910
939- builder := scheme.Builder {GroupVersion : testDefaulterGVK .GroupVersion ()}
940- builder .Register (& TestDefaulterObject {}, & TestDefaulterList {})
941- err = builder .AddToScheme (m .GetScheme ())
942- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
911+ addToScheme (m .GetScheme ())
943912
944913 err = WebhookManagedBy (m , & TestDefaulterObject {}).
945914 WithDefaulter (& testDefaulter {}).
@@ -952,10 +921,7 @@ func runTests(admissionReviewVersion string) {
952921 m , err := manager .New (cfg , manager.Options {})
953922 ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
954923
955- builder := scheme.Builder {GroupVersion : testValidatorGVK .GroupVersion ()}
956- builder .Register (& TestValidatorObject {}, & TestValidatorList {})
957- err = builder .AddToScheme (m .GetScheme ())
958- ExpectWithOffset (1 , err ).NotTo (HaveOccurred ())
924+ addToScheme (m .GetScheme ())
959925
960926 err = WebhookManagedBy (m , & TestValidatorObject {}).
961927 WithValidator (& testValidator {}).
@@ -978,6 +944,18 @@ type TestDefaulterObject struct {
978944
979945var testDefaulterGVK = schema.GroupVersionKind {Group : "foo.test.org" , Version : "v1" , Kind : testDefaulterKind }
980946
947+ func addToScheme (scheme * runtime.Scheme ) {
948+ scheme .AddKnownTypes (testDefaulterGVK .GroupVersion (),
949+ & TestDefaulterObject {},
950+ & TestDefaulterList {},
951+ & TestValidatorObject {},
952+ & TestValidatorList {},
953+ & TestDefaultValidator {},
954+ & TestDefaultValidatorList {},
955+ )
956+ metav1 .AddToGroupVersion (scheme , testDefaulterGVK .GroupVersion ())
957+ }
958+
981959func (d * TestDefaulterObject ) GetObjectKind () schema.ObjectKind { return d }
982960func (d * TestDefaulterObject ) DeepCopyObject () runtime.Object {
983961 return & TestDefaulterObject {
0 commit comments