@@ -35,11 +35,10 @@ import (
3535
3636var _ = Describe ("VirtualMachineRestoreForce" , SIGRestoration (), ginkgoutil .CommonE2ETestDecorators (), func () {
3737 const (
38- viCount = 2
39- vmCount = 1
40- vdCount = 3
41- vmbdaCount = 2
42- hasNoConsumerDiskCount = 1
38+ viCount = 2
39+ vmCount = 1
40+ vdCount = 2
41+ vmbdaCount = 2
4342 )
4443
4544 var (
@@ -49,7 +48,7 @@ var _ = Describe("VirtualMachineRestoreForce", SIGRestoration(), ginkgoutil.Comm
4948 volumeSnapshotClass string
5049 namespace string
5150 testCaseLabel = map [string ]string {"testcase" : "vm-restore-force" }
52- hasNoConsumerLabel = map [string ]string {"hasNoConsumer " : "vm-restore-force" }
51+ additionalDiskLabel = map [string ]string {"additionalDisk " : "vm-restore-force" }
5352 )
5453 BeforeEach (func () {
5554 ctx , cancel = context .WithCancel (context .Background ())
@@ -68,7 +67,6 @@ var _ = Describe("VirtualMachineRestoreForce", SIGRestoration(), ginkgoutil.Comm
6867 var err error
6968 namespace , err = kustomize .GetNamespace (kustomization )
7069 Expect (err ).NotTo (HaveOccurred (), "%w" , err )
71- conf .SetNamespace (namespace )
7270
7371 storageClass , err = GetDefaultStorageClass ()
7472 Expect (err ).NotTo (HaveOccurred (), "failed to get the `DefaultStorageClass`" )
@@ -162,9 +160,9 @@ var _ = Describe("VirtualMachineRestoreForce", SIGRestoration(), ginkgoutil.Comm
162160 By ("Attaching `VirtualDisk` after `VirtualMachine` snapshotting" , func () {
163161 for i , vm := range vms .Items {
164162 vdName := fmt .Sprintf ("%s-%d" , "vd-attached-after-vm-snapshotting" , i )
165- newDisk := NewVirtualDisk (vdName , vm .Namespace , hasNoConsumerLabel , resource .NewQuantity (1 * 1024 * 1024 , resource .BinarySI ))
163+ newDisk := NewVirtualDisk (vdName , vm .Namespace , additionalDiskLabel , resource .NewQuantity (1 * 1024 * 1024 , resource .BinarySI ))
166164 CreateResource (ctx , newDisk )
167- newVmbda := NewVirtualMachineBlockDeviceAttachment (vm .Name , vm .Namespace , newDisk .Name , virtv2 .VMBDAObjectRefKindVirtualDisk , testCaseLabel )
165+ newVmbda := NewVirtualMachineBlockDeviceAttachment (vm .Name , vm .Namespace , newDisk .Name , virtv2 .VMBDAObjectRefKindVirtualDisk , additionalDiskLabel )
168166 CreateResource (ctx , newVmbda )
169167
170168 WaitPhaseByLabel (
@@ -255,14 +253,22 @@ var _ = Describe("VirtualMachineRestoreForce", SIGRestoration(), ginkgoutil.Comm
255253 Resource : virtv2 .VirtualMachineRestoreResource ,
256254 Labels : testCaseLabel ,
257255 },
256+ {
257+ Resource : virtv2 .VirtualDiskResource ,
258+ Labels : additionalDiskLabel ,
259+ },
260+ {
261+ Resource : virtv2 .VirtualMachineBlockDeviceAttachmentResource ,
262+ Labels : additionalDiskLabel ,
263+ },
258264 },
259265 }
260266
261267 if config .IsCleanUpNeeded () {
262268 resourcesToDelete .KustomizationDir = conf .TestData .VMRestoreForce
263269 }
264270
265- DeleteTestCaseResources (resourcesToDelete )
271+ DeleteTestCaseResources (namespace , resourcesToDelete )
266272 })
267273 })
268274})
0 commit comments