Free allocated memory while using 'nltk_get_string_from_element' function

Change-Id: Ic4e1ed18923369918110443620a55da7ebe10d31

diff --git a/tests/src/nlp_service_unittests.cpp b/tests/src/nlp_service_unittests.cpp
index 71377db..74e619c 100644 (file)
--- a/tests/src/nlp_service_unittests.cpp
+++ b/tests/src/nlp_service_unittests.cpp
@@ -84,6 +84,11 @@ TEST_F(NlpServiceTest, utc_nlp_service_language_detect_p)
     char *tmp_str = nltk_get_string_from_element(ld_result);
     string detected_language = string(tmp_str);
 
+    if (tmp_str) {
+        free(tmp_str);
+        tmp_str = NULL;
+    }
+
     Py_DECREF(ld_result);
 
     cout << "Detected language: " << detected_language.c_str() << endl;
@@ -110,11 +115,21 @@ TEST_F(NlpServiceTest, utc_nlp_service_pos_tag_p)
         tmp_str = nltk_get_string_from_element(nltk_get_element_from_tuple_by_index(pt_elm_tuple, 1));
         tags.push_back(string(tmp_str));
 
+        if (tmp_str) {
+            free(tmp_str);
+            tmp_str = NULL;
+        }
+
         cout << "tag : " << tmp_str << endl;
 
         tmp_str = nltk_get_string_from_element(nltk_get_element_from_tuple_by_index(pt_elm_tuple, 0));
         tokens.push_back(string(tmp_str));
 
+        if (tmp_str) {
+            free(tmp_str);
+            tmp_str = NULL;
+        }
+
         Py_DECREF(pt_elm_tuple);
     }