X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Ftimeout.cpp;h=936cc6ab2a8e3ffc19a9b7df50049c14ca99d5f0;hp=bf325279615a060fcd5e6ddf3ef42223e46f49c7;hb=HEAD;hpb=b592218402bfa779c024da2f32c4c782d52eaf88 diff --git a/test/timeout.cpp b/test/timeout.cpp index bf32527..936cc6a 100644 --- a/test/timeout.cpp +++ b/test/timeout.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 @@ -17,9 +32,8 @@ #include -#include -#include -#include +#define BOOST_TEST_DYN_LINK +#include #include #include @@ -33,13 +47,12 @@ #include #include -#ifdef HAVE_CONFIG_H +#include "test_fixtures.hxx" + #include -#endif using namespace std; using namespace libt2n; -using namespace CppUnit; string testfunc2(const string& str) { @@ -112,58 +125,18 @@ class testfunc2_cmd : public libt2n::command BOOST_CLASS_EXPORT(testfunc2_cmd) BOOST_CLASS_EXPORT(testfunc2_res) -// this is an evil hack to get access to real_write, don't ever do this in an app!!! -class real_write_connection: public socket_server_connection +class test_timeoutFixture : public KillChildOnShutdownFixture { - public: - void real_write(const std::string& data) - { socket_write(data); } -}; - -// this is an evil hack to get access to real_write, don't ever do this in an app!!! -class real_write_client_connection: public socket_client_connection -{ - public: - void real_write(const std::string& data) - { socket_write(data); } -}; - -class test_timeout : public TestFixture -{ - CPPUNIT_TEST_SUITE(test_timeout); - - CPPUNIT_TEST(ConnectTimeout); - CPPUNIT_TEST(HelloTimeoutNothing); - CPPUNIT_TEST(HelloTimeoutSlowData); - CPPUNIT_TEST(CommandTimeout); - CPPUNIT_TEST(CommandSlowResponse); - CPPUNIT_TEST(DisconnectOnWrite); - CPPUNIT_TEST(DisconnectOnRead); - CPPUNIT_TEST(BreakAccept); - - CPPUNIT_TEST_SUITE_END(); - - pid_t child_pid; - - public: - +protected: typedef uint32_t packet_size_indicator; - void setUp() - { } - - void tearDown() - { - // make sure the server-child is dead before the next test runs - kill(child_pid,SIGKILL); - sleep(1); - } - - void send_hello(string hello_string, socket_server* ss, unsigned int conn_id) + // this is an evil hack to get access to real_write, don't ever do this in an app!!! + class real_write_client_connection: public socket_client_connection { - server_connection *sc=ss->get_connection(conn_id); - sc->write(hello_string); - } + public: + void real_write(const std::string& data) + { socket_write(data); } + }; void send_slow_raw_socket(string data, socket_server* ss, unsigned int conn_id) { @@ -181,97 +154,115 @@ class test_timeout : public TestFixture usleep(200000); } } +}; + +BOOST_FIXTURE_TEST_SUITE(test_timeout, test_timeoutFixture) - void ConnectTimeout() +BOOST_AUTO_TEST_CASE(ConnectTimeout) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); + default: + // parent + { + string data; - string errormsg; + // wait till server is up + sleep(1); - socket_client_connection sc("./socket"); + string errormsg; - CPPUNIT_ASSERT_EQUAL_MESSAGE("connection not closed",true,sc.connection::is_closed()); + socket_client_connection sc("./socket"); - CPPUNIT_ASSERT_EQUAL_MESSAGE("wrong errormessage",string("no more retries left after connect error"),sc.get_last_error_msg()); - } + BOOST_CHECK_MESSAGE(sc.connection::is_closed() == true, "connection not closed"); + + BOOST_CHECK_MESSAGE(sc.get_last_error_msg() == string("no more retries left after connect error"), "wrong errormessage"); } } +} - void HelloTimeoutNothing() +BOOST_AUTO_TEST_CASE(HelloTimeoutNothing) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); - command_client cc(&sc,1000000,1000000); + default: + // parent + { + string data; - t2n_exception* ep=cc.get_constuctor_exception(); + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - string errormsg; - if (ep) - errormsg=ep->what(); + t2n_exception* ep=cc.get_constuctor_exception(); - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); - } + string errormsg; + if (ep) + errormsg=ep->what(); + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void HelloTimeoutSlowData() +BOOST_AUTO_TEST_CASE(HelloTimeoutSlowData) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); @@ -286,47 +277,54 @@ class test_timeout : public TestFixture std::string send_data(hello.str()); send_data.insert(0,(char*)&psize,sizeof(packet_size_indicator)); - ss.add_callback(new_connection,bind(&test_timeout::send_slow_raw_socket, boost::ref(*this), send_data,&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::HelloTimeoutSlowData::send_slow_raw_socket, boost::ref(*this), send_data,&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); - command_client cc(&sc,1000000,1000000); + default: + // parent + { + string data; - t2n_exception* ep=cc.get_constuctor_exception(); + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + command_client cc(&sc,1000000,1000000); - string errormsg; - if (ep) - errormsg=ep->what(); + t2n_exception* ep=cc.get_constuctor_exception(); - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); - } + string errormsg; + if (ep) + errormsg=ep->what(); + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void CommandTimeout() +BOOST_AUTO_TEST_CASE(CommandTimeout) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); @@ -336,54 +334,59 @@ class test_timeout : public TestFixture hello.write((char*)&byteordercheck,sizeof(byteordercheck)); hello << ';'; - ss.add_callback(new_connection,bind(&test_timeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::CommandTimeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) ss.fill_buffer(1000000); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + default: + // parent + { + string data; - command_client cc(&sc,1000000,1000000); - result_container rc; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); - string errormsg; + command_client cc(&sc,1000000,1000000); + result_container rc; - try - { - cc.send_command(new testfunc2_cmd("hello"),rc); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string errormsg; - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); + try + { + cc.send_command(new testfunc2_cmd("hello"),rc); } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void CommandSlowResponse() +BOOST_AUTO_TEST_CASE(CommandSlowResponse) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); @@ -393,7 +396,7 @@ class test_timeout : public TestFixture hello.write((char*)&byteordercheck,sizeof(byteordercheck)); hello << ';'; - ss.add_callback(new_connection,bind(&test_timeout::send_hello, boost::ref(*this), hello.str(),&ss, _1)); + ss.add_callback(new_connection,bind(&test_timeout::CommandSlowResponse::send_hello, boost::ref(*this), hello.str(),&ss, _1)); // max 10 sec for (int i=0; i < 10; i++) @@ -413,101 +416,172 @@ class test_timeout : public TestFixture send_slow_raw_socket(send_data,&ss,cid); } } - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + default: + // parent + { + string data; - command_client cc(&sc,1000000,1000000); - result_container rc; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); - string errormsg; + command_client cc(&sc,1000000,1000000); + result_container rc; - try - { - cc.send_command(new testfunc2_cmd("hello"),rc); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string errormsg; - CPPUNIT_ASSERT_EQUAL(string("timeout exceeded"),errormsg); + try + { + cc.send_command(new testfunc2_cmd("hello"),rc); } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("timeout exceeded"),errormsg); } } +} - void DisconnectOnWrite() +BOOST_AUTO_TEST_CASE(DisconnectOnWrite) +{ + switch(child_pid=fork()) { - 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 { socket_server ss("./socket"); // bail out as soon as we get something ss.fill_buffer(-1); - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string data; + // don't call atexit and stuff + _exit(0); + } - // don't kill us on broken pipe - signal(SIGPIPE, SIG_IGN); + default: + // parent + { + string data; - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); - string errormsg; + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); - string huge(1000000,'x'); + string errormsg; - try - { - sc.write(huge); - } - catch(t2n_transfer_error &e) - { errormsg=e.what(); } - catch(...) - { throw; } + string huge(5000000,'x'); - CPPUNIT_ASSERT_EQUAL(string("write() returned Broken pipe"),errormsg); + try + { + sc.write(huge); } + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("write() returned Broken pipe"),errormsg); } } +} - void DisconnectOnRead() +BOOST_AUTO_TEST_CASE(WriteTwice) +{ + switch(child_pid=fork()) { - pid_t pid2; + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss("./socket"); + + // bail out as soon as we get something + ss.fill_buffer(-1); + ss.fill_buffer(-1); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } - switch(child_pid=fork()) + // don't call atexit and stuff + _exit(0); + } + + default: + // parent { - case -1: + string data; + + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); + + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + + string errormsg; + + sc.write("somedata"); + + sleep(1); + + // server should disconnect now + try { - CPPUNIT_FAIL("fork error"); - break; + sc.write("other data(2)"); + sleep(1); + sc.write("other data(3)"); } - case 0: - // child + catch(t2n_transfer_error &e) + { errormsg=e.what(); } + + BOOST_CHECK_EQUAL(string("write() returned Broken pipe"),errormsg); + } + } +} + +BOOST_AUTO_TEST_CASE(DisconnectOnRead) +{ + pid_t pid2; + + switch(child_pid=fork()) + { + case -1: + { + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { // wait till server is up sleep(1); @@ -517,165 +591,186 @@ class test_timeout : public TestFixture // this is an evil hack to get access to real_write, don't ever do this in an app!!! real_write_client_connection *rwc=(real_write_client_connection*)≻ rwc->real_write(string(10000,'x')); - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - // don't kill us on broken pipe - signal(SIGPIPE, SIG_IGN); + // don't call atexit and stuff + _exit(0); + } - socket_server ss("./socket"); + default: + // parent + { + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); - time_t t0 = time(NULL); + socket_server ss("./socket"); - // max 5 sec - while (time(NULL) < t0 + 5 ) - { - ss.fill_buffer(1000000); + time_t t0 = time(NULL); - string data; - ss.get_packet(data); - } + // max 5 sec + while (time(NULL) < t0 + 5 ) + { + ss.fill_buffer(1000000); - // are we still alive and able to process data? + string data; + ss.get_packet(data); + } + + // are we still alive and able to process data? - switch(pid2=fork()) + switch(pid2=fork()) + { + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_client_connection *sc=new socket_client_connection("./socket"); sc->write(string(10000,'x')); - delete sc; // socket is closed regularly - - // don't run regular cleanup, otherwise cppunit stuff gets called - _exit(0); + delete sc; + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string received; + // don't run regular cleanup, otherwise cppunit stuff gets called + _exit(0); + } - t0 = time(NULL); + default: + // parent + { + string received; - // max 10 sec - while (time(NULL) < t0 + 10 ) - { - ss.fill_buffer(1000000); + t0 = time(NULL); - if (ss.get_packet(received)) - break; - } + // max 10 sec + while (time(NULL) < t0 + 10 ) + { + ss.fill_buffer(1000000); - CPPUNIT_ASSERT_EQUAL(string(10000,'x'),received); + if (ss.get_packet(received)) + break; } + + BOOST_CHECK_EQUAL(string(10000,'x'),received); } } } - kill(pid2,SIGKILL); } + kill(pid2,SIGKILL); +} - void BreakAccept() - { - pid_t pid2; +BOOST_AUTO_TEST_CASE(BreakAccept) +{ + pid_t pid2; - switch(child_pid=fork()) + 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 { // wait till server is really up and waiting sleep(2); // connect with very tight timeout and only 1 retry socket_client_connection sc("./socket",50,1); - - // don't call atexit and stuff - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - // don't kill us on broken pipe - signal(SIGPIPE, SIG_IGN); + // don't call atexit and stuff + _exit(0); + } - socket_server ss("./socket"); + default: + // parent + { + // don't kill us on broken pipe + signal(SIGPIPE, SIG_IGN); - // server is "working" while client wants to connect - sleep(5); + socket_server ss("./socket"); - time_t t0 = time(NULL); + // server is "working" while client wants to connect + sleep(5); - // max 5 sec - while (time(NULL) < t0 + 5 ) - { - ss.fill_buffer(1000000); + time_t t0 = time(NULL); - string data; - ss.get_packet(data); - } + // max 5 sec + while (time(NULL) < t0 + 5 ) + { + ss.fill_buffer(1000000); + + string data; + ss.get_packet(data); + } - // are we still alive and able to process data? + // are we still alive and able to process data? - switch(pid2=fork()) + switch(pid2=fork()) + { + case -1: { - case -1: - { - CPPUNIT_FAIL("fork error"); - break; - } - case 0: - // child + BOOST_FAIL("fork error"); + break; + } + case 0: + // child + { + try { socket_client_connection *sc=new socket_client_connection("./socket"); sc->write(string(10000,'x')); delete sc; // socket is closed regularly - - // don't run regular cleanup, otherwise cppunit stuff gets called - _exit(0); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } - default: - // parent - { - string received; + // don't run regular cleanup, otherwise cppunit stuff gets called + _exit(0); + } - t0 = time(NULL); + default: + // parent + { + string received; - // max 10 sec - while (time(NULL) < t0 + 10 ) - { - ss.fill_buffer(1000000); + t0 = time(NULL); - if (ss.get_packet(received)) - break; - } + // max 10 sec + while (time(NULL) < t0 + 10 ) + { + ss.fill_buffer(1000000); - CPPUNIT_ASSERT_EQUAL(string(10000,'x'),received); + if (ss.get_packet(received)) + break; } + + BOOST_CHECK_EQUAL(string(10000,'x'),received); } } } - kill(pid2,SIGKILL); } -}; + kill(pid2,SIGKILL); +} -CPPUNIT_TEST_SUITE_REGISTRATION(test_timeout); +BOOST_AUTO_TEST_SUITE_END()