Add setWebSocket method

lib/api.js

@@ -79,6 +79,13 @@ var Steem = function (_EventEmitter) {
   }
 
   _createClass(Steem, [{
+    key: 'setWebSocket',
+    value: function setWebSocket(url) {
+      this.options.url = url;
+      this.stop();
+      this.start();
+    }
+  }, {
     key: 'start',
     value: function start() {
       var _this2 = this;
@@ -106,7 +113,6 @@ var Steem = function (_EventEmitter) {
       this.releases.forEach(function (release) {
         return release();
       });
-      this.ws.removeEventListener();
       this.ws.close();
       delete this.ws;
       delete this.releases;
@@ -326,4 +332,5 @@ _bluebird2.default.promisifyAll(Steem.prototype);
 // Export singleton instance
 var steem = new Steem();
 exports = module.exports = steem;
-exports.Steem = Steem;
+exports.Steem = Steem;
\ No newline at end of file
+exports.Steem.DEFAULTS = DEFAULTS;
\ No newline at end of file

src/api.js

@@ -44,6 +44,12 @@ class Steem extends EventEmitter {
     this.start();
   }
 
+  setWebSocket(url) {
+    this.options.url = url;
+    this.stop();
+    this.start();
+  }
+
   start() {
     this.startP = new Promise((resolve /* , reject*/) => {
       this.ws = new WebSocket(this.options.url);
@@ -69,7 +75,6 @@ class Steem extends EventEmitter {
 
   stop() {
     this.releases.forEach((release) => release());
-    this.ws.removeEventListener();
     this.ws.close();
     delete this.ws;
     delete this.releases;
@@ -272,3 +277,4 @@ Promise.promisifyAll(Steem.prototype);
 const steem = new Steem();
 exports = module.exports = steem;
 exports.Steem = Steem;
+exports.Steem.DEFAULTS = DEFAULTS;

test/api.test.js

@@ -12,6 +12,13 @@ describe('steem', function () {
     await steem.apiIdsP;
   });
 
+  describe('setWebSocket', () => {
+    it('works', () => {
+      steem.setWebSocket('ws://localhost');
+      steem.setWebSocket(steem.Steem.DEFAULTS.url);
+    });
+  });
+
   describe('getFollowers', () => {
     describe('getting ned\'s followers', () => {
       it('works', async () => {