test: Add Survivability test with additional module api's check

Related-To: VLCLJ-2436

Signed-off-by: Raman, Sudharsan <[email protected]>

Author: sraman4

conformance_tests/sysman/test_sysman_init/src/test_init_sysman_survivability_mode.cpp

@@ -19,7 +19,6 @@ namespace {
 TEST(SysmanInitTests,
     GivenSuccessfulSysmanInitWhenzesDeviceGetReturnsValidDevicesThenSurvivabilityModeIsOffIsDetected) {
 
-  //Initialize Sysman and get devices
   ze_result_t result = zesInit(0);
   ASSERT_EQ(result, ZE_RESULT_SUCCESS);
   uint32_t zes_driver_count = 0;
@@ -41,7 +40,6 @@ TEST(SysmanInitTests,
     ASSERT_EQ(result, ZE_RESULT_SUCCESS);
 
     for (auto device : devices) {
-      // Check that zesDeviceGetProperties is successful for all device handles
       zes_device_properties_t properties = {ZES_STRUCTURE_TYPE_DEVICE_PROPERTIES
 	                                       , nullptr};
       result = zesDeviceGetProperties(device, &properties);
@@ -50,4 +48,59 @@ TEST(SysmanInitTests,
   }
 }
 
+TEST(SysmanInitTests,
+     GivenSysmanInitWhenzesDeviceGetPropertiesReturnsSurvivabilityDetectedErrorThenOnlyFormwareAPIsSucceed) {
+
+  ze_result_t result = zesInit(0);
+  ASSERT_EQ(result, ZE_RESULT_SUCCESS);
+  uint32_t zes_driver_count = 0;
+  EXPECT_EQ(ZE_RESULT_SUCCESS, zesDriverGet(&zes_driver_count, nullptr));
+  EXPECT_GT(zes_driver_count, 0);
+  std::vector<zes_driver_handle_t> drivers(zes_driver_count);
+  result = zesDriverGet(&zes_driver_count, drivers.data());
+  ASSERT_EQ(result, ZE_RESULT_SUCCESS);
+
+  for (auto driver : drivers) {
+    uint32_t device_count = 0;
+    result = zesDeviceGet(driver, &device_count, nullptr);
+    ASSERT_EQ(result, ZE_RESULT_SUCCESS);
+    ASSERT_GT(device_count, 0);
+    std::vector<zes_device_handle_t> devices(device_count);
+    result = zesDeviceGet(driver, &device_count, devices.data());
+    ASSERT_EQ(result, ZE_RESULT_SUCCESS);
+
+    for (auto device : devices) {
+      zes_device_properties_t properties = {ZES_STRUCTURE_TYPE_DEVICE_PROPERTIES, nullptr};
+      result = zesDeviceGetProperties(device, &properties);
+      EXPECT_EQ(result, ZE_RESULT_SUCCESS);
+
+      uint32_t temp_count = 0;
+      temp_count = lzt::get_temp_handle_count(device);
+      if (temp_count == 0) {
+        	FAIL() << "No handles found: "
+		      << _ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+      }
+
+      uint32_t standby_count = 0;
+      auto p_standby_handles = lzt::get_standby_handles(device, standby_count);
+      for (auto p_standby_handle : p_standby_handles) {
+        EXPECT_NE(nullptr, p_standby_handle);
+      }
+
+      uint32_t count = 0;
+      count = lzt::get_firmware_handle_count(device);
+      if (count == 0) {
+		      FAIL() << "No handles found: "
+		      <<_ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+	    }
+
+      auto firmware_handles = lzt::get_firmware_handles(device, count);
+      if (count == 0) {
+          FAIL() << "No handles found: "
+		      << _ze_result_t(ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+      }
+    }
+  }
+}
+
 } // namespace