react-use-websocket/src/lib/get-url.test.ts

React Hook for WebSocket communication

import { MutableRefObject } from 'react';
import { getUrl } from './get-url';
import { Options } from './types';

let optionRef: MutableRefObject<Options>;

const URL = 'ws://localhost:1234';
const SOCKET_IO_URL = 'http://localhost:1234';

const getUrlAsync = () => {
  return new Promise<string>(resolve => {
    window.setTimeout(() => {
      resolve(URL);
    }, 500);
  });
};

beforeEach(() => {
  optionRef = { current: {} };
});

test('If passed a funtion, it will return a promise that resolves to the return value of the function, after being parsed', async (done) => {
  const wsUrl = await getUrl(getUrlAsync, optionRef);
  expect(wsUrl).toEqual(URL);
  done();
});

test('If fromSocketIO is passed as an option, socketIO protocol will be used', async (done) => {
  optionRef.current.fromSocketIO = true;

  const wsUrl = await getUrl(SOCKET_IO_URL, optionRef);
  expect(wsUrl.endsWith('socket.io/?EIO=3&transport=websocket')).toBe(true);
  done();
});

test('If query params are passed in the options, object will be converted to stringified params and appended to url', async (done) => {
  optionRef.current.queryParams = { type: 'user', id: 5 };

  const wsUrl = await getUrl(URL, optionRef);
  expect(wsUrl.endsWith('?type=user&id=5')).toBe(true);
  done();
});