meld/api/nested-array.test.ts

Meld: A template language for LLM prompts

import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
import { TestContext } from '@tests/utils/index.js';
import { main } from './index.js';
import type { Services, ProcessOptions } from '@core/types/index.js';

describe('Nested Array Access Tests', () => {
  let context: TestContext;

  beforeEach(async () => {
    context = new TestContext();
    await context.initialize();
    context.enableTransformation();
  });

  afterEach(async () => {
    await context.cleanup();
    vi.resetModules();
  });

  it('should handle nested array access with dot notation', async () => {
    const content = 
`@data nestedArray = [
  ["a", "b", "c"],
  ["d", "e", "f"],
  ["g", "h", "i"]
]

First item of first array: {{nestedArray.0.0}}
Second item of second array: {{nestedArray.1.1}}
Third item of third array: {{nestedArray.2.2}}`;
    
    await context.writeFile('test.meld', content);
    
    const result = await main('test.meld', {
      fs: context.fs,
      services: context.services as unknown as Partial<Services>,
      transformation: true
    });
    
    // Log the content for debugging
    console.log('CONTENT:', content);
    console.log('RESULT:', result);
    
    expect(result.trim()).toBe('First item of first array: a\nSecond item of second array: e\nThird item of third array: i');
  });
});