assistant-stream/src/ai-sdk/index.test.ts

Streaming utilities for AI assistants

import { beforeEach, describe, expect, it, vi } from "vitest";

const loaded = vi.hoisted(() => ({ ai: false }));

describe("ai-sdk entry", () => {
  beforeEach(() => {
    vi.resetModules();
    loaded.ai = false;
    vi.doMock("ai", async (importOriginal) => {
      loaded.ai = true;
      return { ...(await importOriginal<typeof import("ai")>()) };
    });
  });

  it("keeps `ai` out of the root entry's module graph", async () => {
    await import("../index");

    expect(loaded.ai).toBe(false);
  });

  it("keeps `ai` out of the internal entry's module graph", async () => {
    await import("../internal");

    expect(loaded.ai).toBe(false);
  });

  it("loads `ai` only through the ai-sdk subpath", async () => {
    const entry = await import("./index");

    expect(loaded.ai).toBe(true);
    expect(typeof entry.frontendTools).toBe("function");
  });
});