From 25776313347172291f8303ea5351f05c956bf500 Mon Sep 17 00:00:00 2001
From: dansu <daniel.sundberg@sciss.se>
Date: Tue, 13 Jan 2015 10:38:53 +0100
Subject: [PATCH] Added unittests for websocket node

---
 test/nodes/core/io/22-websocket_spec.js | 223 ++++++++++++++++++++++++
 1 file changed, 223 insertions(+)
 create mode 100644 test/nodes/core/io/22-websocket_spec.js

diff --git a/test/nodes/core/io/22-websocket_spec.js b/test/nodes/core/io/22-websocket_spec.js
new file mode 100644
index 000000000..4f1da5319
--- /dev/null
+++ b/test/nodes/core/io/22-websocket_spec.js
@@ -0,0 +1,223 @@
+/**
+ * Copyright 2014 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ **/
+
+var ws = require("ws");
+var when = require("when");
+var should = require("should");
+var helper = require("../../helper.js");
+var websocketNode = require("../../../../nodes/core/io/22-websocket.js");
+
+var sockets = [];
+function createClient(listenerid) {
+    return when.promise(function(resolve, reject) {
+        var node = helper.getNode(listenerid);
+        var url = helper.url().replace(/http/, "ws") + node.path;
+        var sock = new ws(url);
+        sockets.push(sock);
+
+        sock.on("open", function() {
+            resolve(sock);
+        });
+
+        sock.on("error", function(err) {
+            reject(err);
+        });
+    });
+}
+
+function closeAll() {
+    for(var i=0;i<sockets.length;i++) {
+        sockets[i].close();
+    }
+    sockets = [];
+}
+
+describe('websocket node', function() {
+
+    before(function(done) {
+        helper.startServer(done);
+    });
+    
+    afterEach(function() {
+        closeAll();
+        helper.unload();
+    });
+
+    it('should be loaded, listener', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" }];
+        helper.load(websocketNode, flow, function() {
+            var n1 = helper.getNode("n1");
+            n1.should.have.property("path", "/ws");
+            done();
+        });
+    });
+
+    it('should handle wholemsg property', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", type:"websocket-listener", wholemsg: "true", path: "/ws2" }];
+        helper.load(websocketNode, flow, function() {
+            var n1 = helper.getNode("n1");
+            n1.should.have.property("wholemsg", false);
+            var n2 = helper.getNode("n2");
+            n2.should.have.property("wholemsg", true);
+            done();
+        });
+    });
+
+    it('should create socket', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket in" }];
+        helper.load(websocketNode, flow, function() {
+            createClient("n1").then(function(sock) {
+                done();
+            });
+        });
+    });
+
+    it('should close socket on delete', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" }];
+        helper.load(websocketNode, flow, function() {
+            createClient("n1").then(function(sock) {
+                sock.on("close", function(code, msg) {
+                    done();
+                });
+                helper.clearFlows();
+            });
+        });
+    });
+
+    it('should receive data', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket in", wires: [["n3"]] },
+        {id:"n3", type: "helper"}];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                var n3 = helper.getNode("n3");
+                n3.on("input", function(msg) {
+                    msg.should.have.property("payload", "hello");
+                    done();
+                });
+                sock.send("hello");
+            });
+        });
+    });
+
+    it('should receive wholemsg', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", wholemsg: "true", path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket in", wires: [["n3"]] },
+        {id:"n3", type: "helper"}];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                sock.send('{"text":"hello"}');
+                var n3 = helper.getNode("n3");
+                n3.on("input", function(msg) {
+                    msg.should.have.property("text", "hello");
+                    done();
+                });
+            });
+        });
+    });
+
+    it('should send', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", type:"helper", wires: [["n3"]] },
+        {id:"n3", server:"n1", type:"websocket out" }];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                sock.on("message", function(msg, flags) {
+                    msg.should.equal("hello");
+                    done();
+                });
+                var n2 = helper.getNode("n2");
+                n2.send({payload: "hello"});
+            });
+        });
+    });
+
+    it('should send wholemsg', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", wholemsg: "true", path: "/ws" },
+        {id:"n2", type:"helper", wires: [["n3"]] },
+        {id:"n3", server:"n1", type:"websocket out" }];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                sock.on("message", function(msg, flags) {
+                    var jmsg = JSON.parse(msg);
+                    jmsg.should.have.property("text", "hello");
+                    done();
+                });
+                var n2 = helper.getNode("n2");
+                n2.send({text: "hello"});
+            });
+        });
+    });
+
+    it('should echo', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket in", wires: [["n3"]] },
+        {id:"n3", server:"n1", type:"websocket out" }];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                sock.on("message", function(msg, flags) {
+                    msg.should.equal("hello");
+                    done();
+                });
+                sock.send("hello");
+            });
+        });
+    });
+
+    it('should echo wholemsg', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", wholemsg: "true",  path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket in", wires: [["n3"]] },
+        {id:"n3", server:"n1", type:"websocket out" }];
+        helper.load(websocketNode, flow, function() {
+            var sock = createClient("n1").then(function(sock) {
+                sock.on("message", function(msg, flags) {
+                    var jmsg = JSON.parse(msg);
+                    jmsg.should.have.property("text","hello");
+                    done();
+                });
+                sock.send('{"text":"hello"}');
+            });
+        });
+    });
+
+    it('should broadcast', function(done) {
+        var flow = [{id:"n1", type:"websocket-listener", path: "/ws" },
+        {id:"n2", server:"n1", type:"websocket out" },
+        {id:"n3", type: "helper", wires: [["n2"]]}
+        ];
+        helper.load(websocketNode, flow, function() {
+            var def1 = when.defer(), def2 = when.defer();
+            when.all([createClient("n1"), createClient("n1")]).then(function(socks){
+                socks[0].on("message", function(msg, flags) {
+                    msg.should.equal("hello");
+                    def1.resolve();
+                });
+                socks[1].on("message", function(msg, flags) {
+                    msg.should.equal("hello");
+                    def2.resolve();
+                });
+                var n3 = helper.getNode("n3");
+                n3.send({payload:"hello"});
+            });
+
+            when.all([def1.promise, def2.promise]).then(function(){
+                done();
+            });
+        });
+    });
+});