Changing logic of fetching MTU size in server size.

[platform/upstream/iotivity.git] / resource / csdk / connectivity / test / uqueue_test.cpp

diff --git a/resource/csdk/connectivity/test/uqueue_test.cpp b/resource/csdk/connectivity/test/uqueue_test.cpp
index c393cd3..f461d46 100644 (file)
--- a/resource/csdk/connectivity/test/uqueue_test.cpp
+++ b/resource/csdk/connectivity/test/uqueue_test.cpp
@@ -22,6 +22,8 @@
 
 #include "uqueue.h"
 
+#include "oic_malloc.h"
+
 class UQueueF : public testing::Test {
 public:
     UQueueF() :
@@ -47,7 +49,7 @@ protected:
 
 u_queue_message_t *CreateQueueMessage(void *data, uint32_t size)
 {
-    u_queue_message_t *message = (u_queue_message_t *) malloc(sizeof(u_queue_message_t));
+    u_queue_message_t *message = (u_queue_message_t *) OICMalloc(sizeof(u_queue_message_t));
     if (NULL == message)
     {
         return NULL;