cleanup

Signed-off-by: Anton Sergunov <anton.sergunov@flant.com>
deckhouse · Feb 25, 2025 · b27d51e · b27d51e
1 parent bda11a5
commit b27d51e
Showing 1 changed file with 7 additions and 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
@@ -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")