X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=test%2Fcallback.cpp;fp=test%2Fcallback.cpp;h=beb0c777db059f3dd9a054d8342e9eac1f04072e;hp=0000000000000000000000000000000000000000;hb=6cda58a6dad87ff6efe2277db2155be60edb8d48;hpb=28cb45a5725e9c6054d7048a9bf969b9f2c94d64

diff --git a/test/callback.cpp b/test/callback.cpp
new file mode 100644
index 0000000..beb0c77
--- /dev/null
+++ b/test/callback.cpp
@@ -0,0 +1,337 @@
+/***************************************************************************
+ *   Copyright (C) 2004 by Intra2net AG                                    *
+ *   info@intra2net.com                                                    *
+ *                                                                         *
+ ***************************************************************************/
+
+#include <sys/types.h>
+#include <unistd.h>
+#include <errno.h>
+#include <signal.h>
+#include <stdio.h>
+
+#include <iostream>
+#include <string>
+#include <sstream>
+#include <stdexcept>
+#include <vector>
+
+#include <boost/bind.hpp>
+
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <cppunit/ui/text/TestRunner.h>
+#include <cppunit/extensions/HelperMacros.h>
+
+#include <socket_client.hxx>
+#include <socket_server.hxx>
+
+using namespace std;
+using namespace libt2n;
+using namespace CppUnit;
+
+class test_callback : public TestFixture
+{
+    CPPUNIT_TEST_SUITE(test_callback);
+
+    CPPUNIT_TEST(NewConnCallback);
+    CPPUNIT_TEST(ConnClosedCallback);
+    CPPUNIT_TEST(ConnDeletedCallback);
+    CPPUNIT_TEST(CallbackOrder);
+
+    CPPUNIT_TEST_SUITE_END();
+
+    std::vector<bool> 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
+        if (callback_done[ev])
+            throw runtime_error("callback already done for this event");
+
+        callback_done[ev]=true;
+    }
+
+    void NewConnCallback()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                string data;
+                // wait till server is up
+                sleep(1);
+
+                {
+                    socket_client_connection sc("./socket");
+
+                    sc.write("ABC");
+
+                    // wait half a sec
+                    sc.fill_buffer(500000);
+                    sc.get_packet(data);
+
+                    // close the connection
+                }
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                socket_server ss("./socket");
+
+                ss.add_callback(new_connection,bind(&test_callback::callback_func, boost::ref(*this), new_connection, _1));
+
+                // 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<bool>(callback_done[new_connection]));
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[connection_closed]));
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[connection_deleted]));
+            }
+        }
+    }
+
+    void ConnClosedCallback()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                string data;
+                // wait till server is up
+                sleep(1);
+
+                {
+                    socket_client_connection sc("./socket");
+
+                    sc.write("ABC");
+
+                    // wait half a sec
+                    sc.fill_buffer(500000);
+                    sc.get_packet(data);
+
+                    // close the connection
+                }
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                socket_server ss("./socket");
+
+                ss.add_callback(connection_closed,bind(&test_callback::callback_func, boost::ref(*this), connection_closed, _1));
+
+                // 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<bool>(callback_done[new_connection]));
+                CPPUNIT_ASSERT_EQUAL(true,static_cast<bool>(callback_done[connection_closed]));
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[connection_deleted]));
+            }
+        }
+    }
+
+    void ConnDeletedCallback()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                string data;
+                // wait till server is up
+                sleep(1);
+
+                {
+                    socket_client_connection sc("./socket");
+
+                    sc.write("ABC");
+
+                    // wait half a sec
+                    sc.fill_buffer(500000);
+                    sc.get_packet(data);
+
+                    // close the connection
+                }
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                socket_server ss("./socket");
+
+                ss.add_callback(connection_deleted,bind(&test_callback::callback_func, boost::ref(*this), connection_deleted, _1));
+
+                // 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();
+                }
+                ss.cleanup();
+
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[new_connection]));
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[connection_closed]));
+                CPPUNIT_ASSERT_EQUAL(true,static_cast<bool>(callback_done[connection_deleted]));
+            }
+        }
+    }
+
+    void CallbackOrder()
+    {
+        pid_t pid;
+
+        switch(pid=fork())
+        {
+            case -1:
+            {
+                CPPUNIT_FAIL("fork error");
+                break;
+            }
+            case 0:
+            // child
+            {
+                string data;
+                // wait till server is up
+                sleep(1);
+
+                {
+                    socket_client_connection sc("./socket");
+
+                    sc.write("1");
+
+                    // wait half a sec
+                    sc.fill_buffer(500000);
+                    sc.get_packet(data);
+
+                    sc.write("2");
+
+                    // close the connection
+                }
+
+                // don't call atexit and stuff
+                _exit(0);
+            }
+
+            default:
+            // parent
+            {
+                socket_server ss("./socket");
+
+                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));
+
+                bool got_1=false;
+
+                for (int i=0; i < 4; 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();
+
+                CPPUNIT_ASSERT_EQUAL(true,static_cast<bool>(callback_done[connection_closed]));
+                CPPUNIT_ASSERT_EQUAL(false,static_cast<bool>(callback_done[connection_deleted]));
+
+                for (int i=0; i < 4; i++)
+                {
+                    ss.fill_buffer(500000);
+
+                    string data;
+                    unsigned int cid;
+                    ss.get_packet(data,cid);
+                    ss.cleanup();
+                }
+                ss.cleanup();
+
+                CPPUNIT_ASSERT_EQUAL(true,static_cast<bool>(callback_done[connection_closed]));
+                CPPUNIT_ASSERT_EQUAL(true,static_cast<bool>(callback_done[connection_deleted]));
+            }
+        }
+    }
+
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(test_callback);