wspromisify/test/utils.ts

Wraps your WebSockets into Promise-based class with full d.ts typings on client & server

import WebSocketClient from '../src/WSC'
import {AnyFunc, AnyObject} from 'pepka'
import { native_ws } from '../src/utils'
import WS from 'ws'

export const createNew = (config = {}, port: number) => new WebSocketClient(Object.assign({
    url: 'ws://127.0.0.1:' + port,
    // log: (...a) => console.log(...a),
    adapter: (host: string, protocols?: string|string[]) => new (native_ws || WS)(host, protocols)
  }, config)
)

// Inspired by tinchoz49 https://github.com/lukeed/uvu/issues/33#issuecomment-879870292
export const timeout = (time: number, handler: AnyFunc) => async (context: AnyObject) => {
  let timer: NodeJS.Timeout
  try {
    return await Promise.race([
      handler(context),
      new Promise((_resolve, reject) =>
        timer = setTimeout(() => reject(new Error('timeout')), time)
      )
    ])
  } finally {
    if(timer!) clearTimeout(timer)
  }
}