vioser_in_use.py: change logic of 'memory leak check' location

change logic of 'memory leak check' location for some serial case

Signed-off-by: demeng <[email protected]>

qemu/tests/vioser_in_use.py

@@ -64,8 +64,6 @@ def vcpu_hotplug_guest(test, params, vm, session):
         vm.hotplug_vcpu_device(vcpu_id=vcpu_device)
         # make the cpu hotplug has slot during data transfer
         time.sleep(2)
-    if params.get("os_type") == "windows":
-        win_driver_utils.memory_leak_check(vm, test, params)
 
 
 @error_context.context_aware
@@ -148,6 +146,11 @@ def run(test, params, env):
         error_context.context("Check if guest is alive.", test.log.info)
         vm.verify_kernel_crash()
         session = vm.wait_for_login(timeout=timeout)
+        if params.get("memory_leak_check", "no") == "yes":
+            # for windows guest, disable/uninstall driver to get memory leak based on
+            # driver verifier is enabled
+            if params.get("os_type") == "windows":
+                win_driver_utils.memory_leak_check(vm, test, params)
         session.close()
         return
 

qemu/tests/virtio_serial_file_transfer_max_ports.py

@@ -45,8 +45,4 @@ def run(test, params, env):
         transfer_data(params, vm, sender='both')
 
     vm.verify_alive()
-    # for windows guest, disable/uninstall driver to get memory leak based on
-    # driver verifier is enabled
-    if params.get("os_type") == "windows":
-        win_driver_utils.memory_leak_check(vm, test, params)
     vm.destroy()

qemu/tests/virtio_serial_hotplug_port_pci.py

@@ -150,6 +150,4 @@ def run_serial_data_transfer():
         # for windows guest, disable/uninstall driver to get memory leak based on
         # driver verifier is enabled
         if params.get("os_type") == "windows":
-            vm.devices.simple_hotplug(buses[0], vm.monitor)
-            vm.devices.simple_hotplug(serial_devices[0], vm.monitor)
             win_driver_utils.memory_leak_check(vm, test, params)