From 42100e82333e32fffbae8b1b0027b2a95a4897c9 Mon Sep 17 00:00:00 2001
From: Pierre Ossman <ossman@cendio.se>
Date: Sun, 18 Apr 2021 11:07:43 +0200
Subject: [PATCH] Add unit tests for connect/attach errors

---
 tests/test.rfb.js | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/tests/test.rfb.js b/tests/test.rfb.js
index c50379d8..5a1b7131 100644
--- a/tests/test.rfb.js
+++ b/tests/test.rfb.js
@@ -163,6 +163,18 @@ describe('Remote Frame Buffer Protocol Client', function () {
                 expect(open).to.have.been.calledOnceWithExactly('ws://HOST:8675/PATH', []);
             });
 
+            it('should report connection problems via event', function () {
+                open.restore();
+                open = sinon.stub(Websock.prototype, 'open');
+                open.throws(Error('Failure'));
+                const client = new RFB(document.createElement('div'), 'ws://HOST:8675/PATH');
+                let callback = sinon.spy();
+                client.addEventListener('disconnect', callback);
+                this.clock.tick();
+                expect(callback).to.have.been.calledOnce;
+                expect(callback.args[0][0].detail.clean).to.be.false;
+            });
+
             it('should handle WebSocket/RTCDataChannel objects', function () {
                 let sock = new FakeWebSocket('ws://HOST:8675/PATH', []);
                 new RFB(document.createElement('div'), sock);
@@ -170,6 +182,19 @@ describe('Remote Frame Buffer Protocol Client', function () {
                 expect(open).to.not.have.been.called;
                 expect(attach).to.have.been.calledOnceWithExactly(sock);
             });
+
+            it('should report attach problems via event', function () {
+                attach.restore();
+                attach = sinon.stub(Websock.prototype, 'attach');
+                attach.throws(Error('Failure'));
+                let sock = new FakeWebSocket('ws://HOST:8675/PATH', []);
+                const client = new RFB(document.createElement('div'), sock);
+                let callback = sinon.spy();
+                client.addEventListener('disconnect', callback);
+                this.clock.tick();
+                expect(callback).to.have.been.calledOnce;
+                expect(callback.args[0][0].detail.clean).to.be.false;
+            });
         });
 
         describe('#disconnect', function () {