@@ -33,6 +33,7 @@ import (
33
33
"k8s.io/client-go/kubernetes/scheme"
34
34
clusterv1 "sigs.k8s.io/cluster-api/api/v1alpha3"
35
35
"sigs.k8s.io/cluster-api/controllers/external"
36
+ "sigs.k8s.io/cluster-api/util"
36
37
"sigs.k8s.io/controller-runtime/pkg/client"
37
38
"sigs.k8s.io/controller-runtime/pkg/controller"
38
39
"sigs.k8s.io/controller-runtime/pkg/envtest"
@@ -114,7 +115,7 @@ var _ = Describe("MachineHealthCheck Reconciler", func() {
114
115
115
116
Eventually (func () map [string ]string {
116
117
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 )
118
119
if err != nil {
119
120
return nil
120
121
}
@@ -165,7 +166,7 @@ var _ = Describe("MachineHealthCheck Reconciler", func() {
165
166
166
167
Eventually (func () []metav1.OwnerReference {
167
168
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 )
169
170
if err != nil {
170
171
return []metav1.OwnerReference {}
171
172
}
@@ -207,7 +208,7 @@ func cleanupTestMachineHealthChecks(ctx context.Context, c client.Client) error
207
208
func ownerReferenceForCluster (ctx context.Context , c * clusterv1.Cluster ) metav1.OwnerReference {
208
209
// Fetch the cluster to populate the UID
209
210
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 ())
211
212
212
213
return metav1.OwnerReference {
213
214
APIVersion : clusterv1 .GroupVersion .String (),
@@ -341,11 +342,8 @@ func TestClusterToMachineHealthCheck(t *testing.T) {
341
342
gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
342
343
}()
343
344
// Check the cache is populated
344
- key , err := client .ObjectKeyFromObject (& o )
345
- gs .Expect (err ).ToNot (HaveOccurred ())
346
-
347
345
getObj := func () error {
348
- return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
346
+ return r .Client .Get (ctx , util . ObjectKey ( & o ) , & clusterv1.MachineHealthCheck {})
349
347
}
350
348
gs .Eventually (getObj , timeout ).Should (Succeed ())
351
349
}
@@ -539,11 +537,8 @@ func TestMachineToMachineHealthCheck(t *testing.T) {
539
537
gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
540
538
}()
541
539
// Check the cache is populated
542
- key , err := client .ObjectKeyFromObject (& o )
543
- gs .Expect (err ).ToNot (HaveOccurred ())
544
-
545
540
getObj := func () error {
546
- return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
541
+ return r .Client .Get (ctx , util . ObjectKey ( & o ) , & clusterv1.MachineHealthCheck {})
547
542
}
548
543
gs .Eventually (getObj , timeout ).Should (Succeed ())
549
544
}
@@ -675,7 +670,7 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
675
670
expected : []reconcile.Request {mhc1Req },
676
671
},
677
672
{
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" ,
679
674
mhcToCreate : []clusterv1.MachineHealthCheck {* mhc1 , * mhc2 },
680
675
mToCreate : []clusterv1.Machine {* machine1 },
681
676
object : handler.MapObject {
@@ -706,9 +701,7 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
706
701
gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
707
702
}()
708
703
// Check the cache is populated
709
- key , err := client .ObjectKeyFromObject (& o )
710
- gs .Expect (err ).ToNot (HaveOccurred ())
711
-
704
+ key := util .ObjectKey (& o )
712
705
getObj := func () error {
713
706
return r .Client .Get (ctx , key , & clusterv1.MachineHealthCheck {})
714
707
}
@@ -720,14 +713,12 @@ func TestNodeToMachineHealthCheck(t *testing.T) {
720
713
defer func () {
721
714
gs .Expect (r .Client .Delete (ctx , & o )).To (Succeed ())
722
715
}()
723
- // Ensure the status is set (required for matchin node to machine)
716
+ // Ensure the status is set (required for matching node to machine)
724
717
o .Status = obj .Status
725
718
gs .Expect (r .Client .Status ().Update (ctx , & o )).To (Succeed ())
726
719
727
720
// 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 )
731
722
checkStatus := func () clusterv1.MachineStatus {
732
723
m := & clusterv1.Machine {}
733
724
err := r .Client .Get (ctx , key , m )
0 commit comments