enable storage perf test cases for premiumV2 ssd

microsoft · Sep 19, 2024 · cb6f8d7 · cb6f8d7
1 parent 634078f
commit cb6f8d7
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/lisa/messages.py b/lisa/messages.py
@@ -144,6 +144,7 @@ class PerfMessage(MessageBase):
         "unknown",
         "nvme",
         "premiumssd",
+        "premiumv2ssd",
         "ultradisk",
     ],
 )

diff --git a/microsoft/testsuites/performance/storageperf.py b/microsoft/testsuites/performance/storageperf.py
@@ -91,6 +91,51 @@ def perf_ultra_datadisks_1024k(self, node: Node, result: TestResult) -> None:
             disk_type=DiskType.ultradisk,
         )
 
+    @TestCaseMetadata(
+        description="""
+        This test case uses fio to test premiumV2 disk performance with 4K block size.
+        """,
+        priority=3,
+        timeout=TIME_OUT,
+        requirement=simple_requirement(
+            disk=schema.DiskOptionSettings(
+                data_disk_type=schema.DiskType.PremiumV2SSDLRS,
+                os_disk_type=schema.DiskType.PremiumSSDLRS,
+                data_disk_iops=search_space.IntRange(min=160000),
+                data_disk_count=search_space.IntRange(min=2),
+            ),
+        ),
+    )
+    def perf_premiumv2_datadisks_4k(self, node: Node, result: TestResult) -> None:
+        self._perf_premium_datadisks(
+            node=node,
+            test_result=result,
+            disk_type=DiskType.premiumv2ssd,
+        )
+
+    @TestCaseMetadata(
+        description="""
+        This test case uses fio to test premiumV2 disk performance using 1024K block size.
+        """,
+        priority=3,
+        timeout=TIME_OUT,
+        requirement=simple_requirement(
+            disk=schema.DiskOptionSettings(
+                data_disk_type=schema.DiskType.PremiumV2SSDLRS,
+                os_disk_type=schema.DiskType.PremiumSSDLRS,
+                data_disk_iops=search_space.IntRange(min=160000),
+                data_disk_count=search_space.IntRange(min=2),
+            ),
+        ),
+    )
+    def perf_premiumv2_datadisks_1024k(self, node: Node, result: TestResult) -> None:
+        self._perf_premium_datadisks(
+            node=node,
+            test_result=result,
+            block_size=1024,
+            disk_type=DiskType.premiumv2ssd,
+        )
+
     @TestCaseMetadata(
         description="""
         This test case uses fio to test data disk performance with 4K block size.