automaker/apps/server/tests/unit/lib/enhancement-prompts.test.ts

import { describe, it, expect } from 'vitest';
import {
  getEnhancementPrompt,
  getSystemPrompt,
  getExamples,
  buildUserPrompt,
  isValidEnhancementMode,
  getAvailableEnhancementModes,
  IMPROVE_SYSTEM_PROMPT,
  TECHNICAL_SYSTEM_PROMPT,
  SIMPLIFY_SYSTEM_PROMPT,
  ACCEPTANCE_SYSTEM_PROMPT,
  IMPROVE_EXAMPLES,
  TECHNICAL_EXAMPLES,
  SIMPLIFY_EXAMPLES,
  ACCEPTANCE_EXAMPLES,
  type EnhancementMode,
} from '@/lib/enhancement-prompts.js';

describe('enhancement-prompts.ts', () => {
  describe('System Prompt Constants', () => {
    it('should have non-empty improve system prompt', () => {
      expect(IMPROVE_SYSTEM_PROMPT).toBeDefined();
      expect(IMPROVE_SYSTEM_PROMPT.length).toBeGreaterThan(100);
      expect(IMPROVE_SYSTEM_PROMPT).toContain('ANALYZE');
      expect(IMPROVE_SYSTEM_PROMPT).toContain('CLARIFY');
    });

    it('should have non-empty technical system prompt', () => {
      expect(TECHNICAL_SYSTEM_PROMPT).toBeDefined();
      expect(TECHNICAL_SYSTEM_PROMPT.length).toBeGreaterThan(100);
      expect(TECHNICAL_SYSTEM_PROMPT).toContain('technical');
    });

    it('should have non-empty simplify system prompt', () => {
      expect(SIMPLIFY_SYSTEM_PROMPT).toBeDefined();
      expect(SIMPLIFY_SYSTEM_PROMPT.length).toBeGreaterThan(100);
      expect(SIMPLIFY_SYSTEM_PROMPT).toContain('simplify');
    });

    it('should have non-empty acceptance system prompt', () => {
      expect(ACCEPTANCE_SYSTEM_PROMPT).toBeDefined();
      expect(ACCEPTANCE_SYSTEM_PROMPT.length).toBeGreaterThan(100);
      expect(ACCEPTANCE_SYSTEM_PROMPT).toContain('acceptance criteria');
    });
  });

  describe('Example Constants', () => {
    it('should have improve examples with input and output', () => {
      expect(IMPROVE_EXAMPLES).toBeDefined();
      expect(IMPROVE_EXAMPLES.length).toBeGreaterThan(0);
      IMPROVE_EXAMPLES.forEach((example) => {
        expect(example.input).toBeDefined();
        expect(example.output).toBeDefined();
        expect(example.input.length).toBeGreaterThan(0);
        expect(example.output.length).toBeGreaterThan(0);
      });
    });

    it('should have technical examples with input and output', () => {
      expect(TECHNICAL_EXAMPLES).toBeDefined();
      expect(TECHNICAL_EXAMPLES.length).toBeGreaterThan(0);
      TECHNICAL_EXAMPLES.forEach((example) => {
        expect(example.input).toBeDefined();
        expect(example.output).toBeDefined();
      });
    });

    it('should have simplify examples with input and output', () => {
      expect(SIMPLIFY_EXAMPLES).toBeDefined();
      expect(SIMPLIFY_EXAMPLES.length).toBeGreaterThan(0);
      SIMPLIFY_EXAMPLES.forEach((example) => {
        expect(example.input).toBeDefined();
        expect(example.output).toBeDefined();
      });
    });

    it('should have acceptance examples with input and output', () => {
      expect(ACCEPTANCE_EXAMPLES).toBeDefined();
      expect(ACCEPTANCE_EXAMPLES.length).toBeGreaterThan(0);
      ACCEPTANCE_EXAMPLES.forEach((example) => {
        expect(example.input).toBeDefined();
        expect(example.output).toBeDefined();
      });
    });
  });

  describe('getEnhancementPrompt', () => {
    it('should return config for improve mode', () => {
      const config = getEnhancementPrompt('improve');
      expect(config.systemPrompt).toBe(IMPROVE_SYSTEM_PROMPT);
      expect(config.description).toContain('clear');
    });

    it('should return config for technical mode', () => {
      const config = getEnhancementPrompt('technical');
      expect(config.systemPrompt).toBe(TECHNICAL_SYSTEM_PROMPT);
      expect(config.description).toContain('technical');
    });

    it('should return config for simplify mode', () => {
      const config = getEnhancementPrompt('simplify');
      expect(config.systemPrompt).toBe(SIMPLIFY_SYSTEM_PROMPT);
      expect(config.description).toContain('concise');
    });

    it('should return config for acceptance mode', () => {
      const config = getEnhancementPrompt('acceptance');
      expect(config.systemPrompt).toBe(ACCEPTANCE_SYSTEM_PROMPT);
      expect(config.description).toContain('acceptance');
    });

    it('should handle case-insensitive mode', () => {
      const config = getEnhancementPrompt('IMPROVE');
      expect(config.systemPrompt).toBe(IMPROVE_SYSTEM_PROMPT);
    });

    it('should fall back to improve for invalid mode', () => {
      const config = getEnhancementPrompt('invalid-mode');
      expect(config.systemPrompt).toBe(IMPROVE_SYSTEM_PROMPT);
    });

    it('should fall back to improve for empty string', () => {
      const config = getEnhancementPrompt('');
      expect(config.systemPrompt).toBe(IMPROVE_SYSTEM_PROMPT);
    });
  });

  describe('getSystemPrompt', () => {
    it('should return correct system prompt for each mode', () => {
      expect(getSystemPrompt('improve')).toBe(IMPROVE_SYSTEM_PROMPT);
      expect(getSystemPrompt('technical')).toBe(TECHNICAL_SYSTEM_PROMPT);
      expect(getSystemPrompt('simplify')).toBe(SIMPLIFY_SYSTEM_PROMPT);
      expect(getSystemPrompt('acceptance')).toBe(ACCEPTANCE_SYSTEM_PROMPT);
    });
  });

  describe('getExamples', () => {
    it('should return correct examples for each mode', () => {
      expect(getExamples('improve')).toBe(IMPROVE_EXAMPLES);
      expect(getExamples('technical')).toBe(TECHNICAL_EXAMPLES);
      expect(getExamples('simplify')).toBe(SIMPLIFY_EXAMPLES);
      expect(getExamples('acceptance')).toBe(ACCEPTANCE_EXAMPLES);
    });

    it('should return arrays with example objects', () => {
      const modes: EnhancementMode[] = ['improve', 'technical', 'simplify', 'acceptance'];
      modes.forEach((mode) => {
        const examples = getExamples(mode);
        expect(Array.isArray(examples)).toBe(true);
        expect(examples.length).toBeGreaterThan(0);
      });
    });
  });

  describe('buildUserPrompt', () => {
    const testText = 'Add a logout button';

    it('should build prompt with examples by default', () => {
      const prompt = buildUserPrompt('improve', testText);
      expect(prompt).toContain('Example 1:');
      expect(prompt).toContain(testText);
      expect(prompt).toContain('Now, please enhance the following task description:');
    });

    it('should build prompt without examples when includeExamples is false', () => {
      const prompt = buildUserPrompt('improve', testText, false);
      expect(prompt).not.toContain('Example 1:');
      expect(prompt).toContain(testText);
      expect(prompt).toContain('Please enhance the following task description:');
    });

    it('should include all examples for improve mode', () => {
      const prompt = buildUserPrompt('improve', testText);
      IMPROVE_EXAMPLES.forEach((example, index) => {
        expect(prompt).toContain(`Example ${index + 1}:`);
        expect(prompt).toContain(example.input);
      });
    });

    it('should include separator between examples', () => {
      const prompt = buildUserPrompt('improve', testText);
      expect(prompt).toContain('---');
    });

    it('should work with all enhancement modes', () => {
      const modes: EnhancementMode[] = ['improve', 'technical', 'simplify', 'acceptance'];
      modes.forEach((mode) => {
        const prompt = buildUserPrompt(mode, testText);
        expect(prompt).toContain(testText);
        expect(prompt.length).toBeGreaterThan(100);
      });
    });

    it('should preserve the original text exactly', () => {
      const specialText = 'Add feature with special chars: <>&"\'';
      const prompt = buildUserPrompt('improve', specialText);
      expect(prompt).toContain(specialText);
    });
  });

  describe('isValidEnhancementMode', () => {
    it('should return true for valid modes', () => {
      expect(isValidEnhancementMode('improve')).toBe(true);
      expect(isValidEnhancementMode('technical')).toBe(true);
      expect(isValidEnhancementMode('simplify')).toBe(true);
      expect(isValidEnhancementMode('acceptance')).toBe(true);
    });

    it('should return false for invalid modes', () => {
      expect(isValidEnhancementMode('invalid')).toBe(false);
      expect(isValidEnhancementMode('IMPROVE')).toBe(false); // case-sensitive
      expect(isValidEnhancementMode('')).toBe(false);
      expect(isValidEnhancementMode('random')).toBe(false);
    });
  });

  describe('getAvailableEnhancementModes', () => {
    it('should return all four enhancement modes', () => {
      const modes = getAvailableEnhancementModes();
      expect(modes).toHaveLength(4);
      expect(modes).toContain('improve');
      expect(modes).toContain('technical');
      expect(modes).toContain('simplify');
      expect(modes).toContain('acceptance');
    });

    it('should return an array', () => {
      const modes = getAvailableEnhancementModes();
      expect(Array.isArray(modes)).toBe(true);
    });
  });
});