1/*!
  2 * ws: a node.js websocket client
  3 * Copyright(c) 2011 Einar Otto Stangvik <einaros@gmail.com>
  4 * MIT Licensed
  5 */
  6
  7/**
  8 * Benchmark dependencies.
  9 */
 10
 11var benchmark = require('benchmark')
 12  , Receiver = require('../').Receiver
 13  , suite = new benchmark.Suite('Receiver');
 14require('tinycolor');
 15require('./util');
 16
 17/**
 18 * Setup receiver.
 19 */
 20 
 21suite.on('start', function () {
 22  receiver = new Receiver();
 23});
 24
 25suite.on('cycle', function () {
 26  receiver = new Receiver();
 27});
 28
 29/**
 30 * Benchmarks.
 31 */
 32
 33var pingMessage = 'Hello'
 34  , pingPacket1 = getBufferFromHexString('89 ' + (pack(2, 0x80 | pingMessage.length)) + 
 35                                         ' 34 83 a8 68 '+ getHexStringFromBuffer(mask(pingMessage, '34 83 a8 68')));
 36suite.add('ping message', function () {
 37  receiver.add(pingPacket1);  
 38});
 39
 40var pingPacket2 = getBufferFromHexString('89 00')
 41suite.add('ping with no data', function () {
 42  receiver.add(pingPacket2);
 43});
 44
 45var closePacket = getBufferFromHexString('88 00');
 46suite.add('close message', function () {
 47  receiver.add(closePacket);
 48  receiver.endPacket();
 49});
 50
 51var maskedTextPacket = getBufferFromHexString('81 93 34 83 a8 68 01 b9 92 52 4f a1 c6 09 59 e6 8a 52 16 e6 cb 00 5b a1 d5');
 52suite.add('masked text message', function () {
 53  receiver.add(maskedTextPacket);
 54});
 55
 56binaryDataPacket = (function() {
 57  var length = 125
 58    , message = new Buffer(length)
 59  for (var i = 0; i < length; ++i) message[i] = i % 10;
 60  return getBufferFromHexString('82 ' + getHybiLengthAsHexString(length, true) + ' 34 83 a8 68 '
 61       + getHexStringFromBuffer(mask(message), '34 83 a8 68'));
 62})();
 63suite.add('binary data (125 bytes)', function () {
 64  try {
 65    receiver.add(binaryDataPacket);
 66    
 67  }
 68  catch(e) {console.log(e)}
 69});
 70
 71binaryDataPacket2 = (function() {
 72  var length = 65535
 73    , message = new Buffer(length)
 74  for (var i = 0; i < length; ++i) message[i] = i % 10;
 75  return getBufferFromHexString('82 ' + getHybiLengthAsHexString(length, true) + ' 34 83 a8 68 '
 76       + getHexStringFromBuffer(mask(message), '34 83 a8 68'));
 77})();
 78suite.add('binary data (65535 bytes)', function () {
 79  receiver.add(binaryDataPacket2);
 80});
 81
 82binaryDataPacket3 = (function() {
 83  var length = 200*1024
 84    , message = new Buffer(length)
 85  for (var i = 0; i < length; ++i) message[i] = i % 10;
 86  return getBufferFromHexString('82 ' + getHybiLengthAsHexString(length, true) + ' 34 83 a8 68 '
 87       + getHexStringFromBuffer(mask(message), '34 83 a8 68'));
 88})();
 89suite.add('binary data (200 kB)', function () {
 90  receiver.add(binaryDataPacket3);
 91});
 92
 93/**
 94 * Output progress.
 95 */
 96
 97suite.on('cycle', function (bench, details) {
 98  console.log('\n  ' + suite.name.grey, details.name.white.bold);
 99  console.log('  ' + [
100      details.hz.toFixed(2).cyan + ' ops/sec'.grey
101    , details.count.toString().white + ' times executed'.grey
102    , 'benchmark took '.grey + details.times.elapsed.toString().white + ' sec.'.grey
103    , 
104  ].join(', '.grey));
105});
106
107/**
108 * Run/export benchmarks.
109 */
110
111if (!module.parent) {
112  suite.run();
113} else {
114  module.exports = suite;
115}