import { ChatMessageRole } from 'base/common/language-models/languageModels';
import { CustomActionPrompt } from '../prompt-manage/custom-action/CustomActionPrompt';
import { CustomActionType } from '../prompt-manage/custom-action/CustomActionType';
import { InteractionType } from '../prompt-manage/InteractionType';

describe('CustomActionPrompt', () => {
	describe('fromContent', () => {
		it('should parse content with front-matter and chat messages', () => {
			const content = `---
interaction: AppendCursorStream
priority: 1
key1: value1
key2: value2
---
\`\`\`system\`\`\`
Chat message 1
\`\`\`user\`\`\`
Chat message 2
`;
			const expectedPrompt = new CustomActionPrompt(
				undefined,
				InteractionType.AppendCursorStream,
				1,
				CustomActionType.Default,
				{
					key1: 'value1',
					key2: 'value2',
				},
				[
					{ role: ChatMessageRole.System, content: 'Chat message 1\n' },
					{ role: ChatMessageRole.User, content: 'Chat message 2\n\n' },
				],
			);
			const prompt = CustomActionPrompt.fromContent(content);
			expect(prompt).toEqual(expectedPrompt);
		});

		it('should handle invalid content', () => {
			const content = 'Invalid content';
			const expectedPrompt = new CustomActionPrompt(
				undefined,
				InteractionType.AppendCursorStream,
				0,
				CustomActionType.Default,
				{},
				[{ role: ChatMessageRole.User, content }],
			);

			const prompt = CustomActionPrompt.fromContent(content);
			expect(prompt).toEqual(expectedPrompt);
		});
	});
});