From dbb6877e8ae6e400a94092eed15b0c3b1578c3a7 Mon Sep 17 00:00:00 2001
From: Tu Dinh <ngoc-tu.dinh@vates.tech>
Date: Thu, 28 Nov 2024 13:17:43 +0100
Subject: [PATCH] guest-tools/win: Add XenClean tests

Signed-off-by: Tu Dinh <ngoc-tu.dinh@vates.tech>
---
 tests/guest-tools/win/test_xenclean.py | 46 ++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 tests/guest-tools/win/test_xenclean.py

diff --git a/tests/guest-tools/win/test_xenclean.py b/tests/guest-tools/win/test_xenclean.py
new file mode 100644
index 00000000..ec2750b0
--- /dev/null
+++ b/tests/guest-tools/win/test_xenclean.py
@@ -0,0 +1,46 @@
+import logging
+from pathlib import PureWindowsPath
+from typing import Any
+import pytest
+from lib.common import wait_for
+from lib.vm import VM
+
+from . import wait_for_vm_running_and_ssh_up_without_tools
+
+
+def run_xenclean(vm: VM, guest_tools_iso: dict[str, Any]):
+    vm.insert_cd(guest_tools_iso["name"])
+    wait_for(lambda: vm.file_exists("D:/", regular_file=False))
+
+    logging.info("Run XenClean")
+    xenclean_path = PureWindowsPath("D:\\") / guest_tools_iso["xenclean_path"]
+    xenclean_cmd = f"Set-Location C:\\; {xenclean_path} -NoReboot -Confirm:$false; Stop-Computer -Force"
+    vm.start_background_powershell(xenclean_cmd)
+
+    wait_for(vm.is_halted, "Wait for VM halted")
+    vm.eject_cd()
+
+    vm.start()
+    wait_for_vm_running_and_ssh_up_without_tools(vm)
+
+
+@pytest.mark.multi_vms
+@pytest.mark.usefixtures("windows_vm")
+class TestXenClean:
+    def test_xenclean_without_tools(self, running_unsealed_windows_vm, guest_tools_iso):
+        vm = running_unsealed_windows_vm
+        logging.info("XenClean with empty VM")
+        run_xenclean(vm, guest_tools_iso)
+        assert vm.are_windows_tools_uninstalled()
+
+    def test_xenclean_with_test_tools(self, vm_install_test_tools, guest_tools_iso):
+        vm = vm_install_test_tools
+        logging.info("XenClean with test tools")
+        run_xenclean(vm, guest_tools_iso)
+        assert vm.are_windows_tools_uninstalled()
+
+    def test_xenclean_with_other_tools(self, vm_install_other_drivers, guest_tools_iso):
+        vm, _ = vm_install_other_drivers
+        logging.info(f"XenClean with other tools")
+        run_xenclean(vm, guest_tools_iso)
+        assert vm.are_windows_tools_uninstalled()