[Test] alloc failure case

1. Add assertion when allocating memory for test-data is failed.
2. Replace g_free.

Signed-off-by: Jaeyun <jy1210.jung@samsung.com>

diff --git a/tests/unittest_nnstreamer-edge.cc b/tests/unittest_nnstreamer-edge.cc
index 64672210df0d7b9205f3ca7d6bbb776db1dbf027..712e75e12cc77d6612dbe1b7f8dcc667f6641b22 100644 (file)
--- a/tests/unittest_nnstreamer-edge.cc
+++ b/tests/unittest_nnstreamer-edge.cc
@@ -36,11 +36,12 @@ _get_test_data (bool is_server)
 {\r
   ne_test_data_s *_td;\r
 \r
-  _td = (ne_test_data_s *) malloc (sizeof (ne_test_data_s));\r
-  memset (_td, 0, sizeof (ne_test_data_s));\r
+  _td = (ne_test_data_s *) calloc (1, sizeof (ne_test_data_s));\r
 \r
-  _td->loop = g_main_loop_new (NULL, FALSE);\r
-  _td->is_server = is_server;\r
+  if (_td) {\r
+    _td->loop = g_main_loop_new (NULL, FALSE);\r
+    _td->is_server = is_server;\r
+  }\r
 \r
   return _td;\r
 }\r
@@ -175,6 +176,7 @@ TEST(edge, connectLocal)
   _td_server = _get_test_data (true);\r
   _td_client1 = _get_test_data (false);\r
   _td_client2 = _get_test_data (false);\r
+  ASSERT_TRUE (_td_server != NULL && _td_client1 != NULL && _td_client2 != NULL);\r
   port = nns_edge_get_available_port ();\r
 \r
   /* Prepare server (127.0.0.1:port) */\r
@@ -452,6 +454,7 @@ TEST(edge, setEventCbSetNullCallback)
   int ret;\r
 \r
   _td = _get_test_data (false);\r
+  ASSERT_TRUE (_td != NULL);\r
 \r
   ret = nns_edge_create_handle ("temp-id", NNS_EDGE_CONNECT_TYPE_TCP,\r
       (NNS_EDGE_FLAG_RECV | NNS_EDGE_FLAG_SEND), &edge_h);\r
@@ -494,6 +497,7 @@ TEST(edge, setEventCbInvalidParam02_n)
   int ret;\r
 \r
   _td = _get_test_data (false);\r
+  ASSERT_TRUE (_td != NULL);\r
 \r
   ret = nns_edge_create_handle ("temp-id", NNS_EDGE_CONNECT_TYPE_TCP,\r
       (NNS_EDGE_FLAG_RECV | NNS_EDGE_FLAG_SEND), &edge_h);\r
@@ -3155,7 +3159,6 @@ TEST(edgeQueue, getLengthInvalidParam01_n)
 TEST(edgeQueue, pushInvalidParam01_n)\r
 {\r
   void *data;\r
-  unsigned int i;\r
 \r
   data = malloc (5 * sizeof (unsigned int));\r
   ASSERT_TRUE (data != NULL);\r
@@ -3346,6 +3349,7 @@ TEST(edgeMqtt, connectLocal)
 \r
   _td_server = _get_test_data (true);\r
   _td_client = _get_test_data (false);\r
+  ASSERT_TRUE (_td_server != NULL && _td_client != NULL);\r
 \r
   /* Prepare server (127.0.0.1:port) */\r
   nns_edge_create_handle ("temp-server", NNS_EDGE_CONNECT_TYPE_HYBRID,\r
@@ -3407,7 +3411,7 @@ TEST(edgeMqtt, connectLocal)
 \r
   nns_edge_get_info (client_h, "client_id", &val);\r
   nns_edge_data_set_info (data_h, "client_id", val);\r
-  g_free (val);\r
+  nns_edge_free (val);\r
 \r
   ret = nns_edge_data_set_info (data_h, "test-key", "test-value");\r
   EXPECT_EQ (ret, NNS_EDGE_ERROR_NONE);\r