10 Setup Tests

pnpm add -D vitest

import { describe, expect, it, vi } from "vitest";
import { POST } from "./+server";

vi.mock("$lib/server/ai/support-agent", () => ({
  supportAgent: {
    prompt: () => ({
      send: async () => ({
        output: "Reset links expire after 30 minutes.",
        usage: { totalTokens: 12 },
        messages: [],
      }),
    }),
  },
}));

describe("POST /api/support", () => {
  it("returns the agent output", async () => {
    const response = await POST({
      request: new Request("http://test.local/api/support", {
        method: "POST",
        headers: { "content-type": "application/json" },
        body: JSON.stringify({ message: "How long does a reset link last?" }),
      }),
      locals: {},
      params: {},
      url: new URL("http://test.local/api/support"),
    } as never);

    expect(response.status).toBe(200);
    await expect(response.json()).resolves.toMatchObject({
      output: "Reset links expire after 30 minutes.",
    });
  });
});

const stream = new ReadableStream({
  start(controller) {
    controller.enqueue(new TextEncoder().encode('{"type":"final"}\n'));
    controller.close();
  },
});

vi.mock("$lib/server/ai/support-agent", () => ({
  supportAgent: {
    prompt: () => ({
      readableStream: () => stream,
    }),
  },
}));

import { Studio } from "@anvia/studio";
import { supportAgent } from "$lib/server/ai/support-agent";

const studio = new Studio([supportAgent]);
const response = await studio.fetch(
  new Request("http://studio.test/agents/support/runs", {
    method: "POST",
    headers: { "content-type": "application/json" },
    body: JSON.stringify({ message: "Hello" }),
  }),
);

expect(response.status).toBe(200);