Function testWebsocket

test/spec_rule/no_rule_websocket_spec.js:16–128

(protocol, masked = false)

testWebsocket('wss', true);

function testWebsocket(protocol, masked = false) {
  describe('Test WebSocket in protocol : ' + protocol, () => {
    const url = generateWsUrl(protocol, '/test/socket');
    let serverInstance;
    let proxyServer;
    // the message to
    const testMessageArray = [
      'Send the message with default option1',
      'Send the message with default option2',
      'Send the message with default option3',
      'Send the message with default option4'
    ];

    beforeAll((done) => {
      jasmine.DEFAULT_TIMEOUT_INTERVAL = 200000;
      printLog('Start server for no_rule_websocket_spec');
      serverInstance = new Server();

      proxyServer = ProxyServerUtil.defaultProxyServer();

      setTimeout(() => {
        done();
      }, 2000);
    });

    afterAll(() => {
      serverInstance && serverInstance.close();
      proxyServer && proxyServer.close();
      printLog('Closed server for no_rule_websocket_spec');
    });

    it('Default websocket option', done => {
      const directMessages = []; // set the flag for direct message, compare when both direct and proxy got message
      const proxyMessages = [];
      let directHeaders;
      let proxyHeaders;

      const ws = directWs(url);
      const proxyWsRef = proxyWs(url);
      ws.on('open', () => {
        ws.send(testMessageArray[0], masked);
        for (let i = 1; i < testMessageArray.length; i++) {
          setTimeout(() => {
            ws.send(testMessageArray[i], masked);
          }, 1000);
        }
      });

      proxyWsRef.on('open', () => {
        try {
          proxyWsRef.send(testMessageArray[0], masked);
          for (let i = 1; i < testMessageArray.length; i++) {
            setTimeout(() => {
              proxyWsRef.send(testMessageArray[i], masked);
            }, 1000);
          }
        } catch (e) {
          console.error(e);