1 files changed, 69 insertions, 0 deletions

diff --git a/src/tests/itoserver.cpp b/src/tests/itoserver.cpp
new file mode 100644
index 0000000..4f5c644
--- /dev/null
+++ b/src/tests/itoserver.cpp
@@ -0,0 +1,69 @@
+#include "bu/itoserver.h"
+#include "bu/protocol.h"
+#include "bu/client.h"
+
+#define BU_TRACE
+#include "bu/trace.h"
+
+class ProtocolEcho : public Bu::Protocol
+{
+public:
+        ProtocolEcho()
+        {
+                TRACE();
+        }
+        virtual ~ProtocolEcho()
+        {
+                TRACE();
+        }
+
+        virtual void onNewConnection( Bu::Client *pClient )
+        {
+                TRACE();
+                // Huh...
+        }
+
+        virtual void onNewData( Bu::Client *pClient )
+        {
+                TRACE();
+                pClient->write( pClient->getInput().getStr(), pClient->getInputSize() );
+                pClient->seek( pClient->getInputSize() );
+        }
+};
+
+class TestServer : public Bu::ItoServer
+{
+public:
+        TestServer()
+        {
+                TRACE();
+        }
+        virtual ~TestServer()
+        {
+                TRACE();
+        }
+
+        virtual void onNewConnection( Bu::Client *pClient, int iPort )
+        {
+                TRACE();
+                pClient->setProtocol( new ProtocolEcho() );
+        }
+
+        virtual void onClosedConnection( Bu::Client *pClient )
+        {
+                TRACE();
+                delete pClient->getProtocol();
+        }
+};
+
+int main()
+{
+        TRACE();
+        
+        TestServer ts;
+
+        ts.addPort( 5555 );
+        ts.start();
+        ts.join();
+}
+