meld/tests/xml-output-format.test.ts

Meld: A template language for LLM prompts

import { describe, it, expect, beforeEach, afterEach } from 'vitest';
import { TestContext } from './utils/TestContext.js';
import { OutputService } from '../services/pipeline/OutputService/OutputService.js';
import { StateService } from '../services/state/StateService/StateService.js';

describe('XML Output Format', () => {
  let context: TestContext;

  beforeEach(async () => {
    // Set up test context
    context = new TestContext();
    await context.initialize();
    
    // Enable test mode for PathService
    context.services.path.enableTestMode();
  });

  afterEach(async () => {
    // Clean up
    await context.cleanup();
  });

  it('should convert markdown to XML format directly', async () => {
    // Create a simple markdown string
    const markdown = "# Hello World\n\nThis is a test.";
    
    // Create a fresh OutputService
    const outputService = new OutputService();
    const stateService = new StateService();
    
    // Initialize the OutputService with the StateService
    outputService.initialize(stateService);
    
    // Call the convertToXML method directly using type assertion to access private method
    const xmlOutput = await (outputService as any).convertToXML([], stateService, {
      formatOptions: {
        markdown
      }
    });
    
    // Debug log
    console.log('Direct XML Output:', xmlOutput);
    
    // Verify XML output contains proper XML tags
    expect(xmlOutput).toContain('<');  // Should have at least one XML tag
    expect(xmlOutput).toContain('>');  // Should have at least one closing tag
    expect(xmlOutput).toContain('HelloWorld');  // Should contain the content (without space)
    expect(xmlOutput).toContain('This is a test');  // Should contain the content
  });

  it('should properly handle JSON content in markdown when converting to XML', async () => {
    // Create markdown with JSON content
    const markdown = `# Test with JSON

Here's some JSON data:

\`\`\`json
{
  "name": "Test User",
  "age": 30,
  "items": ["apple", "orange", "banana"]
}
\`\`\`

More text after the JSON.`;
    
    // Create a fresh OutputService
    const outputService = new OutputService();
    const stateService = new StateService();
    
    // Initialize the OutputService with the StateService
    outputService.initialize(stateService);
    
    // Call the convertToXML method directly
    const xmlOutput = await (outputService as any).convertToXML([], stateService, {
      formatOptions: {
        markdown
      }
    });
    
    // Debug log
    console.log('XML Output with JSON:', xmlOutput);
    
    // Verify XML output handles the JSON content properly
    expect(xmlOutput).toContain('<');  // Should have XML tags
    expect(xmlOutput).toContain('>');  // Should have closing tags
    
    // Check that the XML contains content from the JSON
    expect(xmlOutput).toContain('TestWithJson');  // llmxml removes spaces and uses PascalCase
    expect(xmlOutput).toContain('name');
    expect(xmlOutput).toContain('Test User');
    expect(xmlOutput).toContain('apple');
    
    // The output should maintain the structure defined by llmxml
    // We're not testing for a specific XML structure, just that it processes without errors
    // and maintains the content from the input
  });
});