houd1ni / WebsocketPromisify

import { createNew } from '../utils'

import mockServer from '../mock/server'

import WS from 'ws'

import { test } from '../suite'

import { equals } from 'pepka'

const addr = (port: number) => 'ws://localhost:' + port

/** If an existing socket connection is provided via config. */

test('existing_socket', () => new Promise(async (ff, rj) => {

  const {port} = await mockServer()

  const to = setTimeout(() => rj(), 4e4)

  const existing_addr = addr(port)

  // This one CANNOT connect as fast as we send to it,

  // So readyState is 0.

  const ws1 = createNew({socket: new WS(existing_addr)}, port)

  if(ws1.socket?.readyState !== 0) return rj('not ready.')

  const msg1 = {echo: true, msg: 'existing_socket!'}

  const response1 = await ws1.send(msg1)

  if(

    ws1.socket?.readyState as number !== 1

    || !equals(response1, msg1)

  ) return rj('not ready.')

  await ws1.close()

  // This one DO CAN connect as fast as we send to it,

  // So readyState should be 1.

  const ws2_0 = new WS(existing_addr)

  ws2_0.addEventListener('open', async () => {

    const ws2 = createNew({socket: ws2_0}, port)

    if(ws2.socket?.readyState !== 1) return rj('not ready.')

    const msg2 = {echo: true, msg: 'existing_socket!'}

    const response2 = await ws2.send(msg2)

    if(

      ws2.socket?.readyState as number !== 1

      || !equals(response2, msg2)

    ) return rj('not ready.')

    await ws2.close()

    clearTimeout(to)

    ff()

  })

}))