1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
import type { AgentEvent } from "@dispatch/core";
import { describe, expect, it, vi } from "vitest";
// Mock @dispatch/core's Agent to avoid real LLM calls
vi.mock("@dispatch/core", async () => {
const actual = await vi.importActual<typeof import("@dispatch/core")>("@dispatch/core");
return {
...actual,
Agent: class MockAgent {
status = "idle";
messages: unknown[] = [];
async *run(_message: string) {
yield { type: "status", status: "running" } as const;
await new Promise<void>((r) => setTimeout(r, 10));
yield { type: "text-delta", delta: "Hello " } as const;
yield { type: "text-delta", delta: "world" } as const;
yield {
type: "done",
message: { role: "assistant", content: "Hello world" },
} as const;
yield { type: "status", status: "idle" } as const;
}
},
};
});
// Import after mock is defined (Vitest hoists vi.mock automatically)
const { AgentManager } = await import("../src/agent-manager.js");
describe("AgentManager", () => {
it("initial status is idle", () => {
const manager = new AgentManager();
expect(manager.getStatus()).toBe("idle");
});
it("initial messageCount is 0", () => {
const manager = new AgentManager();
expect(manager.getMessageCount()).toBe(0);
});
it("event listeners receive events during processMessage", async () => {
const manager = new AgentManager();
const events: AgentEvent[] = [];
manager.onEvent((event) => {
events.push(event);
});
await manager.processMessage("test");
expect(events.length).toBeGreaterThan(0);
expect(events[0]).toEqual({ type: "status", status: "running" });
const lastEvent = events[events.length - 1];
expect(lastEvent).toEqual({ type: "status", status: "idle" });
const doneEvent = events.find((e) => e.type === "done");
expect(doneEvent).toBeDefined();
});
it("emits text-delta events during processMessage", async () => {
const manager = new AgentManager();
const events: AgentEvent[] = [];
manager.onEvent((event) => {
events.push(event);
});
await manager.processMessage("hello");
const textDeltas = events.filter((e) => e.type === "text-delta");
expect(textDeltas.length).toBeGreaterThan(0);
});
it("messageCount increments after processMessage", async () => {
const manager = new AgentManager();
await manager.processMessage("hello");
expect(manager.getMessageCount()).toBe(1);
await manager.processMessage("world");
expect(manager.getMessageCount()).toBe(2);
});
it("status returns to idle after processMessage completes", async () => {
const manager = new AgentManager();
await manager.processMessage("test");
expect(manager.getStatus()).toBe("idle");
});
it("unsubscribe removes listener", async () => {
const manager = new AgentManager();
const events: AgentEvent[] = [];
const unsubscribe = manager.onEvent((event) => {
events.push(event);
});
unsubscribe();
await manager.processMessage("test");
expect(events.length).toBe(0);
});
it("multiple listeners all receive events", async () => {
const manager = new AgentManager();
const listener1 = vi.fn();
const listener2 = vi.fn();
manager.onEvent(listener1);
manager.onEvent(listener2);
await manager.processMessage("test");
expect(listener1).toHaveBeenCalled();
expect(listener2).toHaveBeenCalled();
});
});
|
