Archived
80 lines
2.9 KiB
TypeScript
80 lines
2.9 KiB
TypeScript
import { EventEmitter } from "node:events";
|
|
import { describe, expect, it, vi } from "vitest";
|
|
import { SessionEventHub, type RealtimeSocket } from "./sessionEventHub.js";
|
|
|
|
class FakeSocket extends EventEmitter implements RealtimeSocket {
|
|
readonly OPEN = 1;
|
|
readyState = this.OPEN;
|
|
send = vi.fn();
|
|
}
|
|
|
|
describe("SessionEventHub", () => {
|
|
it("publishes session events only to sockets for that session", () => {
|
|
const hub = new SessionEventHub();
|
|
const sessionSocket = new FakeSocket();
|
|
const otherSocket = new FakeSocket();
|
|
hub.add("s1", sessionSocket);
|
|
hub.add("s2", otherSocket);
|
|
|
|
hub.publish("s1", { type: "assistant.delta", text: "hello" });
|
|
|
|
expect(sessionSocket.send).toHaveBeenCalledWith(JSON.stringify({ type: "assistant.delta", text: "hello" }));
|
|
expect(otherSocket.send).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("omits thinking signatures from final-message payloads without mutating source events", () => {
|
|
const hub = new SessionEventHub();
|
|
const socket = new FakeSocket();
|
|
hub.add("s1", socket);
|
|
const thinkingBlock = { type: "thinking", thinking: "private chain", thinkingSignature: "opaque-provider-payload", redacted: true };
|
|
const message = { role: "assistant", content: [thinkingBlock, { type: "text", text: "visible answer" }] };
|
|
|
|
hub.publish("s1", { type: "message.end", message });
|
|
|
|
expect(socket.send).toHaveBeenCalledWith(JSON.stringify({
|
|
type: "message.end",
|
|
message: { role: "assistant", content: [{ type: "thinking", thinking: "private chain", redacted: true }, { type: "text", text: "visible answer" }] },
|
|
}));
|
|
expect(thinkingBlock.thinkingSignature).toBe("opaque-provider-payload");
|
|
});
|
|
|
|
it("removes session sockets on close and skips non-open sockets", () => {
|
|
const hub = new SessionEventHub();
|
|
const closed = new FakeSocket();
|
|
const removed = new FakeSocket();
|
|
closed.readyState = 3;
|
|
hub.add("s1", closed);
|
|
hub.add("s1", removed);
|
|
removed.emit("close");
|
|
|
|
hub.publish("s1", { type: "session.error", message: "boom" });
|
|
|
|
expect(closed.send).not.toHaveBeenCalled();
|
|
expect(removed.send).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("publishes global events only to global sockets", () => {
|
|
const hub = new SessionEventHub();
|
|
const globalSocket = new FakeSocket();
|
|
const sessionSocket = new FakeSocket();
|
|
hub.addGlobal(globalSocket);
|
|
hub.add("s1", sessionSocket);
|
|
|
|
const status = {
|
|
sessionId: "s1",
|
|
isStreaming: false,
|
|
isCompacting: false,
|
|
isBashRunning: false,
|
|
pendingMessageCount: 0,
|
|
queuedMessages: [],
|
|
tokens: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
cost: 0,
|
|
};
|
|
|
|
hub.publishGlobal({ type: "status.update", status });
|
|
|
|
expect(globalSocket.send).toHaveBeenCalledWith(JSON.stringify({ type: "status.update", status }));
|
|
expect(sessionSocket.send).not.toHaveBeenCalled();
|
|
});
|
|
});
|