X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Fcomm.cpp;h=5bba916297461a5d437473453f882398f78d85c4;hp=55c5632daeaab7db410f35d811661957eb39ccb2;hb=517d1214577af263bb4d6821e4e7ad9ba8c94901;hpb=cc68aabb16ec32278df8b071c4c9efec7e9f0dce

diff --git a/test/comm.cpp b/test/comm.cpp
index 55c5632..5bba916 100644
--- a/test/comm.cpp
+++ b/test/comm.cpp
@@ -26,13 +26,17 @@ using namespace std;
 using namespace libt2n;
 using namespace CppUnit;
 
+
 class test_comm : public TestFixture
 {
     CPPUNIT_TEST_SUITE(test_comm);
 
     CPPUNIT_TEST(UnixCommToServer);
     CPPUNIT_TEST(UnixCommToServerAndBack);
+    CPPUNIT_TEST(UnixCommToServerAndBackBig);
     CPPUNIT_TEST(IPCommToServer);
+    CPPUNIT_TEST(IPCommToServerAndBack);
+    CPPUNIT_TEST(IPCommToServerAndBackBig);
 
     CPPUNIT_TEST_SUITE_END();
 
@@ -155,6 +159,66 @@ class test_comm : public TestFixture
         }
     }
 
+    void UnixCommToServerAndBackBig()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                socket_server ss("./socket");
+                ss.set_logging(&cerr,debug);
+
+                // max 10 sec
+                for (int i=0; i < 10; i++)
+                {
+                    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;
+
+                        con->write(string().insert(0,100*1024,'y'));
+                    }
+                }
+                // 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.write(string().insert(0,100*1024,'x'));
+
+                while (!sc.get_packet(data))
+                    sc.fill_buffer(1000000);
+
+                CPPUNIT_ASSERT_EQUAL(string().insert(0,100*1024,'y'),data);
+
+                sc.write("QUIT");
+            }
+        }
+    }
+
     void IPCommToServer()
     {
         pid_t pid;
@@ -172,7 +236,7 @@ class test_comm : public TestFixture
             {
                 // wait till server is up
                 sleep(1);
-                socket_client_connection sc("localhost",6666,socket_client_connection::max_retries_default);
+                socket_client_connection sc(6666);
                 sc.write("hello");
                 // don't call atexit and stuff
                 _exit(0);
@@ -196,6 +260,141 @@ class test_comm : public TestFixture
         CPPUNIT_ASSERT_EQUAL(string("hello"),data);
     }
 
+    void IPCommToServerAndBack()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                socket_server ss(6666);
+                ss.set_logging(&cerr,debug);
+
+                // max 10 sec
+                for (int i=0; i < 10; i++)
+                {
+                    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=="ABC")
+                            con->write("DEF");
+                        else
+                            con->write("xyz");
+                    }
+                }
+                // 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("ABC");
+
+                sc.fill_buffer(1000000);
+                sc.get_packet(data);
+
+                CPPUNIT_ASSERT_EQUAL(string("DEF"),data);
+
+                sc.write("HAHA");
+
+                sc.fill_buffer(1000000);
+                sc.get_packet(data);
+
+                CPPUNIT_ASSERT_EQUAL(string("xyz"),data);
+
+                sc.write("QUIT");
+            }
+        }
+    }
+
+
+    void IPCommToServerAndBackBig()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                socket_server ss(6666);
+                ss.set_logging(&cerr,debug);
+
+                // max 10 sec
+                for (int i=0; i < 10; i++)
+                {
+                    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'));
+                    }
+                }
+                // 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);