74 lines
3.6 KiB
JavaScript
74 lines
3.6 KiB
JavaScript
import net from 'net';
|
|
import { describe, expect, it } from 'vitest';
|
|
import { ClientManager } from './ClientManager.js';
|
|
describe('ClientManager', () => {
|
|
it('should construct with no tunnels', () => {
|
|
const manager = new ClientManager();
|
|
expect(manager.stats.tunnels).toBe(0);
|
|
});
|
|
it('should create a new client with random id', async () => {
|
|
const manager = new ClientManager();
|
|
const client = await manager.newClient(undefined, 'http://test.example.com', 'example.com');
|
|
expect(manager.hasClient(client.id)).toBe(true);
|
|
manager.removeClient(client.id);
|
|
});
|
|
it('should create a new client with id', async () => {
|
|
const manager = new ClientManager();
|
|
await manager.newClient('foobar', 'http://foobar.example.com', 'example.com');
|
|
expect(manager.hasClient('foobar')).toBe(true);
|
|
manager.removeClient('foobar');
|
|
});
|
|
it('should create a new client with random id if previous exists', async () => {
|
|
const manager = new ClientManager();
|
|
const clientA = await manager.newClient('foobar', 'http://foobar.example.com', 'example.com');
|
|
const clientB = await manager.newClient('foobar', 'http://foobar.example.com', 'example.com');
|
|
expect(clientA.id).toBe('foobar');
|
|
expect(manager.hasClient(clientB.id)).toBe(true);
|
|
expect(clientB.id).not.toBe(clientA.id);
|
|
manager.removeClient(clientB.id);
|
|
manager.removeClient('foobar');
|
|
});
|
|
it('should remove client once it goes offline', async () => {
|
|
const manager = new ClientManager();
|
|
const client = await manager.newClient('foobar', 'http://foobar.example.com', 'example.com');
|
|
const socket = await new Promise((resolve) => {
|
|
const netClient = net.createConnection({ port: client.port }, () => {
|
|
resolve(netClient);
|
|
});
|
|
});
|
|
const closePromise = new Promise(resolve => socket.once('close', resolve));
|
|
socket.end();
|
|
await closePromise;
|
|
// should still have client - grace period has not expired
|
|
expect(manager.hasClient('foobar')).toBe(true);
|
|
// wait past grace period (1s)
|
|
await new Promise(resolve => setTimeout(resolve, 1500));
|
|
expect(manager.hasClient('foobar')).toBe(false);
|
|
}, 5000);
|
|
it('should remove correct client once it goes offline', async () => {
|
|
const manager = new ClientManager();
|
|
const clientFoo = await manager.newClient('foo', 'http://foo.example.com', 'example.com');
|
|
await manager.newClient('bar', 'http://bar.example.com', 'example.com');
|
|
const socket = await new Promise((resolve) => {
|
|
const netClient = net.createConnection({ port: clientFoo.port }, () => {
|
|
resolve(netClient);
|
|
});
|
|
});
|
|
await new Promise(resolve => setTimeout(resolve, 1500));
|
|
// foo should still be ok
|
|
expect(manager.hasClient('foo')).toBe(true);
|
|
// clientBar should be removed - nothing connected to it
|
|
expect(manager.hasClient('bar')).toBe(false);
|
|
manager.removeClient('foo');
|
|
socket.end();
|
|
}, 5000);
|
|
it('should remove clients if they do not connect within 5 seconds', async () => {
|
|
const manager = new ClientManager();
|
|
await manager.newClient('foo', 'http://foo.example.com', 'example.com');
|
|
expect(manager.hasClient('foo')).toBe(true);
|
|
// wait past grace period (1s)
|
|
await new Promise(resolve => setTimeout(resolve, 1500));
|
|
expect(manager.hasClient('foo')).toBe(false);
|
|
}, 5000);
|
|
});
|
|
//# sourceMappingURL=ClientManager.spec.js.map
|