#!/usr/bin/python3 -cimport os, sys; os.execv(os.path.dirname(sys.argv[1]) + "/../common/pywrap", sys.argv)## Copyright (C) 2025 Red Hat, Inc.# SPDX-License-Identifier: LGPL-2.1-or-laterfrom sys import stderrimport storagelibimport testlib@testlib.skipImage("Debian and Ubuntu does not boot, drive is not detected.", "debian-*", "ubuntu-*")class TestStorageSmart(storagelib.StorageSmartCase):def udisks_mock_smart_supported(self):m = self.machine# udisks2 version > 2.10.90 is required to mock SMART data on virtual disksif m.image.startswith("fedora") or m.image.startswith("rhel") or m.image.startswith("centos"):version_str = self.machine.execute("rpm -q udisks2 --qf '%{NAME} %{VERSION}\n'").strip()elif m.image.startswith("debian") or m.image.startswith("ubuntu"):version_str = self.machine.execute("dpkg-query -f '${Package} ${Version}\n' --show udisks2").strip()elif m.image == "arch":version_str = self.machine.execute("pacman -Q udisks2").strip()else:return Falseversion = tuple(int(v) for v in version_str.split()[1].split('-')[0].split('.'))return version > (2, 10, 90)def has_new_libblockdev(self):m = self.machineif m.image.startswith("fedora") or m.image.startswith("rhel") or m.image.startswith("centos"):version_str = self.machine.execute("rpm -q libblockdev --qf '%{NAME} %{VERSION}-%{RELEASE}\n'").strip()elif m.image.startswith("debian") or m.image.startswith("ubuntu"):version_str = self.machine.execute("dpkg-query -f '${Package} ${Version}\n' --show libblockdev").strip()elif m.image == "arch":version_str = self.machine.execute("pacman -Q libblockdev").strip()else:return Falsebase_ver, sub_ver = version_str.split()[1].split('-')base_ver = tuple(int(v) for v in base_ver.split('.'))# split on `dot` to remove extra info from rpmsub_ver = int(sub_ver.split('.')[0])if base_ver > (3, 3, 1):return Trueelif base_ver == (3, 3, 1) and sub_ver > 1: # copr @storage/udisks-dailyreturn Trueelif base_ver == (3, 3, 0) and sub_ver >= 3:return Trueelse:return Falsedef set_smart_dump(self, name: str, block: str):self.machine.execute(f"udisksctl smart-simulate -f /tmp/smart-dumps/{name} -b {block}")def testSmart(self):def check_smart_info(assessment: str, hours: str, status: str, bad_sectors: str | None = None,failing_attrs: str | None = None):self.assertIn(assessment, b.text(self.card_desc("Device health (SMART)", "Assessment")))b.wait_in_text(self.card_desc("Device health (SMART)", "Power on hours"), hours)b.wait_in_text(self.card_desc("Device health (SMART)", "Self-test status"), status)if bad_sectors is not None:b.wait_in_text(self.card_desc("Device health (SMART)", "Number of bad sectors"), bad_sectors)b.wait_visible(self.card_desc("Device health (SMART)", "Number of bad sectors") + " .pf-m-warning")if failing_attrs is not None:b.wait_in_text(self.card_desc("Device health (SMART)", "Attributes failing"), failing_attrs)b.wait_visible(self.card_desc("Device health (SMART)", "Attributes failing") + " .pf-m-warning")m = self.machineb = self.browser# udisks2 version > 2.10.1 is required to mock SMART data on virtual disksif not self.udisks_mock_smart_supported():stderr.write("Image has old udisks2 version, cannot run SMART test.\n")returnm.upload(["verify/files/smart-dumps"], "/tmp")# Failing disk, storage page is not loadedself.set_smart_dump("Maxtor_96147H8--BAC51KJ0--2", "/dev/sda")self.login_and_go("/system")b.wait_in_text("#smart-status", "1 disk is failing")self.set_smart_dump("MCCOE64GEMPP--2.9.09", "/dev/sda")b.wait_not_present("#smart-status")# Clicking the link navigates to storage page, failing disk has red iconself.set_smart_dump("Maxtor_96147H8--BAC51KJ0--2", "/dev/sda")b.wait_in_text("#smart-status", "1 disk is failing")b.click("#smart-status a")b.enter_page("/storage")b.wait_visible(self.card("Storage"))b.wait_visible(self.card_row("Storage", name="sda"))b.wait_visible(self.card_row("Storage", name="sda") + " .ct-icon-times-circle")# new disk, no failing sectorsself.set_smart_dump("MCCOE64GEMPP--2.9.09", "/dev/sda")b.wait_visible(self.card("Storage"))self.click_card_row("Storage", name="sda")b.wait_visible(self.card("Device health (SMART)"))check_smart_info("Disk is OK", "1 hours", "Successful")# Disk with running self testself.set_smart_dump("SAMSUNG_MMCQE28G8MUP--0VA_VAM08L1Q", "/dev/sda")check_smart_info("Disk is OK", "2417 hours", "In progress, 30%")# Interrupted self test, disk is OKself.set_smart_dump("INTEL_SSDSA2MH080G1GC--045C8820", "/dev/sda")check_smart_info("Disk is OK", "2309 hours", "Interrupted")# latest libblockdev builds (since 3.3.0-99) from copr have different behavior when assessing disk as failedhas_new_libblockdev = self.has_new_libblockdev()# Aborted self test and has known bad sector (overall assessment is still OK)self.set_smart_dump("ST9160821AS--3.CLH", "/dev/sda")if has_new_libblockdev:check_smart_info("Disk is OK", "556 hours", "Aborted", bad_sectors="1")else:check_smart_info("Disk is failing", "556 hours", "Aborted", bad_sectors="1")# Multiple bad sectorsself.set_smart_dump("Maxtor_96147H8--BAC51KJ0", "/dev/sda")if has_new_libblockdev:check_smart_info("Disk is OK", "2016 hours", "Successful", bad_sectors="71")else:check_smart_info("Disk is failing", "2016 hours", "Successful", bad_sectors="71")# Multiple bad sectors with failing attributeself.set_smart_dump("Maxtor_96147H8--BAC51KJ0--2", "/dev/sda")check_smart_info("Disk is failing", "2262 hours", "Successful", bad_sectors="71", failing_attrs="1")b.wait_visible(self.card_desc("Device health (SMART)", "Assessment") + " .pf-m-danger")# Check that SMART card is not visible on DVD driveb.go("/storage")self.click_card_row("Storage", name="sr0")b.wait_visible(self.card("Media drive"))b.wait_not_present(self.card("Device health (SMART)"))if __name__ == '__main__':testlib.test_main()