@@ -28,19 +28,16 @@ import (
28
28
. "github.com/onsi/gomega"
29
29
corev1 "k8s.io/api/core/v1"
30
30
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
31
- "k8s.io/apimachinery/pkg/types"
32
31
"k8s.io/client-go/tools/record"
33
32
"k8s.io/utils/ptr"
34
33
ctrl "sigs.k8s.io/controller-runtime"
35
34
"sigs.k8s.io/controller-runtime/pkg/client"
36
35
"sigs.k8s.io/controller-runtime/pkg/client/fake"
37
- "sigs.k8s.io/controller-runtime/pkg/reconcile"
38
36
39
37
infrav1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
40
38
"sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/scope"
41
39
"sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/services"
42
40
"sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/services/mock_services"
43
- "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/logger"
44
41
clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
45
42
"sigs.k8s.io/cluster-api/util"
46
43
)
@@ -60,7 +57,8 @@ func TestAWSClusterReconcilerReconcile(t *testing.T) {
60
57
Kind : "Cluster" ,
61
58
Name : "capi-fail-test" ,
62
59
UID : "1" ,
63
- }}}},
60
+ },
61
+ }}},
64
62
expectError : true ,
65
63
},
66
64
{
@@ -532,97 +530,6 @@ func TestAWSClusterReconcileOperations(t *testing.T) {
532
530
})
533
531
}
534
532
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
-
626
533
func createCluster (g * WithT , awsCluster * infrav1.AWSCluster , namespace string ) {
627
534
if awsCluster != nil {
628
535
awsCluster .Namespace = namespace
0 commit comments