include unistd for sleep method

[libplpbus.git] / src_test / test_client.cc

diff --git a/src_test/test_client.cc b/src_test/test_client.cc
index 29747674190c195dffc8075e81d596e65783f127..dcc22a68f59d0501d9339d36a4fcf34d9ee26c25 100644 (file)
--- a/src_test/test_client.cc
+++ b/src_test/test_client.cc
@@ -4,6 +4,9 @@
  *  Created on: Jun 9, 2010
  *      Author: lamikr
  */
+#include <malloc.h>
+#include <unistd.h>
+
 #include <plp/log.h>
 
 #include "plpbus/BusServer.hh"
@@ -33,21 +36,21 @@ int ClientListenerImpl::response_received(const char *msg_rsp_param) {
 
 int ClientListenerImpl::response_received(const BusMessage *msg_rsp_param) {
        long    type;
+       int     err_flg;
 
-       type    = 0;
-       ((BusMessage *)msg_rsp_param)->get_type(&type);
+       type    = ((BusMessage *)msg_rsp_param)->get_type(&err_flg);
        log_debug("response_received(), type: %ld\n", type);
        ((BusMessage *)msg_rsp_param)->printout();
        return 0;
 }
 
 int ClientListenerImpl::event_received(const char *event_param) {
-       cout << "event_received(char *" << event_param << ") " << endl;
+       log_debug("event_received(%s)\n", event_param);
        return 0;
 }
 
 int ClientListenerImpl::event_received(const BusMessage *event_param) {
-       cout << "event_received(BusMessage *)" << endl;
+       log_debug("event_received(BusMessage *)\n");
        return 0;
 }
 
@@ -56,43 +59,49 @@ int main(int argc, char** argv)
        BusClient               *client;
        ClientListenerImpl      *lstnr;
        int                     err_flg;
-       BusMessage              *busmsg;
+       BusMessage              *bus_msg;
 
        if (argc >= 3) {
                client  = new BusClient();
                lstnr   = new ClientListenerImpl();
                err_flg = client->init(argv[1]);
                if (err_flg == 0) {
-                       cout << "init success" << endl;
+                       log_debug("init success\n");
                        err_flg = client->add_client_listener(lstnr);
                        if (err_flg == 0) {
                                int     err_flg;
                                char    *rsp;
 
-                               busmsg  = new BusMessage(MSG_ID_HELLO);
-                               busmsg->add_string_parameter("a", "hello");
-                               busmsg->add_string_parameter("b", "world");
-                               client->send_message_and_request_response(busmsg);
+                               bus_msg = new BusMessage(MSG_ID_HELLO);
+                               if (bus_msg != NULL) {
+                                       bus_msg->add_string_parameter("a", "hello");
+                                       bus_msg->add_string_parameter("b", "world");
+                                       client->send_message_and_request_response(bus_msg);
 
-                               rsp     = NULL;
-                               err_flg = client->send_message_and_wait_response(argv[2], &rsp);
-                               cout << "rsp: " << rsp << endl;
-                               err_flg = client->send_message_and_request_response(argv[2]);
-                               if (err_flg == 0) {
-                                       cout << "request message send successfully"  << endl;
+                                       rsp     = NULL;
+                                       err_flg = client->send_message_and_wait_response(argv[2], &rsp);
+                                       log_debug("rsp: %s\n", rsp);
+                                       free(rsp);
+                                       err_flg = client->send_message_and_request_response(argv[2]);
+                                       if (err_flg == 0) {
+                                               log_debug("request message send successfully\n");
+                                       }
+                                       sleep(5);
+                                       delete(bus_msg);
                                }
-                               sleep(10);
                        }
                        else {
-                               cout << "client failed to add response listener"  << endl;
+                               log_error("client failed to add response listener\n");
                        }
                }
                else {
-                       cout << "client failed to init"  << endl;
+                       log_error("client failed to init\n");
                }
+               delete(client);
+               delete(lstnr);
        }
        else {
-               cout << "usage: <server_name> <message text>"  << endl;
+               log_info("usage: <server_name> <message text>\n");
        }
        return 0;
 }