var vows = require('vows');
var assert = require('assert');
var events = require('events');
var WebSocketClient = require('websocket').client;


vows.describe('WebSocket-Node Client Connection Test').addBatch({
  'HTTP': {
    topic: function() {
      var promise = new (events.EventEmitter);
      var client = new WebSocketClient();

      client.on('connect', function(connection) {
        connection.sendUTF("foobar");

        connection.on('message', function(message) {
          promise.emit('success', message);
          connection.close();
        });
      });

      client.connect("ws://localhost:8080");

      return promise;
    },
    'should receive message': function(message) {
      assert.equal(message.utf8Data, 'foobar');
    }
  },

  'HTTPS': {
    topic: function() {
      var promise = new (events.EventEmitter);
      var client = new WebSocketClient();

      client.on('connect', function(connection) {
        connection.sendUTF("foobar");

        connection.on('message', function(message) {
          promise.emit('success', message);
          connection.close();
        });
      });

      client.connect("wss://localhost:8081");

      return promise;
    },
    'should receive message': function(message) {
      assert.equal(message.utf8Data, 'foobar');
    }
  }
}).export(module);