Hibernation: Move before case to common file

Author: adityagesh

lisa/microsoft/testsuites/power/common.py

@@ -8,7 +8,7 @@
 from lisa import Environment, Logger, Node, RemoteNode, features
 from lisa.features import StartStop
 from lisa.features.startstop import VMStatus
-from lisa.operating_system import SLES, AlmaLinux, Debian, Redhat, Ubuntu
+from lisa.operating_system import BSD, SLES, AlmaLinux, Debian, Redhat, Ubuntu, Windows
 from lisa.sut_orchestrator.azure.features import AzureExtension
 from lisa.tools import (
     Cat,
@@ -112,6 +112,20 @@ def _prepare_hibernation_environment(node: Node) -> None:
         resize.expand_os_partition()
 
 
+def hibernation_before_case(node: Node, log: Logger) -> None:
+    """
+    Common before_case logic for hibernation tests.
+    Validates OS support and prepares the environment.
+    """
+    if isinstance(node.os, BSD) or isinstance(node.os, Windows):
+        raise SkippedException(f"{node.os} is not supported.")
+
+    # Expand OS partition first (needed for RHEL/LVM before checking disk space)
+    _prepare_hibernation_environment(node)
+
+    check_hibernation_disk_requirements(node)
+
+
 def _perform_hibernation_cycle(
     node: Node, log: Logger, throw_error: bool = True
 ) -> tuple[Any, Any]:

lisa/microsoft/testsuites/power/power.py

@@ -7,9 +7,8 @@
 from assertpy import assert_that
 from func_timeout import func_timeout
 from microsoft.testsuites.power.common import (
-    _prepare_hibernation_environment,
-    check_hibernation_disk_requirements,
     cleanup_env,
+    hibernation_before_case,
     is_distro_supported,
     run_network_workload,
     run_storage_workload,
@@ -28,12 +27,10 @@
 from lisa.features import Disk, HibernationEnabled, Sriov, Synthetic
 from lisa.features.availability import AvailabilityTypeNoRedundancy
 from lisa.node import Node
-from lisa.operating_system import BSD, Windows
 from lisa.search_space import IntRange
 from lisa.sut_orchestrator.azure.features import AzureExtension
 from lisa.testsuite import simple_requirement
 from lisa.tools import Date, Hwclock, StressNg
-from lisa.util import SkippedException
 from lisa.util.perf_timer import create_timer
 
 
@@ -47,13 +44,7 @@
 class Power(TestSuite):
     def before_case(self, log: Logger, **kwargs: Any) -> None:
         node: Node = kwargs["node"]
-        if isinstance(node.os, BSD) or isinstance(node.os, Windows):
-            raise SkippedException(f"{node.os} is not supported.")
-
-        # Expand OS partition first (needed for RHEL/LVM before checking disk space)
-        _prepare_hibernation_environment(node)
-
-        check_hibernation_disk_requirements(node)
+        hibernation_before_case(node, log)
 
     @TestCaseMetadata(
         description="""

lisa/microsoft/testsuites/power/stress.py

@@ -5,6 +5,7 @@
 
 from microsoft.testsuites.power.common import (
     cleanup_env,
+    hibernation_before_case,
     is_distro_supported,
     verify_hibernation,
 )
@@ -20,9 +21,7 @@
 from lisa.features import HibernationEnabled, Sriov
 from lisa.features.availability import AvailabilityTypeNoRedundancy
 from lisa.node import Node
-from lisa.operating_system import BSD, Windows
 from lisa.testsuite import simple_requirement
-from lisa.util import SkippedException
 
 
 @TestSuiteMetadata(
@@ -37,8 +36,7 @@ class PowerStress(TestSuite):
 
     def before_case(self, log: Logger, **kwargs: Any) -> None:
         node: Node = kwargs["node"]
-        if isinstance(node.os, BSD) or isinstance(node.os, Windows):
-            raise SkippedException(f"{node.os} is not supported.")
+        hibernation_before_case(node, log)
 
     @TestCaseMetadata(
         description="""