agent-ecosystem/test/setup.ts

50 lines
1.2 KiB
TypeScript

/**
* Vitest setup file.
* Runs before each test file.
*/
import { afterEach, beforeEach, expect, vi } from 'vitest';
// Mock process.env for tests that need home directory
vi.stubGlobal('process', {
...process,
env: {
...process.env,
HOME: '/home/testuser',
},
});
let errorSpy: ReturnType<typeof vi.spyOn>;
let warnSpy: ReturnType<typeof vi.spyOn>;
function formatConsoleCall(args: unknown[]): string {
return args
.map((arg) => {
if (arg instanceof Error) {
return arg.message;
}
return String(arg);
})
.join(' ');
}
beforeEach(() => {
errorSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
warnSpy = vi.spyOn(console, 'warn').mockImplementation(() => {});
});
afterEach(() => {
const unexpectedErrors = errorSpy.mock.calls.map(formatConsoleCall);
const unexpectedWarnings = warnSpy.mock.calls.map(formatConsoleCall);
errorSpy.mockRestore();
warnSpy.mockRestore();
expect(unexpectedErrors, `Unexpected console.error calls:\n${unexpectedErrors.join('\n')}`).toEqual(
[]
);
expect(
unexpectedWarnings,
`Unexpected console.warn calls:\n${unexpectedWarnings.join('\n')}`
).toEqual([]);
});