refactor

Signed-off-by: Valeriy Khorunzhin <valeriy.khorunzhin@flant.com>
deckhouse · Jan 15, 2025 · 6d1ee4b · 6d1ee4b
1 parent 37b7d4e
commit 6d1ee4b
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/images/virtualization-artifact/pkg/controller/vm/internal/block_devices_test.go b/images/virtualization-artifact/pkg/controller/vm/internal/block_devices_test.go
@@ -42,11 +42,9 @@ var _ = Describe("func areVirtualDisksAllowedToUse", func() {
 	var vdFoo *virtv2.VirtualDisk
 	var vdBar *virtv2.VirtualDisk
 
-	blockDevicesConnectedCount := 3
-
 	blockDeviceHandlerMock := &IBlockDeviceServiceMock{}
 	blockDeviceHandlerMock.CountBlockDevicesAttachedToVmFunc = func(_ context.Context, vm *virtv2.VirtualMachine) (int, error) {
-		return blockDevicesConnectedCount, nil
+		return 1, nil
 	}
 
 	BeforeEach(func() {
@@ -160,11 +158,9 @@ var _ = Describe("BlockDeviceHandler", func() {
 	var vdFoo *virtv2.VirtualDisk
 	var vdBar *virtv2.VirtualDisk
 
-	blockDevicesConnectedCount := 3
-
 	blockDeviceHandlerMock := &IBlockDeviceServiceMock{}
 	blockDeviceHandlerMock.CountBlockDevicesAttachedToVmFunc = func(_ context.Context, vm *virtv2.VirtualMachine) (int, error) {
-		return blockDevicesConnectedCount, nil
+		return 1, nil
 	}
 
 	getBlockDevicesState := func(vi *virtv2.VirtualImage, cvi *virtv2.ClusterVirtualImage, vdFoo, vdBar *virtv2.VirtualDisk) BlockDevicesState {