From d5b336d344e4f4f4542c6c44930d24c018764d68 Mon Sep 17 00:00:00 2001
From: Katerina Koukiou <kkoukiou@redhat.com>
Date: Thu, 15 Aug 2024 14:22:57 +0200
Subject: [PATCH] tests: add some pixel tests for reclaim dialog

---
 test/check-storage-reclaim | 45 ++++++++++++++++++++++++++++++++++++--
 test/reference             |  2 +-
 2 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/test/check-storage-reclaim b/test/check-storage-reclaim
index cec421bc81..356769d539 100755
--- a/test/check-storage-reclaim
+++ b/test/check-storage-reclaim
@@ -17,6 +17,7 @@
 
 from anacondalib import VirtInstallMachineCase
 from installer import Installer
+from operating_systems import WindowsOS
 from review import Review
 from storage import Storage
 from storagelib import StorageCase
@@ -27,7 +28,7 @@ from testlib import (
 
 
 @nondestructive
-class TestStorageUseFreeSpaceScenario(VirtInstallMachineCase, StorageCase):
+class TestReclaim(VirtInstallMachineCase, StorageCase):
     def setup_partitions(self, s, i):
         disk = "/dev/vda"
         btrfsname = "btrfstest"
@@ -236,13 +237,19 @@ class TestReclaimLUKS(VirtInstallMachineCase, StorageCase):
         s.set_partitioning("use-free-space")
         s.reclaim_set_checkbox(True)
         i.next(True)
+
+        b.assert_pixels(
+            "#reclaim-space-modal",
+            "reclaim-space-modal-encrypted",
+        )
+
         s.reclaim_check_device_row("vda1", "", "luks", "4.29 GB", locked=True)
         s.reclaim_check_action_button_present("vda1", "shrink", True, True)
         s.reclaim_check_action_button_present("vda1", "delete", True)
 
 
 @nondestructive
-class TestStorageUseFreeSpaceScenarioExistingSystemFedora(VirtInstallMachineCase, StorageCase):
+class TestReclaimExistingSystemFedora(VirtInstallMachineCase, StorageCase):
     disk_image = "fedora-rawhide"
 
     def testDeletePartition(self):
@@ -257,6 +264,13 @@ class TestStorageUseFreeSpaceScenarioExistingSystemFedora(VirtInstallMachineCase
         s.set_partitioning("use-free-space")
         s.reclaim_set_checkbox(True)
         i.next(True)
+
+        b.assert_pixels(
+            "#reclaim-space-modal",
+            "reclaim-space-modal-fedora",
+            ignore=["td[data-label=Space]"],
+        )
+
         s.reclaim_remove_device("vda4")
         s.reclaim_modal_submit()
 
@@ -265,5 +279,32 @@ class TestStorageUseFreeSpaceScenarioExistingSystemFedora(VirtInstallMachineCase
         r.check_some_erased_checkbox_label()
 
 
+@nondestructive
+class TestReclaimExistingSystemWindows(VirtInstallMachineCase):
+    disk_size = 20
+
+    def setUp(self):
+        super().setUp()
+        WindowsOS(machine=self.machine, browser=self.browser).partition_disk()
+
+    def testBasic(self):
+        b = self.browser
+        m = self.machine
+        i = Installer(b, m)
+        s = Storage(b, m)
+
+        i.open()
+        i.reach(i.steps.INSTALLATION_METHOD)
+        s.set_partitioning("use-free-space")
+        s.reclaim_set_checkbox(True)
+        i.next(True)
+
+        b.assert_pixels(
+            "#reclaim-space-modal",
+            "reclaim-space-modal-windows",
+            ignore=["td[data-label=Space]"],
+        )
+
+
 if __name__ == '__main__':
     test_main()
diff --git a/test/reference b/test/reference
index 11dffb5a3d..d48b1620b8 160000
--- a/test/reference
+++ b/test/reference
@@ -1 +1 @@
-Subproject commit 11dffb5a3d4faba8e26ee07ea9531f87abbcb19f
+Subproject commit d48b1620b869f392bad70e4839cc8ebc4fe1d7f6