File Explorer

1var test = require('tape')2var websocket = require('./')3var echo = require("./echo-server")4var WebSocketServer = require('ws').Server5var http = require('http')6var concat = require('concat-stream')7var Buffer = require('safe-buffer').Buffer8var ts = require('typescript')9 10test('echo server', function(t) {11 12  echo.start(function() {13    var client = websocket(echo.url)14 15    client.on('error', console.error)16 17    client.on('data', function(data) {18      t.ok(Buffer.isBuffer(data), 'is a buffer')19      t.equal(data.toString(), 'hello world')20      client.end()21      echo.stop(function() {22        t.end()23      })24    })25 26    client.write('hello world')27  })28 29})30 31test('emitting not connected errors', function(t) {32 33  echo.start(function() {34    var client = websocket(echo.url)35 36    client.on('error', function() {37      echo.stop(function() {38        t.true(true, 'should emit error')39        t.end()40      })41    })42 43    client.once('data', function(data) {44      client.end()45      client.write('abcde')46    })47 48    client.write('hello world')49  })50 51})52 53test('passes options to websocket constructor', function(t) {54  t.plan(3)55 56  opts = {57    verifyClient: function verifyClient(info) {58      t.equal(info.req.headers['x-custom-header'], 'Custom Value')59      return true60    }61  }62  echo.start(opts, function() {63    var options = {headers: {'x-custom-header': 'Custom Value'}}64    var client = websocket(echo.url, options)65 66    client.on('error', console.error)67 68    client.on('data', function(data) {69      t.ok(Buffer.isBuffer(data), 'is a buffer')70      t.equal(data.toString(), 'hello world')71      client.end()72      echo.stop(function() {})73    })74 75    client.write('hello world')76  })77 78})79 80 81test('destroy', function(t) {82  t.plan(1)83 84  echo.start(function() {85    var client = websocket(echo.url, echo.options)86 87    client.on('close', function() {88      echo.stop(function() {89        t.pass('destroyed')90      })91    })92 93    setTimeout(function() {94      client.destroy()95    }, 200)96  })97 98})99 100test('drain', function(t) {101  t.plan(1)102 103  echo.start(function() {104    var client = websocket(echo.url, echo.options)105 106    client.on('drain', function() {107      client.destroy()108      echo.stop(function() {109        t.pass('drained')110      })111    })112 113    // write until buffer is full114    while (client.write('foobar')) {}115  })116 117})118 119test('emit sending errors if the socket is closed by the other party', function(t) {120 121  var server = http.createServer()122  var wss = new WebSocketServer({ server: server })123 124  server.listen(8344, function() {125    var client = websocket('ws://localhost:8344')126 127    wss.on('connection', function(ws) {128      var stream = websocket(ws)129 130      client.destroy()131 132      setTimeout(function() {133        stream.write('hello world')134      }, 50)135 136      stream.on('error', function(err) {137        t.ok(err, 'client errors')138        server.close(t.end.bind(t))139      })140    })141  })142})143 144test('destroy client pipe should close server pipe', function(t) {145  t.plan(1)146 147  var clientDestroy = function() {148    var client = websocket(echo.url, echo.options)149    client.on('data', function(o) {150      client.destroy()151    })152    client.write(Buffer.from('hello'))153  }154 155  var opts = {}156  var server = http.createServer()157  opts.server = server158  var wss = new WebSocketServer(opts)159  wss.on('connection', function(ws) {160    var stream = websocket(ws)161    stream.on('close', function() {162      server.close(function() {163        t.pass('close is called')164      })165    })166    stream.pipe(stream)167  })168  server.listen(echo.port, clientDestroy)169})170 171 172test('error on socket should forward it to pipe', function(t) {173  t.plan(1)174 175  var clientConnect = function() {176    websocket(echo.url, echo.options)177  }178 179  var opts = {}180  var server = http.createServer()181  opts.server = server182  var wss = new WebSocketServer(opts)183  wss.on('connection', function(ws) {184    var stream = websocket(ws)185    stream.on('error', function() {186      server.close(function() {187        t.pass('error is called')188      })189    })190    stream.socket.emit('error', new Error('Fake error'))191  })192  server.listen(echo.port, clientConnect)193})194 195test('stream end', function(t) {196  t.plan(1)197 198  var server = http.createServer()199  websocket.createServer({ server: server }, handle)200 201  function handle (stream) {202    stream.pipe(concat(function (body) {203      t.equal(body.toString(), 'pizza cats\n')204      server.close()205    }))206  }207  server.listen(0, function () {208    var w = websocket('ws://localhost:' + server.address().port)209    w.end('pizza cats\n')210  })211})212 213test('stream handlers should fire once per connection', function(t) {214  t.plan(2)215 216  var server = http.createServer()217  var wss = websocket.createServer({ server: server }, function() {218    server.close(function() {219      t.equal(m, 1)220    })221  })222 223  var m = 0224  wss.on('stream', function(stream, request) {225    t.ok(request instanceof http.IncomingMessage)226    m++227  })228  server.listen(0, function() {229    var w = websocket('ws://localhost:' + server.address().port)230    w.end('pizza cats\n')231  })232})233 234test('client with writev', function(t) {235  var server = http.createServer()236 237  var str = ''238  var wss = websocket.createServer({239    server: server240  }, function (stream) {241    stream.once('data', function(data) {242      t.ok(Buffer.isBuffer(data), 'is a buffer')243      t.equal(data.toString(), 'hello world')244 245      stream.once('data', function(data) {246        t.ok(Buffer.isBuffer(data), 'is a buffer')247        t.equal(data.toString(), str)248        stream.end()249        server.close()250        t.end()251      })252    })253  })254 255  server.listen(8352, function () {256    var client = websocket('ws://localhost:8352', {257      objectMode: false258    })259 260    client.on('error', console.error)261 262    client.once('connect', function () {263      client.cork()264      do {265        str += 'foobar'266      } while (client.write('foobar'))267      client.uncork()268    })269 270    client.write('hello world')271  })272})273 274test('server with writev', function(t) {275  var server = http.createServer()276 277  var str = ''278  var wss = websocket.createServer({279    server: server,280    objectMode: false281  }, function (stream) {282    stream.cork()283    do {284      str += 'foobar'285    } while (stream.write('foobar'))286    stream.uncork()287  })288 289  server.listen(8352, function () {290    var client = websocket('ws://localhost:8352')291 292    client.on('error', console.error)293 294    client.once('data', function(data) {295      t.ok(Buffer.isBuffer(data), 'is a buffer')296      t.equal(data.toString(), str)297      client.end()298      server.close()299      t.end()300    })301  })302})303 304test('stop echo', function(t) {305  echo.stop(function() {306    t.end()307  })308})309 310test('typescript compilation', function(t) {311  function compile(fileNames, options) {312    const program = ts.createProgram(fileNames, options)313    const emitResult = program.emit()314 315    const allDiagnostics = ts.getPreEmitDiagnostics(program).concat(emitResult.diagnostics)316 317    return allDiagnostics.map(diagnostic => {318      if (diagnostic.file) {319        let { line, character } = diagnostic.file.getLineAndCharacterOfPosition(diagnostic.start)320        let message = ts.flattenDiagnosticMessageText(diagnostic.messageText, "\n")321        return `${diagnostic.file.fileName} (${line + 1},${character + 1}): ${message}`322      } else {323        return `${ts.flattenDiagnosticMessageText(diagnostic.messageText, "\n")}`324      }325    });326  }327 328  const messages = compile(['./ts-tests.ts'], {329    noEmit: true,330    noImplicitAny: true,331    target: ts.ScriptTarget.ES5,332    module: ts.ModuleKind.CommonJS333  })334 335  t.equal(messages.length, 0, 'no errors emitted')336  t.end()337})338 339