test: Add tests for zer API's

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

Author: vishnu-khanth

scripts/templates/nullddi.cpp.mako

@@ -132,6 +132,17 @@ ${tbl['export']['name']}(
     %endfor
     )
 {
+    %if n == 'zer':
+    auto zer_api_disable = getenv_string( "ZEL_TEST_NULL_DRIVER_DISABLE_ZER_API" );
+    #ifndef ZEL_NULL_DRIVER_ID
+    #define ZEL_NULL_DRIVER_ID 1
+    #endif
+    std::string null_driver_id_str = std::to_string(ZEL_NULL_DRIVER_ID);
+    auto zer_api_unsupported = (zer_api_disable == null_driver_id_str);
+    if(zer_api_unsupported)
+        return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+
+    %endif
     if( nullptr == pDdiTable )
         return ${X}_RESULT_ERROR_INVALID_NULL_POINTER;
 

source/drivers/null/zer_nullddi.cpp

@@ -137,6 +137,15 @@ zerGetGlobalProcAddrTable(
     zer_global_dditable_t* pDdiTable                ///< [in,out] pointer to table of DDI function pointers
     )
 {
+    auto zer_api_disable = getenv_string( "ZEL_TEST_NULL_DRIVER_DISABLE_ZER_API" );
+    #ifndef ZEL_NULL_DRIVER_ID
+    #define ZEL_NULL_DRIVER_ID 1
+    #endif
+    std::string null_driver_id_str = std::to_string(ZEL_NULL_DRIVER_ID);
+    auto zer_api_unsupported = (zer_api_disable == null_driver_id_str);
+    if(zer_api_unsupported)
+        return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+
     if( nullptr == pDdiTable )
         return ZE_RESULT_ERROR_INVALID_NULL_POINTER;
 

test/CMakeLists.txt

@@ -620,6 +620,8 @@ function(add_runtime_api_test test_scenario init_method driver_config)
         set(gtest_filter "*GivenLevelZeroLoaderPresentWithLoaderInterceptEnabledAndDdiExtNotSupportedWhenCallingRuntimeApisAfter${init_method}ThenExpectNullDriverIsReachedSuccessfully")
     elseif(test_scenario STREQUAL "ddi_ext_v1_0")
         set(gtest_filter "*GivenLevelZeroLoaderPresentWithLoaderInterceptEnabledAndDdiExtSupportedWithVersion1_0WhenCallingRuntimeApisAfter${init_method}ThenExpectErrorUninitialized")
+    elseif(test_scenario STREQUAL "runtime_api_unsupported")
+        set(gtest_filter "*GivenLevelZeroLoaderPresentWithLoaderInterceptEnabledAndRuntimeApiUnsupportedWhenCallingRuntimeApisAfter${init_method}ThenExpectErrorUnsupportedFeature")
     endif()
 
     # Add the test
@@ -654,4 +656,9 @@ foreach(init_method IN LISTS init_methods)
         # Add multi-driver tests
         add_runtime_api_test(${test_scenario} ${init_method} "multi_driver")
     endforeach()
+    
+    # Add runtime_api_unsupported tests for single driver
+    if(init_method STREQUAL "ZeInitDrivers")
+        add_runtime_api_test("runtime_api_unsupported" ${init_method} "single_driver")
+    endif()
 endforeach()

test/loader_api.cpp

@@ -2509,6 +2509,97 @@ TEST_F(DriverOrderingTest,
     EXPECT_NE(errorDesc, nullptr);
     EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNINITIALIZED"));
   }
+
+  TEST(
+      RuntimeApiLoaderDriverInteraction,
+      GivenLevelZeroLoaderPresentWithLoaderInterceptEnabledAndRuntimeApiUnsupportedWhenCallingRuntimeApisAfterZeInitDriversThenExpectErrorUnsupportedFeature)
+  {
+    uint32_t pInitDriversCount = 0;
+    ze_init_driver_type_desc_t desc = {ZE_STRUCTURE_TYPE_INIT_DRIVER_TYPE_DESC};
+    desc.flags = UINT32_MAX;
+    desc.pNext = nullptr;
+    putenv_safe(const_cast<char *>("ZE_ENABLE_LOADER_INTERCEPT=1"));
+    putenv_safe(const_cast<char *>("ZEL_TEST_NULL_DRIVER_DISABLE_ZER_API=1"));
+
+    std::vector<ze_driver_handle_t> drivers;
+    EXPECT_EQ(ZE_RESULT_SUCCESS, zeInitDrivers(&pInitDriversCount, nullptr, &desc));
+    drivers.resize(pInitDriversCount);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, zeInitDrivers(&pInitDriversCount, drivers.data(), &desc));
+    EXPECT_GT(pInitDriversCount, 0);
+
+    uint32_t deviceId = 0;
+
+    const char *errorDesc = nullptr;
+    ze_result_t errorDescResult{};
+
+    deviceId = zerTranslateDeviceHandleToIdentifier(nullptr);
+    EXPECT_EQ(UINT32_MAX, deviceId);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+
+    ze_device_handle_t translatedDevice = zerTranslateIdentifierToDeviceHandle(deviceId);
+    EXPECT_EQ(nullptr, translatedDevice);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+
+    ze_context_handle_t defaultContext = zerGetDefaultContext();
+    EXPECT_EQ(nullptr, defaultContext);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+  }
+
+  /*
+  TEST(
+      RuntimeApiLoaderDriverInteraction,
+      GivenLevelZeroLoaderPresentWithLoaderInterceptEnabledAndRuntimeApiUnsupportedWhenCallingRuntimeApisAfterZeInitThenExpectErrorUnsupportedFeature)
+  {
+    putenv_safe(const_cast<char *>("ZE_ENABLE_LOADER_INTERCEPT=1"));
+    putenv_safe(const_cast<char *>("ZEL_TEST_NULL_DRIVER_DISABLE_ZER_API=1"));
+
+    uint32_t driverGetCount = 0;
+    EXPECT_EQ(ZE_RESULT_SUCCESS, zeInit(0));
+    EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGet(&driverGetCount, nullptr));
+    EXPECT_GT(driverGetCount, 0);
+
+    uint32_t deviceId = 0;
+
+    const char *errorDesc = nullptr;
+    ze_result_t errorDescResult{};
+
+    deviceId = zerTranslateDeviceHandleToIdentifier(nullptr);
+    EXPECT_EQ(UINT32_MAX, deviceId);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+
+    ze_device_handle_t translatedDevice = zerTranslateIdentifierToDeviceHandle(deviceId);
+    EXPECT_EQ(nullptr, translatedDevice);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+
+    ze_context_handle_t defaultContext = zerGetDefaultContext();
+    EXPECT_EQ(nullptr, defaultContext);
+
+    errorDescResult = zerGetLastErrorDescription(&errorDesc);
+    EXPECT_EQ(ZE_RESULT_SUCCESS, errorDescResult);
+    EXPECT_NE(errorDesc, nullptr);
+    EXPECT_EQ(0, strcmp(errorDesc, "ERROR UNSUPPORTED FEATURE"));
+  }
+    */
 
 
 } // namespace