From b27d51e38b35747e4ef3a5623f708f2a9d241c40 Mon Sep 17 00:00:00 2001 From: Anton Sergunov Date: Mon, 24 Feb 2025 15:45:40 +0600 Subject: [PATCH] cleanup Signed-off-by: Anton Sergunov --- .../src/internal/utils/volume_cleanup_ee_test.go | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/images/agent/src/internal/utils/volume_cleanup_ee_test.go b/images/agent/src/internal/utils/volume_cleanup_ee_test.go index eee8b86d..f1e40a69 100644 --- a/images/agent/src/internal/utils/volume_cleanup_ee_test.go +++ b/images/agent/src/internal/utils/volume_cleanup_ee_test.go @@ -49,14 +49,7 @@ var _ = Describe("Cleaning up volume", func() { Expect(err).To(MatchError("volume cleanup is not supported in your edition")) } } - ItSucceedOnlyWhenFeatureEnabled := func(text string) bool { - return It(text, func() { - doCall() - if feature.VolumeCleanupEnabled() { - Expect(err).ToNot(HaveOccurred()) - } - }) - } + When("method is unknown", func() { BeforeEach(func() { method = "some" @@ -111,7 +104,12 @@ var _ = Describe("Cleaning up volume", func() { device.EXPECT().Close().Return(nil) } }) - ItSucceedOnlyWhenFeatureEnabled("calls device discard") + It("calls device discard", func() { + doCall() + if feature.VolumeCleanupEnabled() { + Expect(err).ToNot(HaveOccurred()) + } + }) }) When("cannot close", func() { closingError := errors.New("closing error")