WebsocketProviderTest extended

web3 · Aug 2, 2019 · 6c14240 · 6c14240
1 parent bf33e1f
commit 6c14240
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/web3-providers/tests/src/providers/WebsocketProviderTest.js b/packages/web3-providers/tests/src/providers/WebsocketProviderTest.js
@@ -13,7 +13,7 @@ describe('WebsocketProviderTest', () => {
         socketMock.addEventListener = jest.fn();
         socketMock.removeEventListener = jest.fn();
 
-        websocketProvider = new WebsocketProvider(socketMock, 1);
+        websocketProvider = new WebsocketProvider(socketMock, 1, 1);
     });
 
     it('constructor check', () => {
@@ -22,6 +22,8 @@ describe('WebsocketProviderTest', () => {
         expect(websocketProvider.connection).toEqual(socketMock);
 
         expect(websocketProvider.timeout).toEqual(1);
+
+        expect(websocketProvider.reconnectionTimeout).toEqual(1);
 
         expect(socketMock.addEventListener.mock.calls[0][0]).toEqual('message');
         expect(socketMock.addEventListener.mock.calls[0][1]).toBeInstanceOf(Function);
@@ -445,4 +447,14 @@ describe('WebsocketProviderTest', () => {
 
         expect(websocketProvider.listenerCount('connect')).toEqual(0);
     });
+
+    it('calls onConnect after the connection got lost', (done) => {
+        websocketProvider.on('reconnected', () => {
+
+            done();
+        });
+
+        websocketProvider.reconnecting = true;
+        websocketProvider.onConnect();
+    });
 });