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', () => {

  return 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 = await 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()

    })

  })

})