add unit tests for RippleAPIBroadcast

test/setup-api.js

@@ -1,52 +1,77 @@
 'use strict';
 const net = require('net');
 const RippleAPI = require('ripple-api').RippleAPI;
+const RippleAPIBroadcast = require('ripple-api').RippleAPIBroadcast;
 const ledgerClosed = require('./fixtures/rippled/ledger-close');
 const createMockRippled = require('./mock-rippled');
 
 // using a free port instead of a constant port enables parallelization
-function getFreePort(callback) {
-  const server = net.createServer();
-  let port;
-  server.on('listening', function() {
-    port = server.address().port;
-    server.close();
+function getFreePort() {
+  return new Promise((resolve, reject) => {
+    const server = net.createServer();
+    let port;
+    server.on('listening', function() {
+      port = server.address().port;
+      server.close();
+    });
+    server.on('close', function() {
+      resolve(port);
+    });
+    server.on('error', function(error) {
+      reject(error);
+    });
+    server.listen(0);
   });
-  server.on('close', function() {
-    callback(null, port);
-  });
-  server.on('error', function(error) {
-    callback(error);
-  });
-  server.listen(0);
 }
 
-function setupMockRippledConnection(testcase, port, done) {
-  testcase.mockRippled = createMockRippled(port);
-  testcase.api = new RippleAPI({server: 'ws://localhost:' + port});
-  testcase.api.connect().then(() => {
-    testcase.api.once('ledger', () => done());
-    testcase.api.connection._ws.emit('message', JSON.stringify(ledgerClosed));
-  }).catch(done);
+function setupMockRippledConnection(testcase, port) {
+  return new Promise((resolve, reject) => {
+    testcase.mockRippled = createMockRippled(port);
+    testcase.api = new RippleAPI({server: 'ws://localhost:' + port});
+    testcase.api.connect().then(() => {
+      testcase.api.once('ledger', () => resolve());
+      testcase.api.connection._ws.emit('message', JSON.stringify(ledgerClosed));
+    }).catch(reject);
+  });
 }
 
-function setup(done) {
-  getFreePort((error, port) => {
-    if (error) {
-      throw new Error('Unable to obtain a free port: ' + error);
-    }
-    setupMockRippledConnection(this, port, done);
+function setupMockRippledConnectionForBroadcast(testcase, ports) {
+  return new Promise((resolve, reject) => {
+    const servers = ports.map(port => 'ws://localhost:' + port);
+    testcase.mocks = ports.map(port => createMockRippled(port));
+    testcase.api = new RippleAPIBroadcast(servers);
+    testcase.api.connect().then(() => {
+      testcase.api.once('ledger', () => resolve());
+      testcase.mocks[0].socket.send(JSON.stringify(ledgerClosed));
+    }).catch(reject);
+  });
+}
+
+function setup() {
+  return getFreePort().then(port => {
+    return setupMockRippledConnection(this, port);
+  });
+}
+
+function setupBroadcast() {
+  return Promise.all([getFreePort(), getFreePort()]).then(ports => {
+    return setupMockRippledConnectionForBroadcast(this, ports);
   });
 }
 
 function teardown(done) {
   this.api.disconnect().then(() => {
-    this.mockRippled.close();
+    if (this.mockRippled !== undefined) {
+      this.mockRippled.close();
+    } else {
+      this.mocks.forEach(mock => mock.close());
+    }
     setImmediate(done);
   }).catch(done);
 }
 
 module.exports = {
   setup: setup,
-  teardown: teardown
+  teardown: teardown,
+  setupBroadcast: setupBroadcast
 };