test-pi-extension

Testando Extensões Pi

O @marcfargas/pi-test-harness permite testar extensões pi com o runtime real — sem LLM. Tudo roda de verdade (loading, hooks, tools, eventos), exceto o modelo que é substituído por um playbook scriptado.

Instalação

npm install --save-dev @marcfargas/pi-test-harness vitest

Peer dependencies: @earendil-works/pi-coding-agent, @earendil-works/pi-ai, @earendil-works/pi-agent-core.

Teste Básico

import { describe, it, expect, afterEach } from "vitest";
import { createTestSession, when, calls, says, type TestSession } from "@marcfargas/pi-test-harness";

describe("minha extensão", () => {
  let t: TestSession;
  afterEach(() => t?.dispose());

  it("registra e executa uma tool", async () => {
    t = await createTestSession({
      extensions: ["./src/index.ts"],
      mockTools: {
        bash: (params) => `$ ${params.command}\noutput`,
        read: "conteúdo do arquivo",
        write: "ok",
        edit: "ok",
      },
    });

    await t.run(
      when("Liste os arquivos", [
        calls("bash", { command: "ls" }),
        says("Encontrei os arquivos."),
      ]),
    );

    expect(t.events.toolResultsFor("bash")).toHaveLength(1);
    expect(t.events.toolResultsFor("bash")[0].text).toContain("output");
  });
});

Arquitetura

┌───────────────────────────────────────┐
│  Ambiente pi REAL                     │
│                                       │
│  Extensions ─── carregadas de verdade │
│  Tool registry ─ hooks reais          │
│  Session state ─ persistência real    │
│                                       │
│  ┌─────────────────────────────────┐  │
│  │  streamFn ── SUBSTITUÍDO       │  │ ← playbook (when/calls/says)
│  │  tool.execute ── INTERCEPTADO  │  │ ← mockTools
│  │  ctx.ui.* ── INTERCEPTADO      │  │ ← mockUI
│  └─────────────────────────────────┘  │
└───────────────────────────────────────┘

Só 3 pontos são substituídos — todo o resto é pi real.

Playbook DSL