From 0198d5cc8c786317ab58cf7bd06e20e6e56383fa Mon Sep 17 00:00:00 2001
From: Alex Hultman <alexhultman@gmail.com>
Date: Wed, 23 Jan 2019 14:10:17 +0100
Subject: [PATCH] Add Autobahn testing non-SSL and SSL

---
 tests/Autobahn.js        | 74 ++++++++++++++++++++++++++++++++++++++++
 tests/Makefile           |  2 ++
 tests/fuzzingclient.json | 17 +++++++++
 3 files changed, 93 insertions(+)
 create mode 100644 tests/Autobahn.js
 create mode 100644 tests/Makefile
 create mode 100644 tests/fuzzingclient.json

diff --git a/tests/Autobahn.js b/tests/Autobahn.js
new file mode 100644
index 0000000..c1b0643
--- /dev/null
+++ b/tests/Autobahn.js
@@ -0,0 +1,74 @@
+/* Test server for autobahn, run with ASAN. /exit route shuts down. */
+
+const uWS = require('../dist/uws.js');
+let listenSocket, listenSocketSSL;
+
+/* todo: A good idea is to test compression 1 for non-SSL and 2 for SSL */
+/* or really just test more than twice */
+
+/* Shared, among SSL and non-SSL, behavior of WebSockets */
+const wsBehavior = {
+  /* Options */
+  compression: 2,
+  maxPayloadLength: 16 * 1024 * 1024,
+  idleTimeout: 60,
+  /* Handlers */
+  open: (ws, req) => {
+    console.log('A WebSocket connected via URL: ' + req.getUrl() + '!');
+  },
+  message: (ws, message, isBinary) => {
+    ws.send(message, isBinary);
+  },
+  drain: (ws) => {
+    console.log('WebSocket backpressure: ' + ws.getBufferedAmount());
+  },
+  close: (ws, code, message) => {
+    console.log('WebSocket closed');
+  }
+};
+
+/* The SSL test server is on port 9002 */
+const sslApp = uWS.SSLApp({
+  key_file_name: 'misc/key.pem',
+  cert_file_name: 'misc/cert.pem',
+  passphrase: '1234'
+}).ws('/*', wsBehavior).any('/exit', (res, req) => {
+  /* Stop listening on both SSL and non-SSL */
+  if (listenSocket) {
+    uWS.us_listen_socket_close(listenSocket);
+    uWS.us_listen_socket_close(listenSocketSSL);
+    listenSocket = null;
+  }
+
+  /* Close this connection */
+  res.close();
+}).listen(9002, (token) => {
+  if (token) {
+    listenSocketSSL = token;
+    console.log('SSL test server is up...');
+  }
+});
+
+/* The non-SSL test server is on port 9001 */
+const app = uWS.App().ws('/*', wsBehavior).any('/exit', (res, req) => {
+  /* Stop listening on both SSL and non-SSL */
+  if (listenSocket) {
+    uWS.us_listen_socket_close(listenSocket);
+    uWS.us_listen_socket_close(listenSocketSSL);
+    listenSocket = null;
+  }
+
+  /* Close this connection */
+  res.close();
+}).listen(9001, (token) => {
+  if (token) {
+    listenSocket = token;
+    console.log('Non-SSL test server is up...');
+  }
+});
+
+/* This is required to check for memory leaks */
+process.on('beforeExit', () => {
+  app.forcefully_free();
+  sslApp.forcefully_free();
+});
diff --git a/tests/Makefile b/tests/Makefile
new file mode 100644
index 0000000..9dbde77
--- /dev/null
+++ b/tests/Makefile
@@ -0,0 +1,2 @@
+default:
+	wstest -m fuzzingclient
diff --git a/tests/fuzzingclient.json b/tests/fuzzingclient.json
new file mode 100644
index 0000000..4dd8621
--- /dev/null
+++ b/tests/fuzzingclient.json
@@ -0,0 +1,17 @@
+
+{
+   "outdir": "./reports/servers",
+   "servers": [
+      {
+         "url": "ws://localhost:9001",
+         "agent": "uWebSockets.js non-SSL, compression 2"
+      },
+      {
+         "url": "wss://localhost:9002",
+         "agent": "uWebSockets.js SSL, compression 2"
+      }
+   ],
+   "cases": ["*.*"],
+   "exclude-cases": [],
+   "exclude-agent-cases": {}
+}