From 7ca123f9e6ca703d1994fa8a72d9d54e1fa643d6 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <oleksandr.pavlyk@intel.com>
Date: Wed, 23 Jun 2021 11:52:15 -0500
Subject: [PATCH] Added test class to test DRef=nullptr case

---
 dpctl-capi/tests/test_sycl_device_manager.cpp | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/dpctl-capi/tests/test_sycl_device_manager.cpp b/dpctl-capi/tests/test_sycl_device_manager.cpp
index 05141009a4..50df3ea8cf 100644
--- a/dpctl-capi/tests/test_sycl_device_manager.cpp
+++ b/dpctl-capi/tests/test_sycl_device_manager.cpp
@@ -93,6 +93,7 @@ struct TestDPCTLGetDevices : public ::testing::TestWithParam<int>
         EXPECT_NO_FATAL_FAILURE(DV = DPCTLDeviceMgr_GetDevices(GetParam()));
         EXPECT_TRUE(DV != nullptr);
         EXPECT_NO_FATAL_FAILURE(nDevices = DPCTLDeviceVector_Size(DV));
+        EXPECT_TRUE(nDevices == DPCTLDeviceMgr_GetNumDevices(GetParam()));
     }
 
     void SetUp()
@@ -206,3 +207,26 @@ INSTANTIATE_TEST_SUITE_P(
                       DPCTLSyclDeviceType::DPCTL_ACCELERATOR,
                       DPCTLSyclDeviceType::DPCTL_GPU,
                       DPCTLSyclDeviceType::DPCTL_CPU));
+
+struct TestDPCTLDeviceMgrNullReference : public ::testing::Test
+{
+    DPCTLSyclDeviceRef nullDRef = nullptr;
+};
+
+TEST_F(TestDPCTLDeviceMgrNullReference, ChkPrintDeviceInfo)
+{
+    EXPECT_NO_FATAL_FAILURE(DPCTLDeviceMgr_PrintDeviceInfo(nullDRef));
+}
+
+TEST_F(TestDPCTLDeviceMgrNullReference, ChkGetRelativeId)
+{
+    EXPECT_NO_FATAL_FAILURE(DPCTLDeviceMgr_GetRelativeId(nullDRef));
+}
+
+TEST_F(TestDPCTLDeviceMgrNullReference, ChkGetPositionInDevices)
+{
+    int mask = DPCTLSyclDeviceType::DPCTL_ALL |
+               DPCTLSyclBackendType::DPCTL_ALL_BACKENDS;
+    EXPECT_NO_FATAL_FAILURE(
+        DPCTLDeviceMgr_GetPositionInDevices(nullDRef, mask));
+}