houd1ni / WebsocketPromisify

import { createNew } from '../utils'

import mockServer from '../mock/server'

import { test } from '../suite'

/** Rejects messages by timout */

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

  const {port, shutDown} = await mockServer()

  const ws = createNew({timeout: 500}, port)

  await shutDown()

  const lauchServ = async () => await mockServer(port)

  setTimeout(async () => {

    const msg = {echo: true, msg: 'hello!'}

    let to: NodeJS.Timeout

    try {

      to = setTimeout(() => {

        return rj()

      }, 600)

      await ws.send(msg)

      if(to) clearTimeout(to)

      await lauchServ()

      return rj()

    } catch(e) {

      if(to!) clearTimeout(to)

      await lauchServ()

      return ff()

    }

  }, 200)

}))