test: Add Sysman VF Management CTS

Related-To: VLCLJ-2312

Signed-off-by: Vishnu Khanth <[email protected]>

Author: vishnu-khanth

conformance_tests/sysman/test_sysman_vf_management/src/test_sysman_vf_management.cpp

@@ -13,10 +13,16 @@ namespace lzt = level_zero_tests;
 namespace {
 
 #ifdef USE_ZESINIT
-class VfManagementZesTest : public lzt::ZesSysmanCtsClass {};
+class VfManagementZesTest : public lzt::ZesSysmanCtsClass {
+public:
+  bool is_vf_enabled = false;
+};
 #define VF_MANAGEMENT_TEST VfManagementZesTest
 #else // USE_ZESINIT
-class VfManagementTest : public lzt::SysmanCtsClass {};
+class VfManagementTest : public lzt::SysmanCtsClass {
+public:
+  bool is_vf_enabled = false;
+};
 #define VF_MANAGEMENT_TEST VfManagementTest
 #endif // USE_ZESINIT
 
@@ -57,6 +63,7 @@ TEST_F(
   for (auto device : devices) {
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 0) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       auto vf_handles = lzt::get_vf_handles(device, count);
 
@@ -67,6 +74,10 @@ TEST_F(
       LOG_INFO << "No VF handles found for this device!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(
@@ -87,13 +98,18 @@ TEST_F(
     }
 
     if (count_initial > 0 && count_later > 0) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       EXPECT_EQ(count_initial, count_later);
       EXPECT_EQ(vf_handles_initial, vf_handles_later);
     } else {
       LOG_INFO << "No VF handles found for this device!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(
@@ -102,6 +118,7 @@ TEST_F(
   for (auto device : devices) {
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 0) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       uint32_t test_count = count + 1;
       auto vf_handles = lzt::get_vf_handles(device, test_count);
@@ -115,6 +132,10 @@ TEST_F(
       LOG_INFO << "No VF handles found for this device!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(
@@ -123,6 +144,7 @@ TEST_F(
   for (auto device : devices) {
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 1) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       uint32_t test_count = count - 1;
       auto vf_handles = lzt::get_vf_handles(device, test_count);
@@ -132,11 +154,16 @@ TEST_F(
         EXPECT_NE(vf_handle, nullptr);
       }
     } else if (count == 1) {
+      is_vf_enabled = true;
       LOG_INFO << "Insufficient number of VF handles to validate this test";
     } else {
       LOG_INFO << "No VF handles found for this device!!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(
@@ -145,6 +172,7 @@ TEST_F(
   for (auto device : devices) {
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 0) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       auto vf_handles = lzt::get_vf_handles(device, count);
 
@@ -156,6 +184,10 @@ TEST_F(
       LOG_INFO << "No VF handles found for this device!!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(
@@ -164,6 +196,7 @@ TEST_F(
   for (auto device : devices) {
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 0) {
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       auto vf_handles = lzt::get_vf_handles(device, count);
 
@@ -176,6 +209,10 @@ TEST_F(
       LOG_INFO << "No VF handles found for this device!!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 TEST_F(VF_MANAGEMENT_TEST,
@@ -184,6 +221,7 @@ TEST_F(VF_MANAGEMENT_TEST,
     uint32_t count = lzt::get_vf_handles_count(device);
     if (count > 1) {
       std::vector<uint32_t> vf_ids{};
+      is_vf_enabled = true;
       LOG_INFO << "VF is enabled on this device!!";
       auto vf_handles = lzt::get_vf_handles(device, count);
 
@@ -195,11 +233,16 @@ TEST_F(VF_MANAGEMENT_TEST,
       std::set<uint32_t> unique_vf_ids(vf_ids.begin(), vf_ids.end());
       EXPECT_EQ(vf_ids.size(), unique_vf_ids.size());
     } else if (count == 1) {
+      is_vf_enabled = true;
       LOG_INFO << "Insufficient number of VF handles to validate this test";
     } else {
       LOG_INFO << "No VF handles found for this device!!";
     }
   }
+
+  if (!is_vf_enabled) {
+    FAIL() << "No VF handles found in all devices!!";
+  }
 }
 
 } // namespace