X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Fcallback.cpp;h=eb600cb28008621e29b800b3472e6be998c8b227;hp=beb0c777db059f3dd9a054d8342e9eac1f04072e;hb=a63e08b83794273da3840e0b8bf15bf0085fe3c4;hpb=6cda58a6dad87ff6efe2277db2155be60edb8d48 diff --git a/test/callback.cpp b/test/callback.cpp index beb0c77..eb600cb 100644 --- a/test/callback.cpp +++ b/test/callback.cpp @@ -1,9 +1,24 @@ -/*************************************************************************** - * Copyright (C) 2004 by Intra2net AG * - * info@intra2net.com * - * * - ***************************************************************************/ +/* +Copyright (C) 2004 by Intra2net AG +The software in this package is distributed under the GNU General +Public License version 2 (with a special exception described below). + +A copy of GNU General Public License (GPL) is included in this distribution, +in the file COPYING.GPL. + +As a special exception, if other files instantiate templates or use macros +or inline functions from this file, or you compile this file and link it +with other works to produce a work based on this file, this file +does not by itself cause the resulting work to be covered +by the GNU General Public License. + +However the source code for this file must still be made available +in accordance with section (3) of the GNU General Public License. + +This exception does not invalidate any other reasons why a work based +on this file might be covered by the GNU General Public License. +*/ #include #include #include @@ -18,42 +33,22 @@ #include -#include -#include -#include +#define BOOST_TEST_DYN_LINK +#include #include #include +#include "test_fixtures.hxx" + using namespace std; using namespace libt2n; -using namespace CppUnit; -class test_callback : public TestFixture +class test_callbackFixture : public KillChildOnShutdownFixture { - CPPUNIT_TEST_SUITE(test_callback); - - CPPUNIT_TEST(NewConnCallback); - CPPUNIT_TEST(ConnClosedCallback); - CPPUNIT_TEST(ConnDeletedCallback); - CPPUNIT_TEST(CallbackOrder); - - CPPUNIT_TEST_SUITE_END(); - +protected: std::vector callback_done; - public: - - void setUp() - { - callback_done.resize(__events_end); - } - - void tearDown() - { - callback_done.clear(); - } - void callback_func(callback_event_type ev, int conn_id) { // we don't care for the conn_id, we just mark the callback as done @@ -63,19 +58,33 @@ class test_callback : public TestFixture callback_done[ev]=true; } - void NewConnCallback() +public: + test_callbackFixture() { - pid_t pid; + callback_done.resize(__events_end); + } + + ~test_callbackFixture() + { + callback_done.clear(); + } +}; + +BOOST_FIXTURE_TEST_SUITE(test_callback, test_callbackFixture) - switch(pid=fork()) +BOOST_AUTO_TEST_CASE(ServerNewConnCallback) +{ + switch(child_pid=fork()) + { + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { string data; // wait till server is up @@ -92,51 +101,55 @@ class test_callback : public TestFixture // close the connection } - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - socket_server ss("./socket"); + // don't call atexit and stuff + _exit(0); + } - ss.add_callback(new_connection,bind(&test_callback::callback_func, boost::ref(*this), new_connection, _1)); + default: + // parent + { + socket_server ss("./socket"); - // max 3 sec - for (int i=0; i < 3; i++) - { - ss.fill_buffer(1000000); + ss.add_callback(new_connection,bind(&test_callback::ServerNewConnCallback::callback_func, boost::ref(*this), new_connection, _1)); - string data; - unsigned int cid; - if(ss.get_packet(data,cid)) - { - server_connection* con=ss.get_connection(cid); - con->write("XYZ"); - } + // max 3 sec + for (int i=0; i < 3; i++) + { + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + con->write("XYZ"); } - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[new_connection])); - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[connection_closed])); - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[connection_deleted])); } + BOOST_CHECK_EQUAL(true,static_cast(callback_done[new_connection])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_closed])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_deleted])); } } +} - void ConnClosedCallback() +BOOST_AUTO_TEST_CASE(ServerConnClosedCallback) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { string data; // wait till server is up @@ -153,51 +166,55 @@ class test_callback : public TestFixture // close the connection } - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - socket_server ss("./socket"); + // don't call atexit and stuff + _exit(0); + } - ss.add_callback(connection_closed,bind(&test_callback::callback_func, boost::ref(*this), connection_closed, _1)); + default: + // parent + { + socket_server ss("./socket"); - // max 3 sec - for (int i=0; i < 3; i++) - { - ss.fill_buffer(1000000); + ss.add_callback(connection_closed,bind(&test_callback::ServerConnClosedCallback::callback_func, boost::ref(*this), connection_closed, _1)); - string data; - unsigned int cid; - if(ss.get_packet(data,cid)) - { - server_connection* con=ss.get_connection(cid); - con->write("XYZ"); - } + // max 3 sec + for (int i=0; i < 3; i++) + { + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + con->write("XYZ"); } - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[new_connection])); - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[connection_closed])); - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[connection_deleted])); } + BOOST_CHECK_EQUAL(false,static_cast(callback_done[new_connection])); + BOOST_CHECK_EQUAL(true,static_cast(callback_done[connection_closed])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_deleted])); } } +} - void ConnDeletedCallback() +BOOST_AUTO_TEST_CASE(ServerConnDeletedCallback) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { string data; // wait till server is up @@ -214,54 +231,58 @@ class test_callback : public TestFixture // close the connection } - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - socket_server ss("./socket"); + // don't call atexit and stuff + _exit(0); + } - ss.add_callback(connection_deleted,bind(&test_callback::callback_func, boost::ref(*this), connection_deleted, _1)); + default: + // parent + { + socket_server ss("./socket"); - // max 3 sec - for (int i=0; i < 3; i++) - { - ss.fill_buffer(1000000); + ss.add_callback(connection_deleted,bind(&test_callback::ServerConnDeletedCallback::callback_func, boost::ref(*this), connection_deleted, _1)); - string data; - unsigned int cid; - if(ss.get_packet(data,cid)) - { - server_connection* con=ss.get_connection(cid); - con->write("XYZ"); - } - ss.cleanup(); + // max 3 sec + for (int i=0; i < 3; i++) + { + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + con->write("XYZ"); } ss.cleanup(); - - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[new_connection])); - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[connection_closed])); - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[connection_deleted])); } + ss.cleanup(); + + BOOST_CHECK_EQUAL(false,static_cast(callback_done[new_connection])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_closed])); + BOOST_CHECK_EQUAL(true,static_cast(callback_done[connection_deleted])); } } +} - void CallbackOrder() +BOOST_AUTO_TEST_CASE(ServerCallbackOrder) +{ + switch(child_pid=fork()) { - pid_t pid; - - switch(pid=fork()) + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { string data; // wait till server is up @@ -280,58 +301,180 @@ class test_callback : public TestFixture // close the connection } + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + socket_server ss("./socket"); + + ss.add_callback(connection_closed,bind(&test_callback::ServerCallbackOrder::callback_func, boost::ref(*this), connection_closed, _1)); + ss.add_callback(connection_deleted,bind(&test_callback::ServerCallbackOrder::callback_func, boost::ref(*this), connection_deleted, _1)); - // don't call atexit and stuff - _exit(0); + bool got_1=false; + + for (int i=0; i < 5; i++) + { + ss.fill_buffer(500000); + + string data; + unsigned int cid; + if(!got_1 && ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + con->write("XYZ"); + got_1=true; + // don't call get_packet anymore + } + ss.cleanup(); } + ss.cleanup(); + + BOOST_CHECK_MESSAGE(static_cast(callback_done[connection_closed]) == true, "closed done"); + BOOST_CHECK_MESSAGE(static_cast(callback_done[connection_deleted]) == false, "not deleted yet"); - default: - // parent + for (int i=0; i < 4; i++) { - socket_server ss("./socket"); + ss.fill_buffer(500000); + + string data; + unsigned int cid; + ss.get_packet(data,cid); + ss.cleanup(); + } + ss.cleanup(); - ss.add_callback(connection_closed,bind(&test_callback::callback_func, boost::ref(*this), connection_closed, _1)); - ss.add_callback(connection_deleted,bind(&test_callback::callback_func, boost::ref(*this), connection_deleted, _1)); + BOOST_CHECK_MESSAGE(static_cast(callback_done[connection_closed]) == true, "closed done (2)"); + BOOST_CHECK_MESSAGE(static_cast(callback_done[connection_deleted]) == true, "not deleted yet(2)"); + } + } +} - bool got_1=false; +BOOST_AUTO_TEST_CASE(ClientConnClosedCallback) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss("./socket"); - for (int i=0; i < 4; i++) + // max 3 sec + for (int i=0; i < 3; i++) { - ss.fill_buffer(500000); + ss.fill_buffer(1000000); string data; unsigned int cid; - if(!got_1 && ss.get_packet(data,cid)) - { - server_connection* con=ss.get_connection(cid); - con->write("XYZ"); - got_1=true; - // don't call get_packet anymore - } - ss.cleanup(); + if(ss.get_packet(data,cid)) + break; } - ss.cleanup(); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + string data; + // wait till server is up + sleep(1); - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[connection_closed])); - CPPUNIT_ASSERT_EQUAL(false,static_cast(callback_done[connection_deleted])); + socket_client_connection sc("./socket"); - for (int i=0; i < 4; i++) + sc.add_callback(connection_closed,bind(&test_callback::ClientConnClosedCallback::callback_func, boost::ref(*this), connection_closed, 0)); + + sc.write("ABC"); + + // wait half a sec + sc.fill_buffer(500000); + sc.get_packet(data); + + BOOST_CHECK_EQUAL(false,static_cast(callback_done[new_connection])); + BOOST_CHECK_EQUAL(true,static_cast(callback_done[connection_closed])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_deleted])); + } + } +} + +BOOST_AUTO_TEST_CASE(ClientConnDeletedCallback) +{ + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss("./socket"); + + // max 3 sec + for (int i=0; i < 3; i++) { - ss.fill_buffer(500000); + ss.fill_buffer(1000000); string data; unsigned int cid; - ss.get_packet(data,cid); - ss.cleanup(); + if(ss.get_packet(data,cid)) + break; } - ss.cleanup(); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[connection_closed])); - CPPUNIT_ASSERT_EQUAL(true,static_cast(callback_done[connection_deleted])); + // don't call atexit and stuff + _exit(0); + } + + default: + // parent + { + string data; + // wait till server is up + sleep(1); + + { + socket_client_connection sc("./socket"); + + sc.add_callback(connection_deleted,bind(&test_callback::ClientConnDeletedCallback::callback_func, boost::ref(*this), connection_deleted, 0)); + + sc.write("ABC"); + + // wait half a sec + sc.fill_buffer(500000); + sc.get_packet(data); } + + BOOST_CHECK_EQUAL(false,static_cast(callback_done[new_connection])); + BOOST_CHECK_EQUAL(false,static_cast(callback_done[connection_closed])); + BOOST_CHECK_EQUAL(true,static_cast(callback_done[connection_deleted])); } } +} -}; - -CPPUNIT_TEST_SUITE_REGISTRATION(test_callback); +BOOST_AUTO_TEST_SUITE_END()