wspromisify/test/mock/server.ts

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

import { WebSocketServer } from 'ws'
import { createServer, killServer } from './WS'

let port: number,
    server: WebSocketServer|null

export default async (new_port?: number) => {
  if(!server) {
    port = new_port || 8000 + Math.ceil(Math.random()*500)
    server = await createServer(port)
  }
  return {
    server, port,
    isRunning() {return Boolean(server)},
    shutDown: async () => { await killServer(); server=null }
  }
}