@@ -17,6 +17,7 @@ import (
1717
1818 "k8s.io/utils/ptr"
1919
20+ "k8s.io/apimachinery/pkg/types"
2021 "k8s.io/apimachinery/pkg/util/intstr"
2122
2223 . "github.com/onsi/ginkgo/v2"
@@ -30,7 +31,6 @@ import (
3031 apierrors "k8s.io/apimachinery/pkg/api/errors"
3132 k8sresource "k8s.io/apimachinery/pkg/api/resource"
3233 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
33- "k8s.io/apimachinery/pkg/types"
3434 runtimeClient "sigs.k8s.io/controller-runtime/pkg/client"
3535)
3636
@@ -60,18 +60,10 @@ var _ = Describe("RabbitmqClusterController", func() {
6060 waitForClusterCreation (ctx , cluster , client )
6161 })
6262
63- AfterEach (func () {
64- Expect (client .Delete (ctx , cluster )).To (Succeed ())
65- Eventually (func () bool {
66- err := client .Get (ctx , types.NamespacedName {Name : cluster .Name , Namespace : cluster .Namespace }, cluster )
67- return apierrors .IsNotFound (err )
68- }, 5 ).Should (BeTrue (), "expected to delete cluster '%s' but it still exists" , cluster .Name )
69- })
70-
7163 It ("works" , func () {
7264 By ("populating the image spec with the default image" , func () {
7365 fetchedCluster := & rabbitmqv1beta1.RabbitmqCluster {}
74- Expect (client .Get (ctx , types. NamespacedName { Name : "rabbitmq-one" , Namespace : defaultNamespace } , fetchedCluster )).To (Succeed ())
66+ Expect (client .Get (ctx , runtimeClient . ObjectKeyFromObject ( cluster ) , fetchedCluster )).To (Succeed ())
7567 Expect (fetchedCluster .Spec .Image ).To (Equal (defaultRabbitmqImage ))
7668 })
7769
@@ -199,10 +191,6 @@ var _ = Describe("RabbitmqClusterController", func() {
199191 waitForClusterCreation (ctx , cluster , client )
200192 })
201193
202- AfterEach (func () {
203- Expect (client .Delete (ctx , cluster )).To (Succeed ())
204- })
205-
206194 It ("adds annotations to child resources" , func () {
207195 headlessSvc := service (ctx , cluster , "nodes" )
208196 Expect (headlessSvc .Annotations ).Should (HaveKeyWithValue ("my-annotation" , "this-annotation" ))
@@ -234,10 +222,6 @@ var _ = Describe("RabbitmqClusterController", func() {
234222 waitForClusterCreation (ctx , cluster , client )
235223 })
236224
237- AfterEach (func () {
238- Expect (client .Delete (ctx , cluster )).To (Succeed ())
239- })
240-
241225 It ("applies the Vault configuration" , func () {
242226 By ("not exposing DefaultUser or its Binding as status" )
243227 Expect (cluster ).NotTo (BeNil ())
@@ -246,7 +230,7 @@ var _ = Describe("RabbitmqClusterController", func() {
246230 Expect (cluster .Status .Binding ).To (BeNil ())
247231 By ("setting the default user updater image to the controller default" )
248232 fetchedCluster := & rabbitmqv1beta1.RabbitmqCluster {}
249- Expect (client .Get (ctx , types. NamespacedName { Name : "rabbitmq-vault" , Namespace : defaultNamespace } , fetchedCluster )).To (Succeed ())
233+ Expect (client .Get (ctx , runtimeClient . ObjectKeyFromObject ( cluster ) , fetchedCluster )).To (Succeed ())
250234 Expect (fetchedCluster .Spec .SecretBackend .Vault .DefaultUserUpdaterImage ).To (PointTo (Equal (defaultUserUpdaterImage )))
251235 })
252236 })
@@ -267,10 +251,6 @@ var _ = Describe("RabbitmqClusterController", func() {
267251 waitForClusterCreation (ctx , cluster , client )
268252 })
269253
270- AfterEach (func () {
271- Expect (client .Delete (ctx , cluster )).To (Succeed ())
272- })
273-
274254 It ("configures the imagePullSecret on sts correctly" , func () {
275255 By ("using the instance spec secret" , func () {
276256 sts := statefulSet (ctx , cluster )
@@ -307,10 +287,6 @@ var _ = Describe("RabbitmqClusterController", func() {
307287 Expect (client .Create (ctx , cluster )).To (Succeed ())
308288 })
309289
310- AfterEach (func () {
311- Expect (client .Delete (ctx , cluster )).To (Succeed ())
312- })
313-
314290 It ("adds the affinity rules to pod spec" , func () {
315291 sts := statefulSet (ctx , cluster )
316292 podSpecAffinity := sts .Spec .Template .Spec .Affinity
@@ -319,11 +295,6 @@ var _ = Describe("RabbitmqClusterController", func() {
319295 })
320296
321297 Context ("Service configurations" , func () {
322- AfterEach (func () {
323- Expect (client .Delete (ctx , cluster )).To (Succeed ())
324- Expect (clientSet .CoreV1 ().Services (cluster .Namespace ).Delete (ctx , cluster .ChildResourceName ("" ), metav1.DeleteOptions {}))
325- })
326-
327298 It ("creates the service type and annotations as configured in instance spec" , func () {
328299 cluster = & rabbitmqv1beta1.RabbitmqCluster {
329300 ObjectMeta : metav1.ObjectMeta {
@@ -360,10 +331,6 @@ var _ = Describe("RabbitmqClusterController", func() {
360331 })
361332
362333 Context ("Resource requirements configurations" , func () {
363- AfterEach (func () {
364- Expect (client .Delete (ctx , cluster )).To (Succeed ())
365- })
366-
367334 It ("uses resource requirements from instance spec when provided" , func () {
368335 cluster = & rabbitmqv1beta1.RabbitmqCluster {
369336 ObjectMeta : metav1.ObjectMeta {
@@ -400,10 +367,6 @@ var _ = Describe("RabbitmqClusterController", func() {
400367 })
401368
402369 Context ("Persistence configurations" , func () {
403- AfterEach (func () {
404- Expect (client .Delete (ctx , cluster )).To (Succeed ())
405- })
406-
407370 It ("creates the RabbitmqCluster with the specified storage from instance spec" , func () {
408371 cluster = & rabbitmqv1beta1.RabbitmqCluster {
409372 ObjectMeta : metav1.ObjectMeta {
@@ -426,11 +389,11 @@ var _ = Describe("RabbitmqClusterController", func() {
426389 })
427390 })
428391
429- Context ("Custom Resource updates" , func () {
392+ Context ("Custom Resource updates" , FlakeAttempts ( 3 ), func () {
430393 BeforeEach (func () {
431394 cluster = & rabbitmqv1beta1.RabbitmqCluster {
432395 ObjectMeta : metav1.ObjectMeta {
433- Name : "rabbitmq- cr-update" ,
396+ Name : fmt . Sprintf ( " cr-update-%d-%d" , GinkgoParallelProcess (), time . Now (). Unix ()) ,
434397 Namespace : defaultNamespace ,
435398 },
436399 }
@@ -439,11 +402,6 @@ var _ = Describe("RabbitmqClusterController", func() {
439402 waitForClusterCreation (ctx , cluster , client )
440403 })
441404
442- AfterEach (func () {
443- Expect (client .Delete (ctx , cluster )).To (Succeed ())
444- waitForClusterDeletion (ctx , cluster , client )
445- })
446-
447405 It ("the service annotations are updated" , func () {
448406 Expect (updateWithRetry (cluster , func (r * rabbitmqv1beta1.RabbitmqCluster ) {
449407 r .Spec .Service .Annotations = map [string ]string {"test-key" : "test-value" }
@@ -867,11 +825,6 @@ var _ = Describe("RabbitmqClusterController", func() {
867825 waitForClusterCreation (ctx , cluster , client )
868826 })
869827
870- AfterEach (func () {
871- Expect (client .Delete (ctx , cluster )).To (Succeed ())
872- waitForClusterDeletion (ctx , cluster , client )
873- })
874-
875828 It ("creates a StatefulSet with the override applied" , func () {
876829 sts := statefulSet (ctx , cluster )
877830 myStorage := k8sresource .MustParse ("100Gi" )
@@ -1081,11 +1034,13 @@ var _ = Describe("RabbitmqClusterController", func() {
10811034 })
10821035
10831036 Context ("Service Override" , func () {
1084-
1037+ var clusterName string
10851038 BeforeEach (func () {
1039+ suffix = fmt .Sprintf ("-%d" , time .Now ().UnixNano ())
1040+ clusterName = "svc-override" + suffix
10861041 cluster = & rabbitmqv1beta1.RabbitmqCluster {
10871042 ObjectMeta : metav1.ObjectMeta {
1088- Name : "svc-override" ,
1043+ Name : clusterName ,
10891044 Namespace : defaultNamespace ,
10901045 },
10911046 Spec : rabbitmqv1beta1.RabbitmqClusterSpec {
@@ -1118,11 +1073,6 @@ var _ = Describe("RabbitmqClusterController", func() {
11181073 waitForClusterCreation (ctx , cluster , client )
11191074 })
11201075
1121- AfterEach (func () {
1122- Expect (client .Delete (ctx , cluster )).To (Succeed ())
1123- waitForClusterDeletion (ctx , cluster , client )
1124- })
1125-
11261076 It ("creates a Service with the override applied" , func () {
11271077 amqpTargetPort := intstr.IntOrString {IntVal : int32 (5672 )}
11281078 managementTargetPort := intstr.IntOrString {IntVal : int32 (15672 )}
@@ -1159,7 +1109,7 @@ var _ = Describe("RabbitmqClusterController", func() {
11591109 TargetPort : additionalTargetPort ,
11601110 },
11611111 ))
1162- Expect (svc .Spec .Selector ).To (Equal (map [string ]string {"a-selector" : "a-label" , "app.kubernetes.io/name" : "svc-override" }))
1112+ Expect (svc .Spec .Selector ).To (Equal (map [string ]string {"a-selector" : "a-label" , "app.kubernetes.io/name" : clusterName }))
11631113 Expect (svc .Spec .SessionAffinity ).To (Equal (corev1 .ServiceAffinityClientIP ))
11641114 Expect (svc .Spec .PublishNotReadyAddresses ).To (BeFalse ())
11651115 })
@@ -1188,10 +1138,6 @@ var _ = Describe("RabbitmqClusterController", func() {
11881138 waitForClusterCreation (ctx , cluster , client )
11891139 })
11901140
1191- AfterEach (func () {
1192- Expect (client .Delete (ctx , cluster )).To (Succeed ())
1193- })
1194-
11951141 It ("works" , func () {
11961142 By ("skipping reconciling if label is set to true" , func () {
11971143 Expect (updateWithRetry (cluster , func (r * rabbitmqv1beta1.RabbitmqCluster ) {
0 commit comments