Add test code used to diagnose bug dhbaird#37.

Author: dhbaird

Makefile

@@ -1,9 +1,12 @@
 CXXFLAGS = -std=gnu++0x -Wall
 LDLIBS = -lstdc++
-.PHONY: all clean
+.PHONY: all clean test
 all: example-client example-client-cpp11
 clean:
 	-rm  example-client example-client-cpp11 *.o
+	cd test; $(MAKE) clean
+test:
+	cd test; $(MAKE) test
 testserver: node_modules
 	node example-server.js
 node_modules:

test/Makefile

@@ -0,0 +1,11 @@
+CXXFLAGS = -std=gnu++1y -Wall -I..
+LDLIBS = -lstdc++
+.PHONY: all clean
+test: easywsclient.t
+	node testServer.js & sleep 1 && ./easywsclient.t
+clean:
+	-rm  easywsclient.t *.o
+vpath %.cpp ../
+vpath %.hpp ../
+easywsclient.t: easywsclient.t.o easywsclient.o
+easywsclient.o: easywsclient.cpp easywsclient.hpp

test/easywsclient.t.cpp

@@ -0,0 +1,167 @@
+#include "easywsclient.hpp"
+#ifdef _WIN32
+#pragma comment( lib, "ws2_32" )
+#include <WinSock2.h>
+#endif
+#include <assert.h>
+#include <stdio.h>
+#include <string>
+#include <memory>
+#include <vector>
+#include <utility>
+#include <iostream>
+#include <sstream>
+
+using easywsclient::WebSocket;
+
+template<class T>
+std::string toString(const T& t, size_t maxLen=32)
+{
+    std::stringstream ss;
+    ss << t;
+    std::string s = ss.str();
+    if (s.length() > maxLen) {
+        return s.substr(0, maxLen-3) + "...";
+    }
+    else {
+        return s;
+    }
+}
+
+#define ASSERT_EQ(a, b) \
+    if ((a) != (b)) { \
+        std::cout << "Failed: " #a " == " #b "\n" \
+                  << "Expected: " << toString(b) << "\n" \
+                  << "  Actual: " << toString(a) << "\n"; \
+        throw std::runtime_error("test assertion failed"); \
+    }
+
+namespace {
+
+#ifdef _WIN32
+class WSAInit
+{
+  public:
+    WSAInit()
+    {
+        INT rc;
+        WSADATA wsaData;
+        rc = WSAStartup(MAKEWORD(2, 2), &wsaData);
+        if (rc) {
+            throw std::runtime_error("WSAStartup failed");
+        }
+    }
+    ~WSAInit()
+    {
+        WSACleanup();
+    }
+};
+#endif
+
+class KillServer
+{
+  public:
+    KillServer() : d_ws(WebSocket::from_url(http://github.com/%3Cspan%20class=%22pl-s%22%3E%3Cspan%20class=%22pl-pds%22%3E"%3C/span%3Ews://localhost:8123/killServer%3Cspan%20class=%22pl-pds%22%3E"%3C/span%3E%3C/span%3E))
+    {
+        assert(d_ws);
+    }
+    ~KillServer()
+    {
+        d_ws->send("exit"); // sending any message instructs the server to die
+        d_ws->poll();
+        d_ws->poll();
+    }
+  private:
+    std::unique_ptr<WebSocket> d_ws;
+};
+
+std::string makeString(size_t length)
+{
+    std::vector<char> v;
+    v.reserve(length + 1);
+    for (size_t i = 0; i < length; ++i) {
+        v.push_back('0' + (i % 10));
+    }
+    v.push_back(0);
+    return &v[0];
+}
+
+}
+
+void test()
+{
+#ifdef _WIN32
+    WSAInit wsaInit;
+#endif
+    KillServer killServer;
+
+    std::unique_ptr<WebSocket> ws(WebSocket::from_url("ws://localhost:8123/echoWithSize"));
+    assert(ws);
+
+    ws->send("four");
+    std::string message;
+    while (ws->getReadyState() != WebSocket::CLOSED) {
+        bool gotMessage = false;
+        ws->poll();
+        ws->dispatch([gotMessageOut=&gotMessage, messageOut=&message, ws=ws.get()](const std::string& message) {
+            *gotMessageOut = true;
+            *messageOut = message;
+        });
+        if (gotMessage) {
+            break;
+        }
+    }
+    ASSERT_EQ("4\nfour", message);
+
+    std::vector<std::pair<std::string, std::string> > v;
+    v.emplace_back(    "0",     makeString(0));
+    v.emplace_back(    "1",     makeString(1));
+    v.emplace_back(  "123",   makeString(123));
+    v.emplace_back(  "124",   makeString(124));
+    v.emplace_back(  "125",   makeString(125));
+    v.emplace_back(  "126",   makeString(126));
+    v.emplace_back(  "127",   makeString(127));
+    v.emplace_back(  "128",   makeString(128));
+    v.emplace_back("10000", makeString(10000));
+    v.emplace_back("32767", makeString(32767));
+    v.emplace_back("65500", makeString(65500));
+    v.emplace_back("65529", makeString(65529));
+    v.emplace_back("65530", makeString(65530));
+    v.emplace_back("65531", makeString(65531));
+    v.emplace_back("65532", makeString(65532));
+    v.emplace_back("65533", makeString(65533));
+    v.emplace_back("65534", makeString(65534));
+    v.emplace_back("65535", makeString(65535));
+    v.emplace_back("65536", makeString(65536));
+    v.emplace_back("65537", makeString(65537));
+
+    for (auto i = v.begin(); i != v.end(); ++i) {
+        ws->send(i->second);
+        std::string message;
+        while (ws->getReadyState() != WebSocket::CLOSED) {
+            bool gotMessage = false;
+            ws->poll();
+            ws->dispatch([gotMessageOut=&gotMessage, messageOut=&message, ws=ws.get()](const std::string& message) {
+                *gotMessageOut = true;
+                *messageOut = message;
+            });
+            if (gotMessage) {
+                break;
+            }
+        }
+        ASSERT_EQ(i->first + "\n" + i->second, message);
+    }
+
+    ws->close();
+}
+
+int main()
+{
+    try {
+        test();
+    }
+    catch (...) {
+        throw;
+    }
+    return 0;
+}

test/testServer.js

@@ -0,0 +1,48 @@
+/*
+  Prerequisites:
+
+    1. Install node.js and npm
+    2. npm install ws
+
+  See also,
+
+    http://einaros.github.com/ws/
+
+  To run,
+
+    node example-server.js
+*/
+
+"use strict"; // http://ejohn.org/blog/ecmascript-5-strict-mode-json-and-more/
+var WebSocketServer = require('ws').Server;
+var http = require('http');
+var app = http.createServer();
+var server; // assigned by app.listen() below
+
+var wssEchoWithSize = new WebSocketServer({server: app, path: '/echoWithSize'});
+wssEchoWithSize.on('connection', function(ws) {
+    ws.on('message', function(data, flags) {
+        if (flags.binary) { return; }
+        ws.send(data.length + '\n' + data);
+    });
+    ws.on('close', function() {
+    });
+    ws.on('error', function(e) {
+    });
+});
+
+var wssKillServer = new WebSocketServer({server: app, path: '/killServer'});
+wssKillServer.on('connection', function(ws) {
+    ws.on('message', function(data, flags) {
+        if (flags.binary) { return; }
+        server.close();
+    });
+    ws.on('close', function() {
+    });
+    ws.on('error', function(e) {
+    });
+});
+
+var PORT = parseInt(process.env.PORT || '8123');
+server = app.listen(PORT);
+console.log('Listening on port ' + PORT + '...');