feat(companion): add generic platform event passthroughs

This commit is contained in:
William Valentin
2026-02-16 19:28:48 -08:00
parent 717e5d60e5
commit 0eb8126349
4 changed files with 76 additions and 1 deletions
+15
View File
@@ -28,9 +28,11 @@ function createRuntimeMock(): {
subscribeAgentStream: ReturnType<typeof vi.fn>;
subscribeAgentTyping: ReturnType<typeof vi.fn>;
subscribeContextWarning: ReturnType<typeof vi.fn>;
subscribeEvents: ReturnType<typeof vi.fn>;
waitForAgentStream: ReturnType<typeof vi.fn>;
waitForAgentTyping: ReturnType<typeof vi.fn>;
waitForContextWarning: ReturnType<typeof vi.fn>;
waitForAnyEvent: ReturnType<typeof vi.fn>;
} {
const connect = vi.fn(async () => undefined);
const disconnect = vi.fn(() => undefined);
@@ -59,9 +61,11 @@ function createRuntimeMock(): {
const subscribeAgentStream = vi.fn(() => () => undefined);
const subscribeAgentTyping = vi.fn(() => () => undefined);
const subscribeContextWarning = vi.fn(() => () => undefined);
const subscribeEvents = vi.fn(() => () => undefined);
const waitForAgentStream = vi.fn(async () => ({ token: 'streamed' }));
const waitForAgentTyping = vi.fn(async () => ({ active: true }));
const waitForContextWarning = vi.fn(async () => ({ thresholdPct: 75, estimatedPct: 90 }));
const waitForAnyEvent = vi.fn(async () => ({ event: 'agent.stream', data: { token: 'any' } }));
const runtime = {
connect,
@@ -84,9 +88,11 @@ function createRuntimeMock(): {
subscribeAgentStream,
subscribeAgentTyping,
subscribeContextWarning,
subscribeEvents,
waitForAgentStream,
waitForAgentTyping,
waitForContextWarning,
waitForAnyEvent,
} as unknown as CompanionRuntimeClient;
return {
@@ -111,9 +117,11 @@ function createRuntimeMock(): {
subscribeAgentStream,
subscribeAgentTyping,
subscribeContextWarning,
subscribeEvents,
waitForAgentStream,
waitForAgentTyping,
waitForContextWarning,
waitForAnyEvent,
};
}
@@ -180,26 +188,33 @@ describe('platform companion clients', () => {
const streamHandler = vi.fn();
const typingHandler = vi.fn();
const warningHandler = vi.fn();
const eventHandler = vi.fn();
const unsubscribeStream = client.subscribeAgentStream(streamHandler);
const unsubscribeTyping = client.subscribeAgentTyping(typingHandler);
const unsubscribeWarning = client.subscribeContextWarning(warningHandler);
const unsubscribeEvents = client.subscribeEvents(eventHandler);
const awaited = await client.waitForAgentStream<{ token: string }>({ timeoutMs: 500 });
const waitedTyping = await client.waitForAgentTyping<{ active: boolean }>({ timeoutMs: 500 });
const waitedWarning = await client.waitForContextWarning<{ thresholdPct: number; estimatedPct: number }>({ timeoutMs: 500 });
const waitedAny = await client.waitForAnyEvent<{ token: string }>(['agent.stream'], { timeoutMs: 500 });
expect(mock.subscribeAgentStream).toHaveBeenCalledWith(streamHandler);
expect(mock.subscribeAgentTyping).toHaveBeenCalledWith(typingHandler);
expect(mock.subscribeContextWarning).toHaveBeenCalledWith(warningHandler);
expect(mock.subscribeEvents).toHaveBeenCalledWith(eventHandler);
expect(mock.waitForAgentStream).toHaveBeenCalledWith({ timeoutMs: 500 });
expect(mock.waitForAgentTyping).toHaveBeenCalledWith({ timeoutMs: 500 });
expect(mock.waitForContextWarning).toHaveBeenCalledWith({ timeoutMs: 500 });
expect(mock.waitForAnyEvent).toHaveBeenCalledWith(['agent.stream'], { timeoutMs: 500 });
expect(awaited).toEqual({ token: 'streamed' });
expect(waitedTyping).toEqual({ active: true });
expect(waitedWarning).toEqual({ thresholdPct: 75, estimatedPct: 90 });
expect(waitedAny).toEqual({ event: 'agent.stream', data: { token: 'any' } });
unsubscribeStream();
unsubscribeTyping();
unsubscribeWarning();
unsubscribeEvents();
});
it('macOS client forwards canvas methods to runtime client', async () => {