promptfoo/jest.setup.js

/* eslint-disable jest/require-top-level-describe */
import { rm } from 'fs/promises';
import fs from 'fs';
import yaml from 'js-yaml';
import nock from 'nock';
import { TEST_CONFIG_DIR } from './.jest/setEnvVars';

// Disable all real network requests
nock.disableNetConnect();
nock.enableNetConnect('127.0.0.1');

nock.emitter.on('no match', (req) => {
  if (!req.host.includes('127.0.0.1') && !req.host.includes('localhost')) {
    console.error(`Unexpected HTTP request: ${req.method} ${req.href}`);
  }
});

// Ensure mocks are loaded before tests
jest.mock('fs');
jest.mock('better-sqlite3', () => {
  const mockDb = {
    prepare: jest.fn().mockReturnValue({
      run: jest.fn(),
      get: jest.fn(),
      all: jest.fn(),
      finalize: jest.fn(),
    }),
    transaction: jest.fn((fn) => fn()),
    exec: jest.fn(),
    close: jest.fn(),
    backup: jest.fn().mockReturnValue({
      step: jest.fn().mockResolvedValue(true),
      finish: jest.fn(),
    }),
  };
  
  const Database = jest.fn().mockImplementation(() => mockDb);
  Database.prototype = mockDb;
  
  return { Database };
});
jest.mock('./src/globalConfig/globalConfig');
jest.mock('./src/globalConfig/cloud');
jest.mock('./src/globalConfig/accounts');
jest.mock('./src/telemetry');

// Reset all mocks before each test
beforeEach(() => {
  jest.clearAllMocks();
  // Reset fs mock state
  const mockFs = fs;
  mockFs.__clearMockFiles();

  // Create necessary directories and files for tests
  mockFs.mkdirSync('/tmp', { recursive: true });
  mockFs.mkdirSync('/mock/config/dir', { recursive: true });
  mockFs.writeFileSync('promptfoo-errors.log', '');
  mockFs.writeFileSync('promptfoo-debug.log', '');

  // Initialize with default config
  const configPath = '/mock/config/dir/promptfoo.yaml';
  const defaultConfig = {
    account: { email: 'mock-user@example.com' },
    cloud: {
      appUrl: 'https://mock.example.com',
      apiHost: 'https://mock-api.example.com',
      apiKey: 'mock-key',
    },
  };
  mockFs.__setMockFileContent(configPath, yaml.dump(defaultConfig));
});

afterAll(async () => {
  try {
    await rm(TEST_CONFIG_DIR, { recursive: true });
  } catch {}
});