libt2n: (tomj) don't use std::auto_ptr in command_client, the pointer will get lost...

[libt2n] / test / simplecmd.cpp

diff --git a/test/simplecmd.cpp b/test/simplecmd.cpp
index b2b5bbd..1c8e09e 100644 (file)
--- a/test/simplecmd.cpp
+++ b/test/simplecmd.cpp
@@ -36,10 +36,13 @@ using namespace CppUnit;
 
 string testfunc(const string& str)
 {
+    string ret;
     if (str=="throw")
         throw libt2n::t2n_runtime_error("throw me around");
-    string ret(str);
-    ret+=", testfunc() was here";
+    if (str=="big")
+        ret.insert(0,100*1024,'x');
+    else
+        ret=str+", testfunc() was here";
     return ret;
 }
 
@@ -113,23 +116,28 @@ class test_simplecmd : public TestFixture
 
     CPPUNIT_TEST(SimpleCmd);
     CPPUNIT_TEST(SimpleException);
+    CPPUNIT_TEST(BigReturn);
+    CPPUNIT_TEST(BigParameter);
 
     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 SimpleCmd()
     {
-        pid_t pid;
-
-        switch(pid=fork())
+        switch(child_pid=fork())
         {
             case -1:
             {
@@ -157,7 +165,7 @@ class test_simplecmd : public TestFixture
                 sleep(1);
                 socket_client_connection sc("./socket");
                 sc.set_logging(&cerr,debug);
-                command_client cc(sc);
+                command_client cc(&sc);
 
                 result_container rc;
                 cc.send_command(new testfunc_cmd("hello"),rc);
@@ -171,9 +179,7 @@ class test_simplecmd : public TestFixture
 
     void SimpleException()
     {
-        pid_t pid;
-
-        switch(pid=fork())
+        switch(child_pid=fork())
         {
             case -1:
             {
@@ -201,7 +207,7 @@ class test_simplecmd : public TestFixture
                 sleep(1);
                 socket_client_connection sc("./socket");
                 sc.set_logging(&cerr,debug);
-                command_client cc(sc);
+                command_client cc(&sc);
 
                 result_container rc;
                 cc.send_command(new testfunc_cmd("throw"),rc);
@@ -222,6 +228,89 @@ class test_simplecmd : public TestFixture
         }
     }
 
+    void BigReturn()
+    {
+        switch(child_pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                socket_server ss("./socket");
+                command_server cs(ss);
+
+                // max 10 sec
+                for (int i=0; i < 10; i++)
+                    cs.handle(1000000);
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                // wait till server is up
+                sleep(1);
+                socket_client_connection sc("./socket");
+                command_client cc(&sc);
+
+                result_container rc;
+                cc.send_command(new testfunc_cmd("big"),rc);
+
+                string ret=dynamic_cast<testfunc_res*>(rc.get_result())->get_data();
+
+                CPPUNIT_ASSERT_EQUAL(string().insert(0,100*1024,'x'),ret);
+            }
+        }
+    }
+
+    void BigParameter()
+    {
+        switch(child_pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                socket_server ss("./socket");
+                command_server cs(ss);
+
+                // max 60 sec - we need atleast 28 handle calls to transfer the buffer
+                for (int i=0; i < 60; i++) {
+                    cs.handle(1000000);
+                }
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                // wait till server is up
+                sleep(1);
+                socket_client_connection sc("./socket");
+                command_client cc(&sc);
+
+                result_container rc;
+                cc.send_command(new testfunc_cmd(string().insert(0,100*1024,'y')),rc);
+
+                string ret=dynamic_cast<testfunc_res*>(rc.get_result())->get_data();
+
+                CPPUNIT_ASSERT_EQUAL(string().insert(0,100*1024,'y')+", testfunc() was here",ret);
+            }
+        }
+    }
+
 };
 
 CPPUNIT_TEST_SUITE_REGISTRATION(test_simplecmd);