feat(companion): add typed agent stream subscription helpers
This commit is contained in:
@@ -196,6 +196,34 @@ describe('CompanionRuntimeClient', () => {
|
||||
expect(streamHandler).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('supports subscribeAgentStream and subscribeAgentTyping helpers', () => {
|
||||
const client = new CompanionRuntimeClient({
|
||||
url: 'ws://127.0.0.1:1',
|
||||
});
|
||||
const streamHandler = vi.fn();
|
||||
const typingHandler = vi.fn();
|
||||
client.subscribeAgentStream(streamHandler);
|
||||
client.subscribeAgentTyping(typingHandler);
|
||||
|
||||
(client as unknown as { handleMessage: (raw: string) => void }).handleMessage(
|
||||
JSON.stringify({
|
||||
id: 52,
|
||||
event: 'agent.stream',
|
||||
data: { token: 'x' },
|
||||
}),
|
||||
);
|
||||
(client as unknown as { handleMessage: (raw: string) => void }).handleMessage(
|
||||
JSON.stringify({
|
||||
id: 53,
|
||||
event: 'agent.typing',
|
||||
data: { active: true },
|
||||
}),
|
||||
);
|
||||
|
||||
expect(streamHandler).toHaveBeenCalledWith({ token: 'x' });
|
||||
expect(typingHandler).toHaveBeenCalledWith({ active: true });
|
||||
});
|
||||
|
||||
it('clears all event subscriptions', () => {
|
||||
const client = new CompanionRuntimeClient({
|
||||
url: 'ws://127.0.0.1:1',
|
||||
|
||||
Reference in New Issue
Block a user