n8n-enterprise-unlocked/packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/test/config.test.ts

import { NodeConnectionType } from 'n8n-workflow';

import { getInputs, nodeProperties } from '../methods/config';

describe('config', () => {
	describe('getInputs', () => {
		it('should return basic inputs for all parameters', () => {
			const inputs = getInputs({});

			expect(inputs).toHaveLength(3);
			expect(inputs[0].type).toBe(NodeConnectionType.Main);
			expect(inputs[1].type).toBe(NodeConnectionType.AiLanguageModel);
			expect(inputs[2].type).toBe(NodeConnectionType.AiOutputParser);
		});

		it('should exclude the OutputParser when hasOutputParser is false', () => {
			const inputs = getInputs({ hasOutputParser: false });

			expect(inputs).toHaveLength(2);
			expect(inputs[0].type).toBe(NodeConnectionType.Main);
			expect(inputs[1].type).toBe(NodeConnectionType.AiLanguageModel);
		});

		it('should include the OutputParser when hasOutputParser is true', () => {
			const inputs = getInputs({ hasOutputParser: true });

			expect(inputs).toHaveLength(3);
			expect(inputs[2].type).toBe(NodeConnectionType.AiOutputParser);
		});
	});

	describe('nodeProperties', () => {
		it('should have the expected properties', () => {
			expect(Array.isArray(nodeProperties)).toBe(true);
			expect(nodeProperties.length).toBeGreaterThan(0);

			const promptParams = nodeProperties.filter((prop) => prop.name === 'prompt');
			expect(promptParams.length).toBeGreaterThan(0);

			const messagesParam = nodeProperties.find((prop) => prop.name === 'messages');
			expect(messagesParam).toBeDefined();
			expect(messagesParam?.type).toBe('fixedCollection');

			const hasOutputParserParam = nodeProperties.find((prop) => prop.name === 'hasOutputParser');
			expect(hasOutputParserParam).toBeDefined();
			expect(hasOutputParserParam?.type).toBe('boolean');
		});
	});
});