[UR] Fix program CTS assesion checks

oneapi-src · Sep 6, 2023 · 845357b · 845357b
1 parent eb061c8
commit 845357b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/test/conformance/program/urProgramGetBuildInfo.cpp b/test/conformance/program/urProgramGetBuildInfo.cpp
@@ -36,17 +36,21 @@ TEST_P(urProgramGetBuildInfoTest, Success) {
 TEST_P(urProgramGetBuildInfoTest, InvalidNullHandleProgram) {
     ur_program_build_status_t programBuildStatus =
         UR_PROGRAM_BUILD_STATUS_ERROR;
-    ASSERT_SUCCESS(urProgramGetBuildInfo(
-        nullptr, device, UR_PROGRAM_BUILD_INFO_STATUS,
-        sizeof(programBuildStatus), &programBuildStatus, nullptr));
+    ASSERT_EQ_RESULT(urProgramGetBuildInfo(nullptr, device,
+                                           UR_PROGRAM_BUILD_INFO_STATUS,
+                                           sizeof(programBuildStatus),
+                                           &programBuildStatus, nullptr),
+                     UR_RESULT_ERROR_INVALID_NULL_HANDLE);
 }
 
 TEST_P(urProgramGetBuildInfoTest, InvalidNullHandleDevice) {
     ur_program_build_status_t programBuildStatus =
         UR_PROGRAM_BUILD_STATUS_ERROR;
-    ASSERT_SUCCESS(urProgramGetBuildInfo(
-        program, nullptr, UR_PROGRAM_BUILD_INFO_STATUS,
-        sizeof(programBuildStatus), &programBuildStatus, nullptr));
+    ASSERT_EQ_RESULT(urProgramGetBuildInfo(program, nullptr,
+                                           UR_PROGRAM_BUILD_INFO_STATUS,
+                                           sizeof(programBuildStatus),
+                                           &programBuildStatus, nullptr),
+                     UR_RESULT_ERROR_INVALID_NULL_HANDLE);
 }
 
 TEST_P(urProgramGetBuildInfoTest, InvalidEnumeration) {

diff --git a/test/conformance/program/urProgramGetInfo.cpp b/test/conformance/program/urProgramGetInfo.cpp
@@ -29,8 +29,9 @@ TEST_P(urProgramGetInfoTest, Success) {
 
 TEST_P(urProgramGetInfoTest, InvalidNullHandleProgram) {
     uint32_t ref_count = 0;
-    ASSERT_SUCCESS(urProgramGetInfo(nullptr, UR_PROGRAM_INFO_REFERENCE_COUNT,
-                                    sizeof(ref_count), &ref_count, nullptr));
+    ASSERT_EQ_RESULT(urProgramGetInfo(nullptr, UR_PROGRAM_INFO_REFERENCE_COUNT,
+                                      sizeof(ref_count), &ref_count, nullptr),
+                     UR_RESULT_ERROR_INVALID_NULL_HANDLE);
 }
 
 TEST_P(urProgramGetInfoTest, InvalidEnumeration) {