Skip to content

Commit 714e9c9

Browse files
committed
Refactor tests to use HaveValue.
1 parent b2bed5b commit 714e9c9

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

pkg/controller/controller_test.go

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -488,8 +488,7 @@ var _ = Describe("controller.Controller", func() {
488488

489489
internalCtrlWithWarmup, ok := ctrlWithWarmup.(*internalcontroller.Controller[reconcile.Request])
490490
Expect(ok).To(BeTrue())
491-
Expect(internalCtrlWithWarmup.NeedWarmup).NotTo(BeNil())
492-
Expect(*internalCtrlWithWarmup.NeedWarmup).To(BeTrue())
491+
Expect(internalCtrlWithWarmup.NeedWarmup).To(HaveValue(BeTrue()))
493492

494493
// Test with NeedWarmup set to false
495494
ctrlWithoutWarmup, err := controller.New("warmup-disabled-ctrl", m, controller.Options{
@@ -500,8 +499,7 @@ var _ = Describe("controller.Controller", func() {
500499

501500
internalCtrlWithoutWarmup, ok := ctrlWithoutWarmup.(*internalcontroller.Controller[reconcile.Request])
502501
Expect(ok).To(BeTrue())
503-
Expect(internalCtrlWithoutWarmup.NeedWarmup).NotTo(BeNil())
504-
Expect(*internalCtrlWithoutWarmup.NeedWarmup).To(BeFalse())
502+
Expect(internalCtrlWithoutWarmup.NeedWarmup).To(HaveValue(BeFalse()))
505503

506504
// Test with NeedWarmup not set (should default to nil)
507505
ctrlWithDefaultWarmup, err := controller.New("warmup-default-ctrl", m, controller.Options{
@@ -529,8 +527,7 @@ var _ = Describe("controller.Controller", func() {
529527

530528
internalCtrlInheritingWarmup, ok := ctrlInheritingWarmup.(*internalcontroller.Controller[reconcile.Request])
531529
Expect(ok).To(BeTrue())
532-
Expect(internalCtrlInheritingWarmup.NeedWarmup).NotTo(BeNil())
533-
Expect(*internalCtrlInheritingWarmup.NeedWarmup).To(BeTrue())
530+
Expect(internalCtrlInheritingWarmup.NeedWarmup).To(BeTrue())
534531

535532
// Test that explicit controller setting overrides manager setting
536533
ctrlOverridingWarmup, err := controller.New("override-warmup-disabled", managerWithWarmup, controller.Options{
@@ -541,8 +538,7 @@ var _ = Describe("controller.Controller", func() {
541538

542539
internalCtrlOverridingWarmup, ok := ctrlOverridingWarmup.(*internalcontroller.Controller[reconcile.Request])
543540
Expect(ok).To(BeTrue())
544-
Expect(internalCtrlOverridingWarmup.NeedWarmup).NotTo(BeNil())
545-
Expect(*internalCtrlOverridingWarmup.NeedWarmup).To(BeFalse())
541+
Expect(internalCtrlOverridingWarmup.NeedWarmup).To(BeFalse())
546542
})
547543
})
548544
})

0 commit comments

Comments
 (0)