feat: Initial clean release of CHM

packages/app-desktop/utils/customProtocols/handleCustomProtocols.plugins.test.ts

@@ -0,0 +1,88 @@
+/** @jest-environment node */
+
+type ProtocolHandler = (request: Request)=> Promise<Response>;
+const customProtocols: Map<string, ProtocolHandler> = new Map();
+
+const handleProtocolMock = jest.fn((scheme: string, handler: ProtocolHandler) => {
+	customProtocols.set(scheme, handler);
+});
+
+jest.doMock('electron', () => {
+	return {
+		protocol: {
+			handle: handleProtocolMock,
+		},
+		Session: jest.fn(),
+	};
+});
+
+import handleCustomProtocols from './handleCustomProtocols';
+
+const setUpProtocolHandler = () => {
+	const mockSession = {
+		protocol: {
+			handle: handleProtocolMock,
+		},
+	};
+	const protocolHandler = handleCustomProtocols(mockSession as never);
+
+	expect(handleProtocolMock).toHaveBeenCalled();
+
+	let onRequestListener;
+	for (const call of handleProtocolMock.mock.calls) {
+		if (call[0] === 'joplin-plugin') {
+			onRequestListener = call[1];
+		}
+	}
+
+	// Should have registered the protocol
+	expect(onRequestListener).toBeDefined();
+
+	return {
+		protocolHandler: protocolHandler.pluginContent,
+		onRequestListener,
+	};
+};
+
+
+describe('handleCustomProtocols.plugins', () => {
+	beforeEach(() => {
+		// Reset mocks between tests to ensure a clean testing environment.
+		customProtocols.clear();
+		handleProtocolMock.mockClear();
+	});
+
+	test('should return a 404 error response for unregistered content script paths', async () => {
+		const { onRequestListener } = setUpProtocolHandler();
+
+		const response = await onRequestListener(new Request('joplin-plugin://plugins/testing'));
+		expect(response.status).toBe(404);
+	});
+
+	test('should return a 405 method not supported response for non-get requests', async () => {
+		const { onRequestListener } = setUpProtocolHandler();
+
+		const response = await onRequestListener(new Request('joplin-plugin://plugins/testing', { method: 'POST' }));
+		expect(response.status).toBe(405);
+	});
+
+
+	test('should return a content script', async () => {
+		const { protocolHandler, onRequestListener } = setUpProtocolHandler();
+
+		const handle = protocolHandler.registerContentScript('some-id-here', 'test()');
+		const response = await onRequestListener(new Request(handle.uri));
+		expect(response.status).toBe(200);
+		expect(await response.text()).toBe('test()');
+		expect(response.headers.get('Content-Type')).toBe('application/javascript');
+	});
+
+	test('should return a 404 error after a content script has been removed', async () => {
+		const { protocolHandler, onRequestListener } = setUpProtocolHandler();
+
+		const handle = protocolHandler.registerContentScript('some-id-here', 'test()');
+		handle.revoke();
+		const response = await onRequestListener(new Request(handle.uri));
+		expect(response.status).toBe(404);
+	});
+});