server/test/websocket.js

import http from 'http';
import url from 'url';
import assert from 'assert';
import localtunnel from 'localtunnel';
import WebSocket from 'ws';
import { Server as WebSocketServer } from 'ws';

import CreateServer from '../server';

const localtunnel_server = CreateServer({
    max_tcp_sockets: 2
});

var lt_server_port

suite('websocket');

before('set up localtunnel server', function(done) {
    var server = localtunnel_server.listen(function() {
        lt_server_port = server.address().port;
        done();
    });
});

before('set up local websocket server', function(done) {
    var wss = new WebSocketServer({ port: 0 }, function() {
        test._fake_port = wss._server.address().port;
        done();
    });

    wss.on('error', function(err) {
        done(err);
    });

    wss.on('connection', function connection(ws) {
        ws.on('error', function(err) {
            done(err);
        });

        ws.on('message', function incoming(message) {
            ws.send(message);
        });
    });
});

before('set up localtunnel client', function(done) {
    var opt = {
        host: 'http://localhost:' + lt_server_port,
    };

    localtunnel(test._fake_port, opt, function(err, tunnel) {
        assert.ifError(err);
        var url = tunnel.url;
        assert.ok(new RegExp('^http:\/\/.*localhost:' + lt_server_port + '$').test(url));
        test._fake_url = url;
        done(err);
    });
});

test('websocket server request', function(done) {
    var hostname = url.parse(test._fake_url).hostname;
    var ws = new WebSocket('http://localhost:' + lt_server_port, {
        headers: {
            host: hostname + '.tld'
        }
    });

    ws.on('message', function(msg) {
        assert.equal(msg, 'something');
        done();
    });

    ws.on('open', function open() {
        ws.send('something');
    });
});