implemented mockClose and automatic listener notification for better testing

Author: Christian Jeschke

dist/angular-websocket-mock.js

@@ -1,8 +1,8 @@
 (function (global, factory) {
   if (typeof define === "function" && define.amd) {
-    define(['module', 'angular'], factory);
+    define(["module", "angular"], factory);
   } else if (typeof exports !== "undefined") {
-    factory(module, require('angular'));
+    factory(module, require("angular"));
   } else {
     var mod = {
       exports: {}
@@ -11,7 +11,7 @@
     global.angularWebsocketMock = mod.exports;
   }
 })(this, function (module, _angular) {
-  'use strict';
+  "use strict";
 
   var _angular2 = _interopRequireDefault(_angular);
 
@@ -29,10 +29,17 @@
     var sendQueue = [];
     var pendingSends = [];
     var mock = false;
+    var existingMocks = {};
 
     function $MockWebSocket(url, protocols) {
+      this.url = url;
       this.protocols = protocols;
       this.ssl = /(wss)/i.test(this.url);
+      if (!existingMocks[url]) {
+        existingMocks[url] = [this];
+      } else {
+        existingMocks[url].push(this);
+      }
     }
 
     $MockWebSocket.prototype.send = function (msg) {
@@ -45,6 +52,14 @@
       }
     };
 
+    this.mockClose = function (url, code) {
+      if (existingMocks[url]) {
+        existingMocks[url].map(function (mockSocket) {
+          mockSocket.close(code);
+        });
+      }
+    };
+
     this.mock = function () {
       mock = true;
     };
@@ -57,8 +72,8 @@
       return connectQueue.indexOf(url) > -1;
     };
 
-    $MockWebSocket.prototype.close = function () {
-      pendingCloses.push(true);
+    $MockWebSocket.prototype.close = function (code) {
+      pendingCloses.push({ url: this.url, code: code ? code : 1000 });
     };
 
     function createWebSocketBackend(url, protocols) {
@@ -76,19 +91,41 @@
     this.create = createWebSocketBackend;
     this.createWebSocketBackend = createWebSocketBackend;
 
+    function callOpenCallbacks(url) {
+      existingMocks[url].map(function (socketMock) {
+        if (socketMock.onopen && typeof socketMock.onopen === "function") {
+          socketMock.onopen();
+        }
+      });
+    }
+
+    function callCloseCallbacks(url, code) {
+      existingMocks[url].map(function (socketMock) {
+        if (socketMock.onclose && typeof socketMock.onclose === "function") {
+          socketMock.onclose({ code: code });
+        }
+      });
+    }
+
     this.flush = function () {
       var url, msg, config;
       while (url = pendingConnects.shift()) {
         var i = connectQueue.indexOf(url);
         if (i > -1) {
           connectQueue.splice(i, 1);
+          callOpenCallbacks(url);
         }
         // if (config && config.url) {
         // }
       }
 
-      while (pendingCloses.shift()) {
-        closeQueue.shift();
+      var pendingClose;
+      while (pendingClose = pendingCloses.shift()) {
+        var i = closeQueue.indexOf(pendingClose.url);
+        if (i > -1) {
+          closeQueue.splice(i, 1);
+          callCloseCallbacks(pendingClose.url, pendingClose.code);
+        }
       }
 
       while (msg = pendingSends.shift()) {
@@ -110,8 +147,8 @@
       // connectQueue.push({url: url, protocols: protocols});
     };
 
-    this.expectClose = function () {
-      closeQueue.push(true);
+    this.expectClose = function (url) {
+      closeQueue.push(url);
     };
 
     this.expectSend = function (msg) {

dist/angular-websocket.js

@@ -25,12 +25,6 @@
     };
   }
 
-  var _typeof = typeof Symbol === "function" && typeof Symbol.iterator === "symbol" ? function (obj) {
-    return typeof obj;
-  } : function (obj) {
-    return obj && typeof Symbol === "function" && obj.constructor === Symbol ? "symbol" : typeof obj;
-  };
-
   var Socket;
 
   if (typeof window === 'undefined') {
@@ -412,4 +406,4 @@
 
   exports.default = _angular2.default.module('ngWebSocket');
   module.exports = exports['default'];
-});
+});