refactor to use http agent interface for tunnels

server.test.js

@@ -0,0 +1,85 @@
+import request from 'supertest';
+import assert from 'assert';
+import { Server as WebSocketServer } from 'ws';
+import WebSocket from 'ws';
+import net from 'net';
+
+import createServer from './server';
+
+describe('Server', () => {
+    it('server starts and stops', async () => {
+        const server = createServer();
+        await new Promise(resolve => server.listen(resolve));
+        await new Promise(resolve => server.close(resolve));
+    });
+
+    it('should redirect root requests to landing page', async () => {
+        const server = createServer();
+        const res = await request(server).get('/');
+        assert.equal('https://localtunnel.github.io/www/', res.headers.location);
+    });
+
+    it('should support custom base domains', async () => {
+        const server = createServer({
+            domain: 'domain.example.com',
+        });
+
+        const res = await request(server).get('/');
+        assert.equal('https://localtunnel.github.io/www/', res.headers.location);
+    });
+
+    it('reject long domain name requests', async () => {
+        const server = createServer();
+        const res = await request(server).get('/thisdomainisoutsidethesizeofwhatweallowwhichissixtythreecharacters');
+        assert.equal(res.body.message, 'Invalid subdomain. Subdomains must be lowercase and between 4 and 63 alphanumeric characters.');
+    });
+
+    it('should upgrade websocket requests', async () => {
+        const hostname = 'websocket-test';
+        const server = createServer({
+            domain: 'example.com',
+        });
+        await new Promise(resolve => server.listen(resolve));
+
+        const res = await request(server).get('/websocket-test');
+        const localTunnelPort = res.body.port;
+
+        const wss = await new Promise((resolve) => {
+            const wsServer = new WebSocketServer({ port: 0 }, () => {
+                resolve(wsServer);
+            });
+        });
+
+        const websocketServerPort = wss.address().port;
+
+        const ltSocket = net.createConnection({ port: localTunnelPort });
+        const wsSocket = net.createConnection({ port: websocketServerPort });
+        ltSocket.pipe(wsSocket).pipe(ltSocket);
+
+        wss.once('connection', (ws) => {
+            ws.once('message', (message) => {
+                ws.send(message);
+            });
+        });
+
+        const ws = new WebSocket('http://localhost:' + server.address().port, {
+            headers: {
+                host: hostname + '.example.com',
+            }
+        });
+
+        ws.on('open', () => {
+            ws.send('something');
+        });
+
+        await new Promise((resolve) => {
+            ws.once('message', (msg) => {
+                assert.equal(msg, 'something');
+                resolve();
+            });
+        });
+
+        wss.close();
+        await new Promise(resolve => server.close(resolve));
+    });
+});