lee/puter
1
0 Fork 0
mirror of https://github.com/HeyPuter/puter.git synced 2026-11-08 06:54:18 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

test(ai): wait for the matching realtime billing batch

Browse Source
This commit is contained in:
404oops committed 2026-10-29 11:59:03 +01:00
1 parent 5071cfd802
commit fbd6682389
1 file changed
+18 -9
Download Patch File Download Diff File Expand all files Collapse all files
src/backend/services/realtime/OpenAIRealtimeRelayService.test.ts
+18 -9
View File
@@ -723,16 +723,25 @@ describe('OpenAIRealtimeRelayService relay', () => {
const c = open(
`${relayUrl}?model=gpt-realtime-mini&auth_token=${user.token}`,
);
await c.waitFor('session.created');
c.ws.send(JSON.stringify({ type: 'response.create' }));
await vi.waitFor(() => expect(spy).toHaveBeenCalled());
c.ws.close();
try {
await c.waitFor('session.created');
c.ws.send(JSON.stringify({ type: 'response.create' }));
// Closed sessions can still record usage from earlier tests.
await vi.waitFor(() =>
expect(
spy.mock.calls.map(([, rows]) =>
rows.map((row) => [row.usageType, row.costOverride]),
),
).toContainEqual([
['openai:gpt-realtime-mini:realtime:text_input', 7 * 60 * 2],
['openai:gpt-realtime-mini:realtime:text_output', 3 * 240 * 2],
]),
);
} finally {
c.ws.close();
await c.closed;
}
});
const [, rows] = spy.mock.calls.at(-1)!;
expect(rows.map((r) => [r.usageType, r.costOverride])).toEqual([
['openai:gpt-realtime-mini:realtime:text_input', 7 * 60 * 2],
['openai:gpt-realtime-mini:realtime:text_output', 3 * 240 * 2],
]);
});
it('refuses a session.update that changes the model', async () => {