mynode/packages/utils/ipc.test.ts

import { readFile } from 'fs/promises';
import { createTempDir } from './fs.test';
import { newHttpError, sendMessage, startServer, stopServer } from './ipc';

const getRandomPort = () => {
	return Math.floor(Math.random() * (65535 - 20000 + 1)) + 20000;
};

describe('ipc', () => {

	it('should send and receive messages', async () => {
		const tempDir = await createTempDir();
		const secretFilePath = `${tempDir}/secret.txt`;
		const serverPort1 = getRandomPort();
		const serverPort2 = serverPort1 + 5;

		const server1 = await startServer(serverPort1, secretFilePath, async (request) => {
			if (request.action === 'testing') {
				return {
					text: 'hello1',
				};
			}

			throw newHttpError(404);
		});

		const server2 = await startServer(serverPort2, secretFilePath, async (request) => {
			if (request.action === 'testing') {
				return {
					text: 'hello2',
				};
			}

			if (request.action === 'ping') {
				return {
					text: 'pong',
				};
			}

			throw newHttpError(404);
		});

		const secretKey = await readFile(secretFilePath, 'utf-8');

		{
			const responses = await sendMessage(serverPort1, {
				action: 'testing',
				data: {
					test: 1234,
				},
				secretKey,
			});

			expect(responses).toEqual([
				{ port: serverPort1, response: { text: 'hello1' } },
				{ port: serverPort2, response: { text: 'hello2' } },
			]);
		}

		{
			const responses = await sendMessage(serverPort1, {
				action: 'ping',
				data: null,
				secretKey,
			});

			expect(responses).toEqual([
				{ port: serverPort2, response: { text: 'pong' } },
			]);
		}

		{
			const responses = await sendMessage(serverPort1, {
				action: 'testing',
				data: {
					test: 1234,
				},
				sourcePort: serverPort1,
				secretKey,
			});

			expect(responses).toEqual([
				{ port: serverPort2, response: { text: 'hello2' } },
			]);
		}

		await stopServer(server1);
		await stopServer(server2);
	});

	it('should not process message if secret is invalid', async () => {
		const tempDir = await createTempDir();
		const secretFilePath = `${tempDir}/secret.txt`;
		const serverPort = getRandomPort();

		const server = await startServer(serverPort, secretFilePath, async (request) => {
			if (request.action === 'testing') {
				return {
					text: 'hello1',
				};
			}

			throw newHttpError(404);
		});

		const secretKey = await readFile(secretFilePath, 'utf-8');

		{
			const responses = await sendMessage(serverPort, {
				action: 'testing',
				data: null,
				secretKey: 'wrong_key',
			});

			expect(responses.length).toBe(0);
		}

		{
			const responses = await sendMessage(serverPort, {
				action: 'testing',
				data: null,
				secretKey,
			});

			expect(responses.length).toBe(1);
		}

		await stopServer(server);
	});

});