120 lines
3.1 KiB
TypeScript
120 lines
3.1 KiB
TypeScript
import { expect, tap } from '@push.rocks/tapbundle';
|
|
import * as net from 'net';
|
|
import { PortProxy } from '../ts/smartproxy.portproxy.js';
|
|
|
|
let testServer: net.Server;
|
|
let portProxy: PortProxy;
|
|
const TEST_SERVER_PORT = 4000;
|
|
const PROXY_PORT = 4001;
|
|
const TEST_DATA = 'Hello through port proxy!';
|
|
|
|
// Helper function to create a test TCP server
|
|
function createTestServer(port: number): Promise<net.Server> {
|
|
return new Promise((resolve) => {
|
|
const server = net.createServer((socket) => {
|
|
socket.on('data', (data) => {
|
|
// Echo the received data back
|
|
socket.write(`Echo: ${data.toString()}`);
|
|
});
|
|
|
|
socket.on('error', (error) => {
|
|
console.error('[Test Server] Socket error:', error);
|
|
});
|
|
});
|
|
|
|
server.listen(port, () => {
|
|
console.log(`[Test Server] Listening on port ${port}`);
|
|
resolve(server);
|
|
});
|
|
});
|
|
}
|
|
|
|
// Helper function to create a test client connection
|
|
function createTestClient(port: number, data: string): Promise<string> {
|
|
return new Promise((resolve, reject) => {
|
|
const client = new net.Socket();
|
|
let response = '';
|
|
|
|
client.connect(port, 'localhost', () => {
|
|
console.log('[Test Client] Connected to server');
|
|
client.write(data);
|
|
});
|
|
|
|
client.on('data', (chunk) => {
|
|
response += chunk.toString();
|
|
client.end();
|
|
});
|
|
|
|
client.on('end', () => {
|
|
resolve(response);
|
|
});
|
|
|
|
client.on('error', (error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
// Setup test environment
|
|
tap.test('setup port proxy test environment', async () => {
|
|
testServer = await createTestServer(TEST_SERVER_PORT);
|
|
portProxy = new PortProxy(PROXY_PORT, TEST_SERVER_PORT);
|
|
});
|
|
|
|
tap.test('should start port proxy', async () => {
|
|
await portProxy.start();
|
|
expect(portProxy.netServer.listening).toBeTrue();
|
|
});
|
|
|
|
tap.test('should forward TCP connections and data', async () => {
|
|
const response = await createTestClient(PROXY_PORT, TEST_DATA);
|
|
expect(response).toEqual(`Echo: ${TEST_DATA}`);
|
|
});
|
|
|
|
tap.test('should handle multiple concurrent connections', async () => {
|
|
const concurrentRequests = 5;
|
|
const requests = Array(concurrentRequests).fill(null).map((_, i) =>
|
|
createTestClient(PROXY_PORT, `${TEST_DATA} ${i + 1}`)
|
|
);
|
|
|
|
const responses = await Promise.all(requests);
|
|
|
|
responses.forEach((response, i) => {
|
|
expect(response).toEqual(`Echo: ${TEST_DATA} ${i + 1}`);
|
|
});
|
|
});
|
|
|
|
tap.test('should handle connection timeouts', async () => {
|
|
const client = new net.Socket();
|
|
|
|
await new Promise<void>((resolve) => {
|
|
client.connect(PROXY_PORT, 'localhost', () => {
|
|
// Don't send any data, just wait for timeout
|
|
client.on('close', () => {
|
|
resolve();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
tap.test('should stop port proxy', async () => {
|
|
await portProxy.stop();
|
|
expect(portProxy.netServer.listening).toBeFalse();
|
|
});
|
|
|
|
// Cleanup
|
|
tap.test('cleanup port proxy test environment', async () => {
|
|
await new Promise<void>((resolve) => testServer.close(() => resolve()));
|
|
});
|
|
|
|
process.on('exit', () => {
|
|
if (testServer) {
|
|
testServer.close();
|
|
}
|
|
if (portProxy && portProxy.netServer) {
|
|
portProxy.stop();
|
|
}
|
|
});
|
|
|
|
export default tap.start();
|