From c45e310c82b4bfdbf0f2df23001392a40c2cc735 Mon Sep 17 00:00:00 2001
From: Radek Vykydal <rvykydal@redhat.com>
Date: Thu, 3 Oct 2024 16:34:50 +0200
Subject: [PATCH] home reuse: add a test checking rebooted installed system

Resolves: INSTALLER-4054
---
 test/check-storage-home-reuse | 72 +++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)

diff --git a/test/check-storage-home-reuse b/test/check-storage-home-reuse
index 00283bff0..d93324cc4 100755
--- a/test/check-storage-home-reuse
+++ b/test/check-storage-home-reuse
@@ -19,6 +19,7 @@ import os
 
 from anacondalib import VirtInstallMachineCase
 from installer import Installer
+from progress import Progress
 from review import Review
 from storage import Storage
 from storagelib import StorageCase  # pylint: disable=import-error
@@ -148,5 +149,76 @@ class TestHomeReuseFedoraEFI(VirtInstallMachineCase, StorageCase):
         r.check_disk_row(dev, "/home", f"{dev}4", "12.8 GB", False, "btrfs", is_encrypted=False,
                          action="mount")
 
+class TestFedoraPlansHomeReuseReboot(VirtInstallMachineCase, StorageCase):
+    disk_image = "fedora-rawhide"
+    efi = False
+
+    def install(self, needs_confirmation):
+        b = self.browser
+        m = self.machine
+
+        i = Installer(b, m)
+        p = Progress(b)
+
+        i.begin_installation(button_text="Reinstall Fedora", needs_confirmation=needs_confirmation)
+        with b.wait_timeout(300):
+            p.wait_done()
+
+        self.handleReboot()
+
+    def verifyHomeReuse(self, root_file, home_file):
+        # root_file shouldn't exist, home_file should exist
+        m = self.machine
+
+        home_preserved = m.execute(f"if [ -e /home/{home_file} ] ; then echo pass ; fi")
+        assert home_preserved
+        root_formatted = m.execute(f"if [ ! -e /{root_file} ] ; then echo pass ; fi")
+        assert root_formatted
+
+    def verifyHomeMountOpts(self, opts):
+        m = self.machine
+
+        opts_found = m.execute("grep /home /etc/fstab | tr -s ' ' | cut -d ' ' -f 4")
+        assert opts_found == opts
+
+    def testBasic(self):
+        b = self.browser
+        m = self.machine
+        i = Installer(b, m, scenario="home-reuse")
+        s = Storage(b, m)
+
+        pretend_default_scheme(self, "BTRFS")
+
+        disk = "/dev/vda"
+
+        old_root_file = "old_root_file"
+        old_home_file = "old_home_file"
+
+        home_mount_options = m.execute(f"""
+        # Mark existing root by a file
+        mkdir /m
+        mount -o subvol=root,compress=zstd:1 {disk}4 /m
+        grep /home /m/etc/fstab | tr -s ' ' | cut -d ' ' -f 4
+        touch /m/{old_root_file}
+        umount /m
+        # Mark existing home by a file
+        mount -o subvol=home,compress=zstd:1 {disk}4 /m
+        touch /m/{old_home_file}
+        umount /m
+        rmdir /m
+        """)
+        s.udevadm_settle()
+
+        i.open()
+        i.reach(i.steps.INSTALLATION_METHOD)
+        s.rescan_disks()
+
+        s.set_partitioning("home-reuse")
+        i.reach(i.steps.REVIEW)
+
+        self.install(needs_confirmation=True)
+        self.verifyHomeReuse(old_root_file, old_home_file)
+        self.verifyHomeMountOpts(home_mount_options)
+
 if __name__ == '__main__':
     test_main()