to Enum

to Enum

lisa/tools/__init__.py

@@ -36,7 +36,7 @@
 from .fallocate import Fallocate
 from .fdisk import Fdisk
 from .find import Find
-from .fio import FIOMODES, Fio, FIOResult
+from .fio import FIOMODES, Fio, FIOResult, IoEngine
 from .firewall import Firewall, Iptables
 from .free import Free
 from .gcc import Gcc
@@ -159,6 +159,7 @@
     "Ip",
     "IpInfo",
     "Iperf3",
+    "IoEngine",
     "HibernationSetup",
     "Hostname",
     "Hugepages",

lisa/tools/fio.py

@@ -37,6 +37,16 @@ class FIOResult:
 )
 
 
+class IoEngine(Enum):
+    IO_URING = "io_uring"
+    LIBAIO = "libaio"
+    POSIXAIO = "posixaio"
+    NONE = ""
+
+    def __str__(self) -> str:
+        return self.value
+
+
 class Fio(Tool):
     fio_repo = "https://github.com/axboe/fio/"
     branch = "fio-3.29"
@@ -93,7 +103,7 @@ def launch(
         verify_dump: bool = False,
         verify_fatal: bool = False,
         verify: str = "",
-        ioengine: str = "",
+        ioengine: IoEngine = IoEngine.NONE,
         cwd: Optional[pathlib.PurePath] = None,
     ) -> FIOResult:
         cmd = self._get_command(
@@ -151,7 +161,7 @@ def launch_async(
         verify_dump: bool = False,
         verify_fatal: bool = False,
         verify: str = "",
-        ioengine: str = "",
+        ioengine: IoEngine = IoEngine.NONE,
         cwd: Optional[pathlib.PurePath] = None,
     ) -> Process:
         cmd = self._get_command(
@@ -268,13 +278,15 @@ def _get_command(  # noqa: C901
         verify_dump: bool = False,
         verify_fatal: bool = False,
         verify: str = "",
-        ioengine: str = "",
+        ioengine: IoEngine = IoEngine.NONE,
     ) -> str:
-        if ioengine == "":
-            ioengine = "posixaio" if isinstance(self.node.os, BSD) else "libaio"
+        if isinstance(self.node.os, BSD):
+            ioengine = IoEngine.POSIXAIO
+        elif ioengine == IoEngine.NONE:
+            ioengine = IoEngine.LIBAIO
 
         cmd = (
-            f"--ioengine={ioengine} --filename={filename} "
+            f"--ioengine={ioengine.value} --filename={filename} "
             f"--readwrite={mode} --iodepth={iodepth} "
             f"--name={name}"
         )

microsoft/testsuites/performance/common.py

@@ -44,6 +44,7 @@
     Ssh,
     Sysctl,
 )
+from lisa.tools.fio import IoEngine
 from lisa.tools.ntttcp import (
     NTTTCP_TCP_CONCURRENCY,
     NTTTCP_TCP_CONCURRENCY_BSD,
@@ -70,7 +71,7 @@ def perf_disk(
     size_mb: int = 0,
     numjob: int = 0,
     overwrite: bool = False,
-    ioengine: str = "",
+    ioengine: IoEngine = IoEngine.NONE,
     cwd: Optional[pathlib.PurePath] = None,
 ) -> None:
     fio_result_list: List[FIOResult] = []
@@ -87,7 +88,7 @@ def perf_disk(
     # read: https://www.kernel.org/doc/Documentation/sysctl/fs.txt
     # So we set numjob to 256 if numjob is larger than 256.
     # This limitation is only needed for 'libaio' ioengine but not for 'io_uring'.
-    if ioengine == "libaio":
+    if ioengine == IoEngine.LIBAIO:
         numjob = min(numjob, 256)
     for mode in FIOMODES:
         iodepth = start_iodepth

microsoft/testsuites/performance/nvmeperf.py

@@ -13,6 +13,7 @@
 from lisa.messages import DiskSetupType, DiskType
 from lisa.testsuite import TestResult
 from lisa.tools import Echo, Lscpu
+from lisa.tools.fio import IoEngine
 from microsoft.testsuites.performance.common import perf_disk
 
 
@@ -81,7 +82,7 @@ def perf_nvme(self, node: Node, result: TestResult) -> None:
         priority=3,
         timeout=TIME_OUT,
         requirement=simple_requirement(
-            supported_features=[NvmeSettings(disk_count=8)],
+            supported_features=[Nvme],
         ),
     )
     def perf_nvme_io_uring(self, node: Node, result: TestResult) -> None:
@@ -119,5 +120,5 @@ def perf_nvme_io_uring(self, node: Node, result: TestResult) -> None:
             disk_setup_type=DiskSetupType.raw,
             disk_type=DiskType.nvme,
             test_result=result,
-            ioengine="io_uring",
+            ioengine=IoEngine.IO_URING,
         )