X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Fcomm.cpp;h=47ca385f13027f9d41cbe1dc689b942aaf9a93d5;hp=169c1839741dd72a0860d7f40f3fdf0607ae0244;hb=12fbab79c05ca1cb5530a22a8e35b1fb48811d0a;hpb=58b327c6c65474fdfde7d0d99d66d3986400f208 diff --git a/test/comm.cpp b/test/comm.cpp index 169c183..47ca385 100644 --- a/test/comm.cpp +++ b/test/comm.cpp @@ -9,6 +9,7 @@ #include #include #include +#include #include #include @@ -26,6 +27,7 @@ using namespace std; using namespace libt2n; using namespace CppUnit; + class test_comm : public TestFixture { CPPUNIT_TEST_SUITE(test_comm); @@ -35,23 +37,29 @@ class test_comm : public TestFixture CPPUNIT_TEST(UnixCommToServerAndBackBig); CPPUNIT_TEST(IPCommToServer); CPPUNIT_TEST(IPCommToServerAndBack); + CPPUNIT_TEST(IPCommToServerAndBackBig); CPPUNIT_TEST_SUITE_END(); + pid_t child_pid; + public: void setUp() { } void tearDown() - { } + { + // make sure the server-child is dead before the next test runs + kill(child_pid,SIGKILL); + sleep(1); + } void UnixCommToServer() { - pid_t pid; string data; - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -61,10 +69,17 @@ class test_comm : public TestFixture case 0: // child { - // wait till server is up - sleep(1); - socket_client_connection sc("./socket"); - sc.write("hello"); + try + { + // wait till server is up + sleep(1); + socket_client_connection sc("./socket"); + sc.write("hello"); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + // don't call atexit and stuff _exit(0); } @@ -74,8 +89,10 @@ class test_comm : public TestFixture { socket_server ss("./socket"); + time_t t0 = time(NULL); + // max 10 sec - for (int i=0; i < 10; i++) + while (time(NULL) < t0 + 10 ) { ss.fill_buffer(1000000); @@ -89,9 +106,7 @@ class test_comm : public TestFixture void UnixCommToServerAndBack() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -101,30 +116,39 @@ class test_comm : public TestFixture case 0: // child { - socket_server ss("./socket"); - ss.set_logging(&cerr,debug); - - // max 10 sec - for (int i=0; i < 10; i++) + try { - ss.fill_buffer(1000000); + socket_server ss("./socket"); + ss.set_logging(&cerr,debug); - string data; - unsigned int cid; + time_t t0 = time(NULL); - if(ss.get_packet(data,cid)) + // max 10 sec + while (time(NULL) < t0 + 10 ) { - server_connection* con=ss.get_connection(cid); + ss.fill_buffer(1000000); - if (data=="QUIT") - break; + string data; + unsigned int cid; - if (data=="ABC") - con->write("DEF"); - else - con->write("xyz"); + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + + if (data=="QUIT") + break; + + if (data=="ABC") + con->write("DEF"); + else + con->write("xyz"); + } } + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } + // don't call atexit and stuff _exit(0); } @@ -159,9 +183,7 @@ class test_comm : public TestFixture void UnixCommToServerAndBackBig() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -171,27 +193,37 @@ class test_comm : public TestFixture case 0: // child { - socket_server ss("./socket"); - ss.set_logging(&cerr,debug); - - // max 10 sec - for (int i=0; i < 10; i++) + try { - ss.fill_buffer(1000000); + socket_server ss("./socket"); + ss.set_logging(&cerr,debug); - string data; - unsigned int cid; + time_t t0 = time(NULL); - if(ss.get_packet(data,cid)) + // max 10 sec + while (time(NULL) < t0 + 10 ) { - server_connection* con=ss.get_connection(cid); + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); - if (data=="QUIT") - break; + if (data=="QUIT") + break; - con->write(string().insert(0,100*1024,'y')); + con->write(string().insert(0,100*1024,'y')); + } } + std::cerr << "child: OVER" << std::endl; + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } + // don't call atexit and stuff _exit(0); } @@ -219,10 +251,9 @@ class test_comm : public TestFixture void IPCommToServer() { - pid_t pid; string data; - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -232,10 +263,17 @@ class test_comm : public TestFixture case 0: // child { - // wait till server is up - sleep(1); - socket_client_connection sc(6666); - sc.write("hello"); + try + { + // wait till server is up + sleep(1); + socket_client_connection sc(6666); + sc.write("hello"); + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; + } + // don't call atexit and stuff _exit(0); } @@ -245,8 +283,10 @@ class test_comm : public TestFixture { socket_server ss(6666); + time_t t0 = time(NULL); + // max 10 sec - for (int i=0; i < 10; i++) + while (time(NULL) < t0 + 10 ) { ss.fill_buffer(1000000); @@ -260,9 +300,7 @@ class test_comm : public TestFixture void IPCommToServerAndBack() { - pid_t pid; - - switch(pid=fork()) + switch(child_pid=fork()) { case -1: { @@ -272,29 +310,37 @@ class test_comm : public TestFixture case 0: // child { - socket_server ss(6666); - ss.set_logging(&cerr,debug); - - // max 10 sec - for (int i=0; i < 10; i++) + try { - ss.fill_buffer(1000000); + socket_server ss(6666); + ss.set_logging(&cerr,debug); - string data; - unsigned int cid; + time_t t0 = time(NULL); - if(ss.get_packet(data,cid)) + // max 10 sec + while (time(NULL) < t0 + 10 ) { - server_connection* con=ss.get_connection(cid); + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); - if (data=="QUIT") - break; + if (data=="QUIT") + break; - if (data=="ABC") - con->write("DEF"); - else - con->write("xyz"); + if (data=="ABC") + con->write("DEF"); + else + con->write("xyz"); + } } + } catch(...) + { + std::cerr << "exception in child. ignoring\n"; } // don't call atexit and stuff _exit(0); @@ -327,6 +373,75 @@ class test_comm : public TestFixture } } + + void IPCommToServerAndBackBig() + { + switch(child_pid=fork()) + { + case -1: + { + CPPUNIT_FAIL("fork error"); + break; + } + case 0: + // child + { + try + { + socket_server ss(6666); + ss.set_logging(&cerr,debug); + + time_t t0 = time(NULL); + + // max 10 sec + while (time(NULL) < t0 + 10 ) + { + ss.fill_buffer(1000000); + + string data; + unsigned int cid; + + if(ss.get_packet(data,cid)) + { + server_connection* con=ss.get_connection(cid); + + socket_handler* alias= dynamic_cast< socket_handler* >(con); + + if (data=="QUIT") + break; + + alias->set_write_block_size( 4093 ); + con->write(string().insert(0,2048*1024,'y')); + } + } + } 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); + socket_client_connection sc(6666); + + sc.write(string().insert(0,100*1024,'x')); + + while (!sc.get_packet(data)) + sc.fill_buffer(1000000); + + CPPUNIT_ASSERT_EQUAL(string().insert(0,2048*1024,'y'),data); + + sc.write("QUIT"); + } + } + } // eo IPCommToServerAndBackBig() }; CPPUNIT_TEST_SUITE_REGISTRATION(test_comm);