[unittest] fix segmentation fault

This patch fixes segmentation fault in ne_core_npu_test.
If npu engine can't create driver, api should not be called.

Signed-off-by: yelini-jeong <yelini.jeong@samsung.com>

diff --git a/tests/unittests/ne_core_npu_test.cc b/tests/unittests/ne_core_npu_test.cc
index 152026c143e9fb08a6636d0c304ab5a28e7ecd22..b9f9064d821b31590634084972ae5886fb660783 100644 (file)
--- a/tests/unittests/ne_core_npu_test.cc
+++ b/tests/unittests/ne_core_npu_test.cc
@@ -79,7 +79,7 @@ TEST (ne_core_npu_test, check_dev_status_triv2) {
   std::unique_ptr<DriverAPI> api;
 
   api = DriverAPI::createDriverAPI (NPUCOND_TRIV2_CONN_SOCIP, 0);
-  EXPECT_NE (api.get (), nullptr);
+  ASSERT_NE (api.get (), nullptr);
   EXPECT_GE (api->getDeviceFD (), 0);
   EXPECT_EQ (api->getDeviceID (), 0);
   EXPECT_EQ (api->isReady (), device_state_t::TRINITY_STATE_READY);
@@ -87,7 +87,7 @@ TEST (ne_core_npu_test, check_dev_status_triv2) {
   /** open same device */
   std::unique_ptr<DriverAPI> api2;
   api2 = DriverAPI::createDriverAPI (NPUCOND_TRIV2_CONN_SOCIP, 0);
-  EXPECT_NE (api2.get (), nullptr);
+  ASSERT_NE (api2.get (), nullptr);
   EXPECT_GE (api2->getDeviceFD (), 0);
   EXPECT_EQ (api2->getDeviceID (), 0);
   EXPECT_EQ (api2->isReady (), device_state_t::TRINITY_STATE_READY);
@@ -198,7 +198,7 @@ TEST (ne_core_npu_test, manage_mem_triv2) {
 
   std::unique_ptr<DriverAPI> api;
   api = DriverAPI::createDriverAPI (NPUCOND_TRIV2_CONN_SOCIP, 0);
-  EXPECT_NE (api.get (), nullptr);
+  ASSERT_NE (api.get (), nullptr);
 
   size_t size = 4096;
   int dmabuf = api->alloc (size);
@@ -231,7 +231,7 @@ TEST (ne_core_npu_test, manage_mem_triv2_n) {
 
   std::unique_ptr<DriverAPI> api;
   api = DriverAPI::createDriverAPI (NPUCOND_TRIV2_CONN_SOCIP, 0);
-  EXPECT_NE (api.get (), nullptr);
+  ASSERT_NE (api.get (), nullptr);
 
   size_t size = 4096;
   int dmabuf = api->alloc (size);