From 3b1655e030519f1cfd28e52893d36682971c54b3 Mon Sep 17 00:00:00 2001
From: Ben Lesh <blesh@netflix.com>
Date: Tue, 12 Jan 2016 13:34:12 -0800
Subject: [PATCH] fix(WebSocketSubject): ensure error codes passed to WebSocket
 close method

---
 spec/observables/dom/webSocket-spec.js | 51 +++++++++++++++++++++++---
 src/observable/dom/webSocket.ts        | 13 ++++++-
 2 files changed, 57 insertions(+), 7 deletions(-)

diff --git a/spec/observables/dom/webSocket-spec.js b/spec/observables/dom/webSocket-spec.js
index d7ae11aa23..6da61da677 100644
--- a/spec/observables/dom/webSocket-spec.js
+++ b/spec/observables/dom/webSocket-spec.js
@@ -15,7 +15,7 @@ describe('Observable.webSocket', function () {
     teardownMockWebSocket();
   });
 
-  it ('should send a message', function () {
+  it('should send and receive messages', function () {
     var messageReceived = false;
     var subject = Observable.webSocket('ws://mysocket');
 
@@ -27,6 +27,7 @@ describe('Observable.webSocket', function () {
     });
 
     var socket = MockWebSocket.lastSocket();
+    expect(socket.url).toBe('ws://mysocket');
 
     socket.open();
     expect(socket.lastMessageSent()).toBe('ping');
@@ -35,7 +36,7 @@ describe('Observable.webSocket', function () {
     expect(messageReceived).toBe(true);
   });
 
-  it ('receive multiple messages', function () {
+  it('receive multiple messages', function () {
     var expected = ['what', 'do', 'you', 'do', 'with', 'a', 'drunken', 'sailor?'];
     var results = [];
     var subject = Observable.webSocket('ws://mysocket');
@@ -55,7 +56,7 @@ describe('Observable.webSocket', function () {
     expect(results).toEqual(expected);
   });
 
-  it ('should queue messages prior to subscription', function () {
+  it('should queue messages prior to subscription', function () {
     var expected = ['make', 'him', 'walk', 'the', 'plank'];
     var subject = Observable.webSocket('ws://mysocket');
 
@@ -75,7 +76,7 @@ describe('Observable.webSocket', function () {
     expect(socket.sent.length).toBe(expected.length);
   });
 
-  it('should send messages immediately if alreayd open', function () {
+  it('should send messages immediately if already open', function () {
     var subject = Observable.webSocket('ws://mysocket');
     subject.subscribe();
     var socket = MockWebSocket.lastSocket();
@@ -103,7 +104,20 @@ describe('Observable.webSocket', function () {
     expect(socket.readyState).toBe(3); // closed
   });
 
-  it('should allow resubscription after closure', function () {
+  it('should close the socket with a code and a reason when errored', function () {
+    var subject = Observable.webSocket('ws://mysocket');
+    subject.subscribe();
+    var socket = MockWebSocket.lastSocket();
+    socket.open();
+
+    spyOn(socket, 'close').and.callThrough();
+    expect(socket.close).not.toHaveBeenCalled();
+
+    subject.error({ code: 1337, reason: 'Too bad, so sad :('});
+    expect(socket.close).toHaveBeenCalledWith(1337, 'Too bad, so sad :(');
+  });
+
+  it('should allow resubscription after closure via complete', function () {
     var subject = Observable.webSocket('ws://mysocket');
     subject.subscribe();
     var socket1 = MockWebSocket.lastSocket();
@@ -118,6 +132,22 @@ describe('Observable.webSocket', function () {
     expect(socket2).not.toBe(socket1);
     expect(socket2.lastMessageSent()).toBe('a mariner yer not. yarrr.');
   });
+
+  it('should allow resubscription after closure via error', function () {
+    var subject = Observable.webSocket('ws://mysocket');
+    subject.subscribe();
+    var socket1 = MockWebSocket.lastSocket();
+    socket1.open();
+    subject.error({ code: 1337 });
+
+    subject.next('yo-ho! yo-ho!');
+    subject.subscribe();
+    var socket2 = MockWebSocket.lastSocket();
+    socket2.open();
+
+    expect(socket2).not.toBe(socket1);
+    expect(socket2.lastMessageSent()).toBe('yo-ho! yo-ho!');
+  });
 });
 
 var sockets = [];
@@ -145,6 +175,15 @@ MockWebSocket.prototype = {
     return sent.length > 0 ? sent[sent.length - 1] : undefined;
   },
 
+  closeDirty: function (code, reason) {
+    if (this.readyState < 2) {
+      this.readyState = 2;
+      this.closeCode = code;
+      this.closeReason = reason;
+      this.triggerClose({ wasClean: false });
+    }
+  },
+
   triggerClose: function (e) {
     this.readyState = 3;
     this.trigger('close', e);
@@ -176,7 +215,7 @@ MockWebSocket.prototype = {
       this.readyState = 2;
       this.closeCode = code;
       this.closeReason = reason;
-      this.triggerClose({ wasClean: (!code || code === 1000) });
+      this.triggerClose({ wasClean: true });
     }
   },
 
diff --git a/src/observable/dom/webSocket.ts b/src/observable/dom/webSocket.ts
index cddc5beebe..8ca3382329 100644
--- a/src/observable/dom/webSocket.ts
+++ b/src/observable/dom/webSocket.ts
@@ -109,7 +109,18 @@ export class WebSocketSubject<T> extends Subject<T> {
 
         self.destination = Subscriber.create(
           (x) => socket.readyState === 1 && socket.send(x),
-          (e) => socket.close(e),
+          (e) => {
+            const closingObserver = self.closingObserver;
+            if (closingObserver) {
+              closingObserver.next(undefined);
+            }
+            if (e && e.code) {
+              socket.close(e.code, e.reason);
+            } else {
+              self._finalError(new TypeError('WebSocketSubject.error must be called with an object with an error code, ' +
+                'and an optional reason: { code: number, reason: string }'));
+            }
+          },
           ( ) => {
             const closingObserver = self.closingObserver;
             if (closingObserver) {