@@ -33,6 +33,7 @@ import (
3333 "k8s.io/client-go/kubernetes/scheme"
3434 clusterv1 "sigs.k8s.io/cluster-api/api/v1alpha3"
3535 "sigs.k8s.io/cluster-api/controllers/external"
36+ "sigs.k8s.io/cluster-api/util"
3637 "sigs.k8s.io/controller-runtime/pkg/client"
3738 "sigs.k8s.io/controller-runtime/pkg/controller"
3839 "sigs.k8s.io/controller-runtime/pkg/envtest"
@@ -114,7 +115,7 @@ var _ = Describe("MachineHealthCheck Reconciler", func() {
114115
115116 Eventually (func () map [string ]string {
116117 mhc := & clusterv1.MachineHealthCheck {}
117- err := k8sClient .Get (ctx , client .ObjectKey { Namespace : mhcToCreate . GetNamespace (), Name : mhcToCreate . GetName ()} , mhc )
118+ err := k8sClient .Get (ctx , util .ObjectKey ( mhcToCreate ) , mhc )
118119 if err != nil {
119120 return nil
120121 }
@@ -165,7 +166,7 @@ var _ = Describe("MachineHealthCheck Reconciler", func() {
165166
166167 Eventually (func () []metav1.OwnerReference {
167168 mhc := & clusterv1.MachineHealthCheck {}
168- err := k8sClient .Get (ctx , client .ObjectKey { Namespace : mhcToCreate . GetNamespace (), Name : mhcToCreate . GetName ()} , mhc )
169+ err := k8sClient .Get (ctx , util .ObjectKey ( mhcToCreate ) , mhc )
169170 if err != nil {
170171 return []metav1.OwnerReference {}
171172 }
@@ -207,7 +208,7 @@ func cleanupTestMachineHealthChecks(ctx context.Context, c client.Client) error
207208func ownerReferenceForCluster (ctx context.Context , c * clusterv1.Cluster ) metav1.OwnerReference {
208209 // Fetch the cluster to populate the UID
209210 cc := & clusterv1.Cluster {}
210- Expect (k8sClient .Get (ctx , client .ObjectKey { Namespace : c . GetNamespace (), Name : c . GetName ()} , cc )).To (Succeed ())
211+ Expect (k8sClient .Get (ctx , util .ObjectKey ( c ) , cc )).To (Succeed ())
211212
212213 return metav1.OwnerReference {
213214 APIVersion : clusterv1 .GroupVersion .String (),
@@ -341,11 +342,8 @@ func TestClusterToMachineHealthCheck(t *testing.T) {
341342 gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
342343 }()
343344 // Check the cache is populated
344- key , err := client .ObjectKeyFromObject (& o )
345- gs .Expect (err ).ToNot (HaveOccurred ())
346-
347345 getObj := func () error {
348- return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
346+ return r .Client .Get (ctx , util . ObjectKey ( & o ) , & clusterv1.MachineHealthCheck {})
349347 }
350348 gs .Eventually (getObj , timeout ).Should (Succeed ())
351349 }
@@ -539,11 +537,8 @@ func TestMachineToMachineHealthCheck(t *testing.T) {
539537 gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
540538 }()
541539 // Check the cache is populated
542- key , err := client .ObjectKeyFromObject (& o )
543- gs .Expect (err ).ToNot (HaveOccurred ())
544-
545540 getObj := func () error {
546- return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
541+ return r .Client .Get (ctx , util . ObjectKey ( & o ) , & clusterv1.MachineHealthCheck {})
547542 }
548543 gs .Eventually (getObj , timeout ).Should (Succeed ())
549544 }
@@ -675,7 +670,7 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
675670 expected : []reconcile.Request {mhc1Req },
676671 },
677672 {
678- name : "when two NachineHealthChecks exist for the Node in the Machine's namespace" ,
673+ name : "when two MachineHealthChecks exist for the Node in the Machine's namespace" ,
679674 mhcToCreate : []clusterv1.MachineHealthCheck {* mhc1 , * mhc2 },
680675 mToCreate : []clusterv1.Machine {* machine1 },
681676 object : handler.MapObject {
@@ -706,9 +701,7 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
706701 gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
707702 }()
708703 // Check the cache is populated
709- key , err := client .ObjectKeyFromObject (& o )
710- gs .Expect (err ).ToNot (HaveOccurred ())
711-
704+ key := util .ObjectKey (& o )
712705 getObj := func () error {
713706 return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
714707 }
@@ -720,14 +713,12 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
720713 defer func () {
721714 gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
722715 }()
723- // Ensure the status is set (required for matchin node to machine)
716+ // Ensure the status is set (required for matching node to machine)
724717 o .Status = obj .Status
725718 gs .Expect (r .Client .Status ().Update (ctx , & o )).To (Succeed ())
726719
727720 // Check the cache is up to date with the status update
728- key , err := client .ObjectKeyFromObject (& o )
729- gs .Expect (err ).ToNot (HaveOccurred ())
730-
721+ key := util .ObjectKey (& o )
731722 checkStatus := func () clusterv1.MachineStatus {
732723 m := & clusterv1.Machine {}
733724 err := r .Client .Get (ctx , key , m )
0 commit comments