repomix-mirror/tests/core/file/fileTreeGenerate.test.ts

import { describe, expect, test } from 'vitest';
import {
  type FilesByRoot,
  generateTreeString,
  generateTreeStringWithRoots,
} from '../../../src/core/file/fileTreeGenerate.js';

describe('fileTreeGenerate', () => {
  describe('generateTreeString', () => {
    test('generates a flat tree for single directory files', () => {
      const files = ['file1.txt', 'file2.txt', 'subdir/nested.txt'];
      const result = generateTreeString(files);

      expect(result).toContain('file1.txt');
      expect(result).toContain('file2.txt');
      expect(result).toContain('subdir/');
      expect(result).toContain('nested.txt');
    });
  });

  describe('generateTreeStringWithRoots', () => {
    test('returns standard flat tree for single root', () => {
      const filesByRoot: FilesByRoot[] = [{ rootLabel: 'project', files: ['file1.txt', 'file2.txt'] }];

      const result = generateTreeStringWithRoots(filesByRoot);

      // Should not have root label for single root
      expect(result).not.toContain('[project]');
      expect(result).toContain('file1.txt');
      expect(result).toContain('file2.txt');
    });

    test('generates labeled sections for multiple roots', () => {
      const filesByRoot: FilesByRoot[] = [
        { rootLabel: 'cli', files: ['cliRun.ts', 'types.ts'] },
        { rootLabel: 'config', files: ['configLoad.ts', 'configSchema.ts'] },
      ];

      const result = generateTreeStringWithRoots(filesByRoot);

      // Should have root labels
      expect(result).toContain('[cli]/');
      expect(result).toContain('[config]/');

      // Should have files under each label
      expect(result).toContain('cliRun.ts');
      expect(result).toContain('types.ts');
      expect(result).toContain('configLoad.ts');
      expect(result).toContain('configSchema.ts');
    });

    test('generates labeled sections with nested directories', () => {
      const filesByRoot: FilesByRoot[] = [
        { rootLabel: 'src', files: ['index.ts', 'utils/helper.ts', 'utils/format.ts'] },
        { rootLabel: 'tests', files: ['index.test.ts'] },
      ];

      const result = generateTreeStringWithRoots(filesByRoot);

      expect(result).toContain('[src]/');
      expect(result).toContain('[tests]/');
      expect(result).toContain('utils/');
      expect(result).toContain('helper.ts');
    });

    test('skips empty root sections', () => {
      const filesByRoot: FilesByRoot[] = [
        { rootLabel: 'cli', files: ['cliRun.ts'] },
        { rootLabel: 'empty', files: [] },
        { rootLabel: 'config', files: ['configLoad.ts'] },
      ];

      const result = generateTreeStringWithRoots(filesByRoot);

      expect(result).toContain('[cli]/');
      expect(result).not.toContain('[empty]/');
      expect(result).toContain('[config]/');
    });

    test('handles single root falling back to standard behavior', () => {
      const filesByRoot: FilesByRoot[] = [{ rootLabel: 'project', files: ['a.txt', 'b/c.txt'] }];

      const singleRootResult = generateTreeStringWithRoots(filesByRoot);
      const standardResult = generateTreeString(['a.txt', 'b/c.txt']);

      // Should be identical
      expect(singleRootResult).toBe(standardResult);
    });
  });
});