@@ -35,6 +35,7 @@ import (
3535 clientgoscheme "k8s.io/client-go/kubernetes/scheme"
3636 "k8s.io/utils/ptr"
3737 clusterv1beta1 "sigs.k8s.io/cluster-api/api/core/v1beta1"
38+ clusterv1 "sigs.k8s.io/cluster-api/api/core/v1beta2"
3839 clusterctlv1 "sigs.k8s.io/cluster-api/cmd/clusterctl/api/v1alpha3"
3940 "sigs.k8s.io/controller-runtime/pkg/client"
4041 "sigs.k8s.io/controller-runtime/pkg/client/fake"
@@ -82,7 +83,7 @@ func TestAzureClusterToAzureMachinesMapper(t *testing.T) {
8283 {
8384 Name : clusterName ,
8485 Kind : "Cluster" ,
85- APIVersion : clusterv1beta1 .GroupVersion .String (),
86+ APIVersion : clusterv1 .GroupVersion .String (),
8687 },
8788 },
8889 },
@@ -376,28 +377,28 @@ func setupScheme(g *WithT) *runtime.Scheme {
376377 g .Expect (clientgoscheme .AddToScheme (scheme )).To (Succeed ())
377378 g .Expect (infrav1 .AddToScheme (scheme )).To (Succeed ())
378379 g .Expect (clusterv1beta1 .AddToScheme (scheme )).To (Succeed ())
380+ g .Expect (clusterv1 .AddToScheme (scheme )).To (Succeed ())
379381 return scheme
380382}
381383
382- func newMachine (clusterName , machineName string ) * clusterv1beta1 .Machine {
383- return & clusterv1beta1 .Machine {
384+ func newMachine (clusterName , machineName string ) * clusterv1 .Machine {
385+ return & clusterv1 .Machine {
384386 ObjectMeta : metav1.ObjectMeta {
385387 Labels : map [string ]string {
386- clusterv1beta1 .ClusterNameLabel : clusterName ,
388+ clusterv1 .ClusterNameLabel : clusterName ,
387389 },
388390 Name : machineName ,
389391 Namespace : "default" ,
390392 },
391393 }
392394}
393395
394- func newMachineWithInfrastructureRef (clusterName , machineName string ) * clusterv1beta1 .Machine {
396+ func newMachineWithInfrastructureRef (clusterName , machineName string ) * clusterv1 .Machine {
395397 m := newMachine (clusterName , machineName )
396- m .Spec .InfrastructureRef = corev1.ObjectReference {
397- Kind : "AzureMachine" ,
398- Namespace : "default" ,
399- Name : "azure" + machineName ,
400- APIVersion : infrav1 .GroupVersion .String (),
398+ m .Spec .InfrastructureRef = clusterv1.ContractVersionedObjectReference {
399+ Kind : "AzureMachine" ,
400+ Name : "azure" + machineName ,
401+ APIGroup : infrav1 .GroupVersion .Group ,
401402 }
402403 return m
403404}
0 commit comments