mirror of
https://github.com/yamadashy/repomix.git
synced 2026-05-30 11:18:53 +02:00
Kazuki Yamada
e5f7a1f311
- shared/errorHandle: recognize duck-typed OperationCancelledError from worker boundaries in isRepomixError (it extends RepomixError but the name was missing from the structured-clone fallback comparison). Add a regression test for the worker-boundary case. Test improvements per coderabbit / claude review: - cliReport: assert skill-directory + relative path on the same log line. - processConcurrency: restore process.versions.bun by removing the property when it didn't originally exist, instead of leaving it defined-as-undefined. - logger: drop the no-op `process.env.REPOMIX_LOG_LEVEL = undefined` (it coerces to the string "undefined" and is overwritten by the next delete). - unifiedWorker: replace the tautological cache test with one that proves cache uniqueness via onWorkerTermination cleanup count; add a test for task-based inference overriding workerData (bundled-env reuse). - calculateMetricsWorker: new direct test for the default export's items vs. single-mode dispatch — unifiedWorker mocks this module so the branch was otherwise untested. - packRemoteRepositoryTool: hard-code the expected output path instead of expect.any(String) to catch arg-swap regressions. - memoryUtils: tighten getMemoryStats assertions with sanity bounds (heapUsed <= heapTotal, rss > 0, heapUsagePercent <= 100) so a unit-conversion regression (bytes vs MB) would fail the test.
71 lines
2.5 KiB
TypeScript
71 lines
2.5 KiB
TypeScript
import { afterEach, describe, expect, test, vi } from 'vitest';
|
|
import { logger } from '../../src/shared/logger.js';
|
|
import {
|
|
getMemoryStats,
|
|
logMemoryDifference,
|
|
logMemoryUsage,
|
|
withMemoryLogging,
|
|
} from '../../src/shared/memoryUtils.js';
|
|
|
|
vi.mock('../../src/shared/logger', () => ({
|
|
logger: {
|
|
trace: vi.fn(),
|
|
},
|
|
}));
|
|
|
|
describe('memoryUtils', () => {
|
|
afterEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
test('getMemoryStats returns numeric MB values and a heap percentage', () => {
|
|
const stats = getMemoryStats();
|
|
// Sanity bounds — these would catch unit-conversion regressions
|
|
// (e.g., returning bytes instead of MB) that `expect.any(Number)` misses.
|
|
expect(stats.heapTotal).toBeGreaterThan(0);
|
|
expect(stats.rss).toBeGreaterThan(0);
|
|
expect(stats.heapUsed).toBeGreaterThan(0);
|
|
expect(stats.heapUsed).toBeLessThanOrEqual(stats.heapTotal);
|
|
expect(stats.external).toBeGreaterThanOrEqual(0);
|
|
expect(stats.heapUsagePercent).toBeGreaterThanOrEqual(0);
|
|
expect(stats.heapUsagePercent).toBeLessThanOrEqual(100);
|
|
});
|
|
|
|
test('logMemoryUsage emits a trace line tagged with context', () => {
|
|
logMemoryUsage('parse');
|
|
expect(logger.trace).toHaveBeenCalledWith(expect.stringContaining('Memory [parse]'));
|
|
});
|
|
|
|
test('logMemoryDifference formats positive and negative deltas with sign', () => {
|
|
const before = { heapUsed: 10, heapTotal: 20, external: 1, rss: 50, heapUsagePercent: 50 };
|
|
const after = { heapUsed: 15, heapTotal: 20, external: 0, rss: 48, heapUsagePercent: 75 };
|
|
|
|
logMemoryDifference('parse', before, after);
|
|
|
|
const message = (logger.trace as ReturnType<typeof vi.fn>).mock.calls[0][0] as string;
|
|
expect(message).toContain('+5.00MB'); // heap diff
|
|
expect(message).toContain('-2.00MB'); // rss diff
|
|
expect(message).toContain('-1.00MB'); // external diff
|
|
});
|
|
|
|
test('withMemoryLogging returns the inner result on success', async () => {
|
|
const result = await withMemoryLogging('task', async () => 'ok');
|
|
expect(result).toBe('ok');
|
|
// Before, After, Delta — three trace lines.
|
|
expect(logger.trace).toHaveBeenCalledTimes(3);
|
|
});
|
|
|
|
test('withMemoryLogging rethrows but still logs After (Error)', async () => {
|
|
const boom = new Error('boom');
|
|
|
|
await expect(
|
|
withMemoryLogging('task', async () => {
|
|
throw boom;
|
|
}),
|
|
).rejects.toBe(boom);
|
|
|
|
const messages = (logger.trace as ReturnType<typeof vi.fn>).mock.calls.map((c) => String(c[0]));
|
|
expect(messages.some((m) => m.includes('After (Error)'))).toBe(true);
|
|
});
|
|
});
|