@@ -28,19 +28,16 @@ import (
2828 . "github.com/onsi/gomega"
2929 corev1 "k8s.io/api/core/v1"
3030 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
31- "k8s.io/apimachinery/pkg/types"
3231 "k8s.io/client-go/tools/record"
3332 "k8s.io/utils/ptr"
3433 ctrl "sigs.k8s.io/controller-runtime"
3534 "sigs.k8s.io/controller-runtime/pkg/client"
3635 "sigs.k8s.io/controller-runtime/pkg/client/fake"
37- "sigs.k8s.io/controller-runtime/pkg/reconcile"
3836
3937 infrav1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
4038 "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/scope"
4139 "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/services"
4240 "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/services/mock_services"
43- "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/logger"
4441 clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
4542 "sigs.k8s.io/cluster-api/util"
4643)
@@ -60,7 +57,8 @@ func TestAWSClusterReconcilerReconcile(t *testing.T) {
6057 Kind : "Cluster" ,
6158 Name : "capi-fail-test" ,
6259 UID : "1" ,
63- }}}},
60+ },
61+ }}},
6462 expectError : true ,
6563 },
6664 {
@@ -532,97 +530,6 @@ func TestAWSClusterReconcileOperations(t *testing.T) {
532530 })
533531}
534532
535- func TestAWSClusterReconcilerRequeueAWSClusterForUnpausedCluster (t * testing.T ) {
536- testCases := []struct {
537- name string
538- awsCluster * infrav1.AWSCluster
539- ownerCluster * clusterv1.Cluster
540- requeue bool
541- }{
542- {
543- name : "Should create reconcile request successfully" ,
544- awsCluster : & infrav1.AWSCluster {
545- ObjectMeta : metav1.ObjectMeta {GenerateName : "aws-test-" }, TypeMeta : metav1.TypeMeta {Kind : "AWSCluster" , APIVersion : infrav1 .GroupVersion .String ()},
546- },
547- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" }},
548- requeue : true ,
549- },
550- {
551- name : "Should not create reconcile request if AWSCluster is externally managed" ,
552- awsCluster : & infrav1.AWSCluster {
553- ObjectMeta : metav1.ObjectMeta {GenerateName : "aws-test-" , Annotations : map [string ]string {clusterv1 .ManagedByAnnotation : "capi-test" }},
554- TypeMeta : metav1.TypeMeta {Kind : "AWSCluster" , APIVersion : infrav1 .GroupVersion .String ()},
555- },
556- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" }},
557- requeue : false ,
558- },
559- {
560- name : "Should not create reconcile request for deleted clusters" ,
561- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" , DeletionTimestamp : & metav1.Time {Time : time .Now ()}}},
562- requeue : false ,
563- },
564- {
565- name : "Should not create reconcile request if infrastructure ref for AWSCluster on owner cluster is not set" ,
566- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" }},
567- requeue : false ,
568- },
569- {
570- name : "Should not create reconcile request if infrastructure ref type on owner cluster is not AWSCluster" ,
571- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" }, Spec : clusterv1.ClusterSpec {InfrastructureRef : & corev1.ObjectReference {
572- APIVersion : clusterv1 .GroupVersion .String (),
573- Kind : "Cluster" ,
574- Name : "aws-test" }}},
575- requeue : false ,
576- },
577- {
578- name : "Should not create reconcile request if AWSCluster not found" ,
579- ownerCluster : & clusterv1.Cluster {ObjectMeta : metav1.ObjectMeta {Name : "capi-test" }, Spec : clusterv1.ClusterSpec {InfrastructureRef : & corev1.ObjectReference {
580- APIVersion : clusterv1 .GroupVersion .String (),
581- Kind : "AWSCluster" ,
582- Name : "aws-test" }}},
583- requeue : false ,
584- },
585- }
586- for _ , tc := range testCases {
587- t .Run (tc .name , func (t * testing.T ) {
588- g := NewWithT (t )
589- log := logger .FromContext (ctx )
590- reconciler := & AWSClusterReconciler {
591- Client : testEnv .Client ,
592- }
593-
594- ns , err := testEnv .CreateNamespace (ctx , fmt .Sprintf ("namespace-%s" , util .RandomString (5 )))
595- g .Expect (err ).To (BeNil ())
596- createCluster (g , tc .awsCluster , ns .Name )
597- defer cleanupCluster (g , tc .awsCluster , ns )
598-
599- if tc .ownerCluster != nil {
600- if tc .awsCluster != nil {
601- tc .ownerCluster .Spec = clusterv1.ClusterSpec {InfrastructureRef : & corev1.ObjectReference {
602- APIVersion : infrav1 .GroupVersion .String (),
603- Kind : "AWSCluster" ,
604- Name : tc .awsCluster .Name ,
605- Namespace : ns .Name ,
606- }}
607- }
608- tc .ownerCluster .Namespace = ns .Name
609- }
610- handlerFunc := reconciler .requeueAWSClusterForUnpausedCluster (ctx , log )
611- result := handlerFunc (ctx , tc .ownerCluster )
612- if tc .requeue {
613- g .Expect (result ).To (ContainElement (reconcile.Request {
614- NamespacedName : types.NamespacedName {
615- Namespace : ns .Name ,
616- Name : tc .awsCluster .Name ,
617- },
618- }))
619- } else {
620- g .Expect (result ).To (BeNil ())
621- }
622- })
623- }
624- }
625-
626533func createCluster (g * WithT , awsCluster * infrav1.AWSCluster , namespace string ) {
627534 if awsCluster != nil {
628535 awsCluster .Namespace = namespace
0 commit comments