feat(fistpoint): Add fistpoint interface in lib

And a `exit_on_fistpoint` fixture to enable the behavior of exiting
instead of pausing 30 seconds at normal fistpoint

Signed-off-by: Damien Thenot <[email protected]>

Author: Nambrok

conftest.py

@@ -742,6 +742,15 @@ def nfs_iso_sr(host, nfs_iso_device_config):
     # teardown
     sr.forget()
 
+@pytest.fixture(scope='module')
+def exit_on_fistpoint(host):
+    from lib.fistpoint import FistPoint
+    logging.info(">> Enabling exit on fistpoint")
+    FistPoint.enable_exit_on_fistpoint(host)
+    yield
+    logging.info("<< Disabling exit on fistpoint")
+    FistPoint.disable_exit_on_fistpoint(host)
+
 @pytest.fixture(scope='module')
 def cifs_iso_sr(host, cifs_iso_device_config):
     """ A Samba/CIFS SR. """

lib/fistpoint.py

@@ -0,0 +1,48 @@
+import logging
+from typing import Final
+
+from lib.host import Host
+
+FISTPOINT_DIR: Final = "/tmp"
+LVHDRT_EXIT_FIST: Final = "fist_LVHDRT_exit"
+
+class FistPoint:
+    fistpointName: str
+
+    def __init__(self, host: Host, name: str):
+        self.fistpointName = self._get_name(name)
+        self.host = host
+
+    @staticmethod
+    def enable_exit_on_fistpoint(host: Host):
+        host.create_file(FistPoint._get_path(LVHDRT_EXIT_FIST), "")
+
+    @staticmethod
+    def disable_exit_on_fistpoint(host: Host):
+        host.ssh(["rm", FistPoint._get_path(LVHDRT_EXIT_FIST)])
+
+    @staticmethod
+    def _get_name(name: str) -> str:
+        if name.startswith("fist_"):
+            return name
+        else:
+            return f"fist_{name}"
+
+    @staticmethod
+    def _get_path(name) -> str:
+        return f"{FISTPOINT_DIR}/{name}"
+
+    def enable(self):
+        logging.info(f"Enable fistpoint {self.fistpointName}")
+        self.host.create_file(self._get_path(self.fistpointName), "")
+
+    def disable(self):
+        logging.info(f"Disabling fistpoint {self.fistpointName}")
+        self.host.ssh(["rm", self._get_path(self.fistpointName)])
+
+    def __enter__(self):
+        self.enable()
+        return self
+
+    def __exit__(self, *_):
+        self.disable()