bbpro/src/services/instance/pptr-console-server/websocket_chat_app_homework/public/tests/messageTest.js

BrowserBox - remote browser isolation tool

import {sleep} from './helpers.js';

// send TOTAL messages
export default async function run(TOTAL) {
  const MESSAGE_CONTENT = 'Hello there you have a wonderful beautiful day.'
  let received = 0;
  let resolve;

  console.log('Running message tests');

  const testSocket = new WebSocket(`wss://${location.host}`);

  const p = new Promise(res => resolve = res);

  testSocket.onmessage = m => {
    m = JSON.parse(m.data);
    if( m.message == MESSAGE_CONTENT ) {
      received++;
      if ( received == TOTAL ) {
        resolve();
      }
    }
  };

  console.log(`Sending ${TOTAL} messages...`);

  testSocket.onopen = async () => {
    // send messages
    for( let i = 0; i < TOTAL; i++ ) {
      newMessage(); 
      await sleep(10);
    }
  };

  await Promise.race([sleep(10000), p]);

  testSocket.close();

  await sleep(1000);

  console.assert(received == TOTAL, `Expected ${TOTAL} messages, but ${received} received.`);

  console.log('Completed message tests');

  return received == TOTAL;

  function newMessage() {
    testSocket.send(JSON.stringify({message:MESSAGE_CONTENT}));
  }
}