Fix tests

openethereum · Dec 20, 2016 · 66ee764 · 66ee764
1 parent 2bc8786
commit 66ee764
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/js/src/api/transport/ws/ws.js b/js/src/api/transport/ws/ws.js
@@ -22,7 +22,7 @@ import TransportError from '../error';
 
 /* global WebSocket */
 export default class Ws extends JsonRpcBase {
-  constructor (url, token) {
+  constructor (url, token, connect = true) {
     super();
 
     this._url = url;
@@ -39,14 +39,18 @@ export default class Ws extends JsonRpcBase {
     this._connectPromise = null;
     this._connectPromiseFunctions = {};
 
-    // this.connect();
+    if (connect) {
+      this.connect();
+    }
   }
 
-  updateToken (token) {
+  updateToken (token, connect = true) {
     this._token = token;
     // this._autoConnect = true;
 
-    // this.connect();
+    if (connect) {
+      this.connect();
+    }
   }
 
   connect () {

diff --git a/js/src/secureApi.js b/js/src/secureApi.js
@@ -20,7 +20,7 @@ const sysuiToken = window.localStorage.getItem('sysuiToken');
 
 export default class SecureApi extends Api {
   constructor (url, nextToken) {
-    super(new Api.Transport.Ws(url, sysuiToken));
+    super(new Api.Transport.Ws(url, sysuiToken, false));
 
     this._url = url;
     this._isConnecting = true;
@@ -140,7 +140,7 @@ export default class SecureApi extends Api {
   }
 
   updateToken (token) {
-    this._transport.updateToken(token.replace(/[^a-zA-Z0-9]/g, ''));
+    this._transport.updateToken(token.replace(/[^a-zA-Z0-9]/g, ''), false);
     return this._followConnection();
     // DEBUG: console.log('SecureApi:updateToken', this._transport.token, connectState);
   }