tests/limit: Restore the VM to its original state

Signed-off-by: Andrii Sultanov <[email protected]>

Author: last-genius

lib/vif.py

@@ -30,3 +30,6 @@ def device_id(self):
 
     def move(self, network_uuid):
         self.vm.host.xe('vif-move', {'uuid': self.uuid, 'network-uuid': network_uuid})
+
+    def destroy(self):
+        self.vm.host.xe('vif-destroy', {'uuid': self.uuid})

tests/limits/test_vif_limit.py

@@ -32,12 +32,16 @@ def test_vif_limit(self, host_with_saved_yum_state, imported_vm):
         existing_vifs = len(vm.vifs())
 
         logging.info(f'Get {vcpus} vCPUs for the VM')
+        original_vcpus_max = vm.param_get('VCPUs-max')
+        original_vcpus_at_startup = vm.param_get('VCPUs-at-startup')
         vm.param_set('VCPUs-max', vcpus)
         vm.param_set('VCPUs-at-startup', vcpus)
 
         logging.info('Create VIFs before starting the VM')
+        vifs = []
         for i in range(existing_vifs, vif_limit):
-            vm.create_vif(i, network_uuid=network_uuid)
+            vif = vm.create_vif(i, network_uuid=network_uuid)
+            vifs.append(vif)
 
         vm.start()
         vm.wait_for_os_booted()
@@ -92,4 +96,9 @@ def test_vif_limit(self, host_with_saved_yum_state, imported_vm):
             logging.info(stdout)
         finally:
             vm.ssh(['pkill iperf3 || true'])
+            vm.shutdown(verify=True)
+            vm.param_set('VCPUs-at-startup', original_vcpus_at_startup)
+            vm.param_set('VCPUs-max', original_vcpus_max)
+            for vif in vifs:
+                vif.destroy()
             host.ssh('killall iperf3')