WebsocketProviderTest updated

web3 · Mar 27, 2019 · 8240a27 · 8240a27
1 parent a4ad5e6
commit 8240a27
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/packages/web3-providers/src/providers/WebsocketProvider.js b/packages/web3-providers/src/providers/WebsocketProvider.js
@@ -208,7 +208,7 @@ export default class WebsocketProvider extends AbstractSocketProvider {
 
                 try {
                     this.connection.send(JSON.stringify(payload));
-                } catch(error) {
+                } catch (error) {
                     reject(error);
                 }
 
@@ -236,7 +236,9 @@ export default class WebsocketProvider extends AbstractSocketProvider {
             }
 
             this.once('connect', () => {
-                this.sendPayload(payload).then(resolve).catch(reject);
+                this.sendPayload(payload)
+                    .then(resolve)
+                    .catch(reject);
             });
         });
     }

diff --git a/packages/web3-providers/tests/src/providers/WebsocketProviderTest.js b/packages/web3-providers/tests/src/providers/WebsocketProviderTest.js
@@ -353,6 +353,20 @@ describe('WebsocketProviderTest', () => {
         expect(socketMock.send).toHaveBeenCalledWith('{"id":"0x0"}');
     });
 
+    it('calls sendPayload and returns with a rejected promise because of the connection.send() method', async () => {
+        socketMock.OPEN = 4;
+        socketMock.readyState = 4;
+        socketMock.CONNECTING = 0;
+        socketMock.send = jest.fn(() => {
+            throw new Error('Nope');
+        });
+        websocketProvider.timeout = 2;
+
+        await expect(websocketProvider.sendPayload({id: '0x0'})).rejects.toThrow('Nope');
+
+        expect(socketMock.send).toHaveBeenCalledWith('{"id":"0x0"}');
+    });
+
     it('calls sendPayload with a timeout defined and returns with a resolved promise', async () => {
         socketMock.OPEN = 4;
         socketMock.readyState = 4;