Function useAimock

src/jest.ts:53–107

(options: UseAimockOptions = {})

 * in the test environment. This works with the default jest configuration.
 */
export function useAimock(options: UseAimockOptions = {}): () => AimockHandle {
  let handle: AimockHandle | null = null;
  let origOpenaiUrl: string | undefined;
  let origAnthropicUrl: string | undefined;

  beforeAll(async () => {
    const { fixtures: fixturePath, patchEnv, ...serverOpts } = options;
    const llm = new LLMock(serverOpts);

    if (fixturePath) {
      const resolved = resolve(fixturePath);
      const loadedFixtures = loadFixtures(resolved);
      for (const f of loadedFixtures) {
        llm.addFixture(f);
      }
    }

    const url = await llm.start();

    if (patchEnv !== false) {
      origOpenaiUrl = process.env.OPENAI_BASE_URL;
      origAnthropicUrl = process.env.ANTHROPIC_BASE_URL;
      process.env.OPENAI_BASE_URL = `${url}/v1`;
      process.env.ANTHROPIC_BASE_URL = `${url}/v1`;
    }

    handle = { llm, url };
  });

  beforeEach(() => {
    if (handle) {
      handle.llm.resetMatchCounts();
    }
  });

  afterAll(async () => {
    if (handle) {
      if (options.patchEnv !== false) {
        if (origOpenaiUrl !== undefined) process.env.OPENAI_BASE_URL = origOpenaiUrl;
        else delete process.env.OPENAI_BASE_URL;
        if (origAnthropicUrl !== undefined) process.env.ANTHROPIC_BASE_URL = origAnthropicUrl;
        else delete process.env.ANTHROPIC_BASE_URL;
      }
      await handle.llm.stop();
      handle = null;
    }
  });

  return () => {
    if (!handle) {
      throw new Error("useAimock(): server not started — are you calling this inside a test?");
    }
    return handle;
  };
}

function loadFixtures(fixturePath: string): Fixture[] {
  try {