repomix-mirror/tests/core/packager/splitOutput.test.ts

import { describe, expect, it, vi } from 'vitest';
import { pack } from '../../../src/core/packager.js';
import { createMockConfig } from '../../testing/testUtils.js';

describe('packager split output', () => {
  it('passes split output results correctly through the packager', async () => {
    const processedFiles = [
      { path: 'a/file1.txt', content: '11111' },
      { path: 'b/file2.txt', content: '22222' },
    ];

    const allFilePaths = ['a/file1.txt', 'b/file2.txt'];
    const mockConfig = createMockConfig({
      cwd: '/test',
      output: {
        filePath: 'repomix-output.xml',
        splitOutput: 12,
        copyToClipboard: false,
        stdout: false,
        git: {
          includeDiffs: false,
          includeLogs: false,
        },
      },
    });

    const produceOutput = vi.fn().mockResolvedValue({
      outputFiles: ['repomix-output.1.xml', 'repomix-output.2.xml'],
      outputForMetrics: ['x'.repeat(10), 'x'.repeat(10)],
    });

    const calculateMetrics = vi.fn().mockResolvedValue({
      totalFiles: 2,
      totalCharacters: 0,
      totalTokens: 0,
      fileCharCounts: {},
      fileTokenCounts: {},
      gitDiffTokenCount: 0,
      gitLogTokenCount: 0,
    });

    const result = await pack(['root'], mockConfig, () => {}, {
      searchFiles: vi.fn().mockResolvedValue({ filePaths: allFilePaths, emptyDirPaths: [] }),
      sortPaths: vi.fn().mockImplementation((paths) => paths),
      collectFiles: vi.fn().mockResolvedValue({ rawFiles: processedFiles, skippedFiles: [] }),
      processFiles: vi.fn().mockReturnValue(processedFiles),
      validateFileSafety: vi.fn().mockResolvedValue({
        safeFilePaths: allFilePaths,
        safeRawFiles: processedFiles,
        suspiciousFilesResults: [],
        suspiciousGitDiffResults: [],
        suspiciousGitLogResults: [],
      }),
      getGitDiffs: vi.fn().mockResolvedValue(undefined),
      getGitLogs: vi.fn().mockResolvedValue(undefined),
      produceOutput,
      calculateMetrics,
      createMetricsTaskRunner: vi.fn().mockReturnValue({
        taskRunner: {
          run: vi.fn().mockResolvedValue(0),
          cleanup: vi.fn().mockResolvedValue(undefined),
        },
        warmupPromise: Promise.resolve(),
      }),
      createSecurityTaskRunner: vi.fn().mockReturnValue({
        taskRunner: {
          run: vi.fn().mockResolvedValue([]),
          cleanup: vi.fn().mockResolvedValue(undefined),
        },
        warmupPromise: Promise.resolve(),
      }),
    });

    expect(produceOutput).toHaveBeenCalledWith(
      ['root'],
      mockConfig,
      processedFiles,
      allFilePaths,
      undefined,
      undefined,
      expect.any(Function),
      [{ rootLabel: 'root', files: allFilePaths }],
      undefined,
    );

    expect(calculateMetrics).toHaveBeenCalledWith(
      processedFiles,
      expect.anything(),
      expect.anything(),
      mockConfig,
      undefined,
      undefined,
      expect.objectContaining({ taskRunner: expect.anything() }),
    );

    // Verify that calculateMetrics received a promise that resolves to the expected split output
    const outputArg = calculateMetrics.mock.calls[0][1];
    await expect(outputArg).resolves.toEqual(['x'.repeat(10), 'x'.repeat(10)]);

    expect(result.outputFiles).toEqual(['repomix-output.1.xml', 'repomix-output.2.xml']);
  });
});