106 lines
2.7 KiB
JavaScript
106 lines
2.7 KiB
JavaScript
var http = require('http');
|
|
var url = require('url');
|
|
var assert = require('assert');
|
|
var localtunnel = require('localtunnel');
|
|
var WebSocket = require('ws');
|
|
var WebSocketServer = require('ws').Server;
|
|
|
|
var localtunnel_server = require('../server')({
|
|
max_tcp_sockets: 2,
|
|
subHost: ['sub']
|
|
});
|
|
|
|
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');
|
|
});
|
|
});
|
|
test('set up localtunnel client (sub host)', function (done) {
|
|
var opt = {
|
|
host: 'http://sub.localhost:' + lt_server_port
|
|
};
|
|
|
|
localtunnel(test._fake_port, opt, function (err, tunnel) {
|
|
assert.ifError(err);
|
|
var url = tunnel.url;
|
|
assert.ok(new RegExp('^http:\/\/.*sub.localhost:' + lt_server_port + '$').test(url));
|
|
test._fake_url = url;
|
|
done(err);
|
|
});
|
|
});
|
|
|
|
test('websocket server request (sub-host)', function (done) {
|
|
var hostname = url.parse(test._fake_url).hostname;
|
|
var ws = new WebSocket('http://sub.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');
|
|
});
|
|
}); |