added test.js

luzhlon · luzhlon · commit 67557e6a8769 · 2017-07-23T14:59:08.000+08:00
diff --git a/test/test.cpp b/test/test.cpp
@@ -2,7 +2,6 @@
 #include "../src/websocket.hpp"
 
 #include <string>
-#include <Windows.h>
 
 const char *opcodes[] = {
     "Additional",           // 0
@@ -18,20 +17,42 @@ const char *opcodes[] = {
 using namespace std;
 using namespace websocket;
 
-int main(int argc, char **argv) {
-    tstream::server ser(5333);
+class MyHandler : public WebSocketHandler {
+public:
+    static void onopen(MyHandler& h) {
+        cout << "OPEN: " << h.getHost() << endl;
+        cout << "GET " << h.getPath() << endl;
+        cout << "Host: " << h.getHost() << endl;
+        cout << "Sec-WebSocket-Key: " << h.getKey() << endl;
+        cout << "Sec-WebSocket-Protocol: " << h.getProtocol() << endl;
+    }
+    static void onmessage(MyHandler& h) {
+        cout << h.data() << endl;
+    }
+    static void onclose(MyHandler& h) {
+        cout << "CLOSED" << endl;
+    }
+};
+
+void test1() {
+    tstream::server ser("127.0.0.1", 5333);
     WebSocketHandler h = ser.accept();
     if (h) {
         cout << "GET " << h.getPath() << endl;
         cout << "Host: " << h.getHost() << endl;
         cout << "Sec-WebSocket-Key: " << h.getKey() << endl;
-        cout << "Sec-WebSocket-Protocol: " << h.getSubProtocol() << endl;
+        cout << "Sec-WebSocket-Protocol: " << h.getProtocol() << endl;
         while (1) {
             if (h.recv())
                 cout << h.data() << endl,
                 h.send(h.data(), true);
             getchar();
         }
     }
+}
+
+int main(int argc, char **argv) {
+    WebSocketServer<MyHandler> ser("127.0.0.1", 5333);
+    ser.run();
     return 0;
 }
diff --git a/test/test.html b/test/test.html
@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
+    <script src="./test.js"></script>
+    <title>test</title>
+  </head>
+  <body>
+    <button onclick="connect()">Connect</button>
+  </body>
+</html>
diff --git a/test/test.js b/test/test.js
@@ -0,0 +1,21 @@
+function connect() {
+    ws = new WebSocket('ws://localhost:5333');
+    // ws.binaryType = 'arraybuffer'
+    ws.onopen = function(ev) {
+        console.log('opened');
+        ws.send('abcdefg');
+        ws.send('12345678');
+        ws.send('ABCDEFG');
+    };
+    ws.onmessage = function(ev) {
+        console.log(ev);
+        last = ev.data;
+    };
+    ws.onclose = function(ev) {
+        console.log('CLOSED');
+    };
+    ws.onerror = function(ev) {
+        console.log('ERROR', ev);
+    };
+}
+connect();
diff --git a/test/xmake.lua b/test/xmake.lua
@@ -2,5 +2,15 @@
 target 'test'
     set_kind 'binary'
     add_headers '../src/*.hpp'
-    add_files '../src/*.cpp'
     add_files 'test.cpp'
+
+    add_linkdirs '$(buildir)'
+    add_links 'websocket'
+
+    add_deps 'websocket'
+
+    if os.host() ~= 'windows' then
+        add_cxxflags '-std=c++11'
+    elseif is_mode 'debug' then
+        add_cxxflags '/MTd'
+    end
