Tests/UnitTests: Fix SVACE issues 211290 and 211291

This patch resolves the SVACE issues 211290 and 211291, which are
reported by the DEREF_OF_NULL.RET.ALLOC.MINOR checker.

Change-Id: Ida26ed0ed59d90961f5ddf066f625e01cb725da0
Signed-off-by: Wook Song <wook16.song@samsung.com>

diff --git a/tests/unittests/ne_libnpuhost_test.cc b/tests/unittests/ne_libnpuhost_test.cc
index 61366ea6f85903f6ada9ff03707fa59f95742f2a..3e97d6de19827b68b7862ee4ec8dcbdb5023c200 100644 (file)
--- a/tests/unittests/ne_libnpuhost_test.cc
+++ b/tests/unittests/ne_libnpuhost_test.cc
@@ -304,6 +304,10 @@ TEST (ne_libnpuhost_test, get_npumodel_metadata_02) {
   }
 
   extra = (uint8_t *) malloc (extra_size * NPUBIN_META_SIZE);
+  if (extra == nullptr) {
+    fclose (meta_only_bin_fp);
+    FAIL () << "Failed to allocate memory for extra data\n";
+  }
   memset (extra, 0, extra_size * NPUBIN_META_SIZE);
 
   cnt = fwrite (extra, extra_size * NPUBIN_META_SIZE, 1, meta_only_bin_fp);
@@ -483,6 +487,10 @@ TEST (ne_libnpuhost_test, get_npumodel_metadata_04_n) {
   /** Wrong meta data is given */
 
   extra = (uint8_t *) malloc (extra_size * NPUBIN_META_SIZE);
+  if (extra == nullptr) {
+    fclose (meta_only_bin_fp);
+    FAIL () << "Failed to allocate memory for extra data\n";
+  }
   memset (extra, 0, extra_size * NPUBIN_META_SIZE);
 
   cnt = fwrite (extra, extra_size * (NPUBIN_META_SIZE / 2), 1, meta_only_bin_fp);