mirror of
https://github.com/wassname/pi-goals.git
synced 2026-09-20 13:10:47 +08:00
wassname
47cc054582
Borrow the provider tracker queries from cecb1e9. Keep unavailable state unknown, bind incremental views to acknowledged source entries, reset after compaction, bound serialized payloads, and recheck tracked work at sign-off.
32 lines
2.0 KiB
TypeScript
32 lines
2.0 KiB
TypeScript
import { EventEmitter } from "node:events";
|
|
import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
|
|
import { expect, it, vi } from "vitest";
|
|
import { backgroundState } from "../src/background.js";
|
|
|
|
function api(tools: string[], processes?: unknown, subagents?: number) {
|
|
const bus = new EventEmitter();
|
|
if (processes !== undefined) bus.on("processes:request:list", request => request.reply(processes));
|
|
if (subagents !== undefined) bus.on("subagents:rpc:v1:request", request => bus.emit(`subagents:rpc:v1:reply:${request.requestId}`, { requestId: request.requestId, success: true, data: { fleet: { version: 1, totalActive: subagents } } }));
|
|
return { getAllTools: () => tools.map(name => ({ name })), events: { emit: (name: string, value: unknown) => bus.emit(name, value), on: (name: string, fn: (...args: any[]) => void) => { bus.on(name, fn); return () => { bus.off(name, fn); }; } } } as unknown as ExtensionAPI;
|
|
}
|
|
|
|
it("reports tracked running work, rather than equating idle agent with finished jobs", async () => {
|
|
const active = await backgroundState(api(["process", "subagent"], [{ name: "generation", status: "running" }], 1));
|
|
expect(active.quiet).toBe(false);
|
|
expect(active.description).toContain("processes: 1 (generation)");
|
|
expect(active.description).toContain("subagents: 1");
|
|
const finished = await backgroundState(api(["process", "subagent"], [{ status: "exited" }], 0));
|
|
expect(finished.quiet).toBe(true);
|
|
});
|
|
|
|
it("distinguishes missing providers from an unavailable installed tracker", async () => {
|
|
expect((await backgroundState(api([]))).quiet).toBe(true);
|
|
expect(await backgroundState(api(["process"]))).toMatchObject({ quiet: false, description: expect.stringContaining("processes: unknown") });
|
|
vi.useFakeTimers();
|
|
try {
|
|
const unavailable = backgroundState(api(["subagent"]));
|
|
await vi.advanceTimersByTimeAsync(2000);
|
|
expect(await unavailable).toMatchObject({ quiet: false, description: expect.stringContaining("subagents: unknown") });
|
|
} finally { vi.useRealTimers(); }
|
|
});
|