ask-claude.test.ts

// node --test ask-claude.test.ts
import { test } from "node:test";
import { query } from "@anthropic-ai/claude-code";

test("Ask", async () => {
  const EXPECTED = "paris";
  const abortController = new AbortController();
  for await (const message of query({
    prompt: "What is the capital of France?",
    options: { maxTurns: 1, abortController },
  })) {
    if (message.type === "assistant") {
      const c = message.message.content;
      const res: string = Array.isArray(c) ? c.map((c) => c.text).join("") : c;
      if (res.toLowerCase().includes(EXPECTED)) {
        abortController.abort();
        return;
      }
    }
  }
  throw new Error("No response");
});

test("Read with tools", async () => {
  const EXPECTED = "APE";
  const abortController = new AbortController();
  for await (const message of query({
    prompt: "Read README.md and tell me about it",
    options: { maxTurns: 7, allowedTools: ["Read(*)"], cwd: process.cwd() },
  })) {
    if (message.type === "result" && message.subtype === "success") {
      if (message.result.includes(EXPECTED)) {
        abortController.abort();
        return;
      }
    }
  }
  throw new Error("No response");
});