Integration test with socket.io

testSocket.js

const assert = require('chai').assert;
const socket = require('socket.io-client')("http://localhost:3000");


/**
 * function used to wait response on a channel
 * 
 * @param {String} channel 
 * @param {Function} callback 
 */
const waitMessage = async (channel, socket, callback) => {
  return new Promise((resolve, reject) => {
    try {
      socket.on(channel, async (data) => {
        await callback(data);
        resolve(data);
      });
    } catch (err) {
      reject(err)
    }
  });
}



describe('Test Connection', () => {
  
  it("Should connect, send ping and receive pong", () => {
    return new Promise((resolve, reject) => {
      socket.emit("pingMsg", "ping");
      
      // wait for response
      await waitMessage("pongMsg", socket, (data) => {
        try {
          assert.equal(data, "pong");
          socket.removeAllListeners(["pongMsg"]);  // remove socket listener
          resolve("done");
        } catch (e) {
          reject(e);
        }
      }); 
      
    });
    
  });
  
});