feat: emitting connection_error if a connection fails (#216)

HathorNetwork · Jul 20, 2021 · 77deff8 · 77deff8
1 parent 057ae0e
commit 77deff8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/__tests__/websocket.test.js b/__tests__/websocket.test.js
@@ -28,7 +28,6 @@ test('Ping', (done) => {
 }, 10000)
 
 test('Close', () => {
-
   expect(WebSocketHandler.ws.started).toBe(true);
   expect(WebSocketHandler.ws.connected).toBe(true);
   expect(WebSocketHandler.ws.isOnline).toBe(true);
@@ -38,3 +37,13 @@ test('Close', () => {
   expect(WebSocketHandler.ws.connected).toBe(false);
   expect(WebSocketHandler.ws.isOnline).toBe(false);
 })
+
+test('Error on ws should emit connection_error event', () => {
+  expect.hasAssertions();
+
+  WebSocketHandler.ws.on('connection_error', (evt) => {
+    expect(evt.message).toStrictEqual('expect-me');
+  });
+
+  WebSocketHandler.ws.onError(new Error('expect-me'));
+})
diff --git a/src/websocket.js b/src/websocket.js
@@ -208,6 +208,7 @@ class WS extends EventEmitter {
    * @param {Object} evt Event that contains the error
    */
   onError(evt) {
+    this.emit('connection_error', evt);
     this.onClose();
     // console.log('ws error', window.navigator.onLine, evt);
   }