From b36a3176dfb080b5ce58713dcb83d9489948acb4 Mon Sep 17 00:00:00 2001 From: Bent Bisballe Nyeng Date: Thu, 3 May 2012 18:35:37 +0200 Subject: Add test client. --- src/Makefile.am | 2 + src/testclient.cc | 208 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ src/testclient.h | 30 ++++++++ 3 files changed, 240 insertions(+) create mode 100644 src/testclient.cc create mode 100644 src/testclient.h diff --git a/src/Makefile.am b/src/Makefile.am index 9fccb45..a56ae39 100644 --- a/src/Makefile.am +++ b/src/Makefile.am @@ -19,6 +19,7 @@ muniad_SOURCES = \ task.cc \ taskmanager.cc \ tasktree.cc \ + testclient.cc \ xml_encode_decode.cc \ xmlparser.cc @@ -35,6 +36,7 @@ EXTRA_DIST = \ task.h \ taskmanager.h \ tasktree.h \ + testclient.h \ xml_encode_decode.h \ xmlparser.h diff --git a/src/testclient.cc b/src/testclient.cc new file mode 100644 index 0000000..5df0d56 --- /dev/null +++ b/src/testclient.cc @@ -0,0 +1,208 @@ +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set et sw=2 ts=2: */ +/*************************************************************************** + * testclient.cc + * + * Thu May 3 10:06:49 CEST 2012 + * Copyright 2012 Bent Bisballe Nyeng + * deva@aasimon.org + ****************************************************************************/ + +/* + * This file is part of Munia. + * + * Munia is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * Munia is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with Munia; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA. + */ +#include "testclient.h" + +#ifdef TEST_TESTCLIENT +//deps: +//cflags: $(LIBWEBSOCKETS_CFLAGS) +//libs: $(LIBWEBSOCKETS_LIBS) +#include "test.h" + +#include +#include +#include +#include +#include + +#include + +struct test { + const char *command; + const char *result; + bool success; +}; + +enum demo_protocols { + PROTOCOL_TASK, +}; + +struct test *current_test = NULL; + +static int callback_task(struct libwebsocket_context *me, + struct libwebsocket *wsi, + enum libwebsocket_callback_reasons reason, + void *user, void *in, size_t len) +{ + unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 4096 + + LWS_SEND_BUFFER_POST_PADDING]; + int l; + + switch (reason) { + + case LWS_CALLBACK_CLOSED: + fprintf(stderr, "mirror: LWS_CALLBACK_CLOSED\n"); + // wsi_mirror = NULL; + break; + + case LWS_CALLBACK_CLIENT_ESTABLISHED: + + /* + * start the ball rolling, + * LWS_CALLBACK_CLIENT_WRITEABLE will come next service + */ + + libwebsocket_callback_on_writable(me, wsi); + break; + + case LWS_CALLBACK_CLIENT_RECEIVE: + fprintf(stderr, "rx %d '%s'\n", (int)len, (char *)in); + current_test->success = strncmp(current_test->result, (char*)in, len) == 0; + current_test++; + break; + + case LWS_CALLBACK_CLIENT_WRITEABLE: + + + l = sprintf((char *)&buf[LWS_SEND_BUFFER_PRE_PADDING], + "%s", current_test->command); + + libwebsocket_write(wsi, &buf[LWS_SEND_BUFFER_PRE_PADDING], + l, LWS_WRITE_TEXT); + + if(current_test->result == NULL) current_test++; + + libwebsocket_callback_on_writable(me, wsi); + + /* + * without at least this delay, we choke the browser + * and the connection stalls, despite we now take care about + * flow control + */ + + //usleep(200); + sleep(1); + break; + + default: + break; + } + + return 0; +} + +static struct libwebsocket_protocols protocols[] = { + { "lws-task-protocol", callback_task, 0, }, + { NULL, NULL, 0 } +}; + +int client(const char *address, int port, struct test tests[]) +{ + current_test = &tests[0]; + + struct libwebsocket_context *context; + struct libwebsocket *wsi_task = NULL; + int ietf_version = -1; // latest + + context = libwebsocket_create_context(CONTEXT_PORT_NO_LISTEN, NULL, + protocols, + libwebsocket_internal_extensions, + NULL, NULL, -1, -1, 0); + if (context == NULL) { + fprintf(stderr, "Creating libwebsocket context failed\n"); + return 1; + } + + + /* + * sit there servicing the websocket context to handle incoming + * packets, and drawing random circles on the mirror protocol websocket + */ + int n = 1; + while(n >= 0 && current_test->command) { + n = libwebsocket_service(context, 1000); + + if (wsi_task == NULL) { + wsi_task = libwebsocket_client_connect(context, address, port, + 0, "/", address, address, + protocols[PROTOCOL_TASK].name, + ietf_version); + + if (wsi_task == NULL) { + fprintf(stderr, "libwebsocket task connect failed\n"); + return -1; + } + + } else { + + /* + fprintf(stderr, "closing mirror session\n"); + libwebsocket_close_and_free_session(context, + wsi_mirror, LWS_CLOSE_STATUS_GOINGAWAY); + */ + } + } + + fprintf(stderr, "Exiting\n"); + + libwebsocket_context_destroy(context); + + return 0; +} + +TEST_BEGIN; + +#define BASE "0 create 0 -1; 0 update 0 \"root\"; " +static struct test tests[] = { + { "observe 0", BASE"", false }, + { "unobserve 0", "0 remove 0", false }, + /* + { "create 0", "", false }, + { "observe 0", BASE"0 create 0 5", false }, + { "unobserve 0", NULL, false }, + { "observe 0", BASE"0 add title description 5", false }, + { "add title2 description 0", BASE"0 add title2 description 6", false }, + { "unobserve 0", NULL, false }, + { "observe 0", BASE"0 add title description 5; add title description 6", false }, + { "unobserve 0", NULL, false }, + */ + { NULL, NULL, false } +}; + +client("holger", 10001, tests); + +struct test *t = &tests[0]; +while(t->command) { + TEST_TRUE(t->success, "%s", t->command); + t++; +} +// TODO: Put some testcode here (see test.h for usable macros). +//TEST_TRUE(false, "No tests yet!"); + +TEST_END; + +#endif/*TEST_TESTCLIENT*/ diff --git a/src/testclient.h b/src/testclient.h new file mode 100644 index 0000000..d92bd9c --- /dev/null +++ b/src/testclient.h @@ -0,0 +1,30 @@ +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set et sw=2 ts=2: */ +/*************************************************************************** + * testclient.h + * + * Thu May 3 10:06:49 CEST 2012 + * Copyright 2012 Bent Bisballe Nyeng + * deva@aasimon.org + ****************************************************************************/ + +/* + * This file is part of Munia. + * + * Munia is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * Munia is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with Munia; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA. + */ +#ifndef __MUNIA_TESTCLIENT_H__ +#define __MUNIA_TESTCLIENT_H__ +#endif/*__MUNIA_TESTCLIENT_H__*/ -- cgit v1.2.3