@amodalai/runtime 0.3.64 → 0.3.65

package/dist/src/sinks/http-usage-sink.test.js

@@ -0,0 +1,105 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { HttpUsageSink } from './http-usage-sink.js';
+function captureFetch(status = 200) {
+    const calls = [];
+    const fetchMock = vi.fn(async (input, init) => {
+        const url = String(input);
+        const body = typeof init?.body === 'string'
+            ? JSON.parse(init.body)
+            : { reports: [] };
+        const headers = init?.headers;
+        const authorization = headers?.['Authorization'] ?? '';
+        calls.push({ url, body, authorization });
+        return new Response(null, { status });
+    });
+    vi.stubGlobal('fetch', fetchMock);
+    return { fetchMock, calls };
+}
+function makeReport(scopeId) {
+    return {
+        sessionId: 'session-1',
+        scopeId,
+        model: 'claude-sonnet-4-20250514',
+        taskAgentRuns: 0,
+        tokens: { inputTokens: 100, outputTokens: 50, cachedTokens: 0 },
+    };
+}
+describe('HttpUsageSink', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+        vi.unstubAllGlobals();
+    });
+    it('substitutes {agentId} in the URL template and sends with bearer auth', async () => {
+        const { calls } = captureFetch();
+        const sink = new HttpUsageSink({
+            url: 'https://api.example.com/api/agents/{agentId}/usage',
+            getBearerToken: () => 'deploy-jwt',
+            maxBatchSize: 1,
+        });
+        sink.log('agent-1', makeReport('end-user-x'));
+        await vi.runOnlyPendingTimersAsync();
+        expect(calls).toHaveLength(1);
+        expect(calls[0]?.url).toBe('https://api.example.com/api/agents/agent-1/usage');
+        expect(calls[0]?.authorization).toBe('Bearer deploy-jwt');
+        expect(calls[0]?.body.reports).toHaveLength(1);
+        expect(calls[0]?.body.reports[0]?.scopeId).toBe('end-user-x');
+        await sink.shutdown();
+    });
+    it('batches reports per agent regardless of scope', async () => {
+        const { calls } = captureFetch();
+        const sink = new HttpUsageSink({
+            url: 'https://api.example.com/usage',
+            getBearerToken: () => 'jwt',
+            maxBatchSize: 100,
+        });
+        sink.log('agent-1', makeReport('user-a'));
+        sink.log('agent-1', makeReport('user-b'));
+        sink.log('agent-1', makeReport(''));
+        await sink.flush();
+        expect(calls).toHaveLength(1);
+        expect(calls[0]?.body.reports.map((r) => r.scopeId).sort()).toEqual([
+            '',
+            'user-a',
+            'user-b',
+        ]);
+        await sink.shutdown();
+    });
+    it('keeps batches separate per agent', async () => {
+        const { calls } = captureFetch();
+        const sink = new HttpUsageSink({
+            url: 'https://api.example.com/api/agents/{agentId}/usage',
+            getBearerToken: () => 'jwt',
+            maxBatchSize: 100,
+        });
+        sink.log('agent-1', makeReport('user-a'));
+        sink.log('agent-2', makeReport('user-a'));
+        await sink.flush();
+        expect(calls).toHaveLength(2);
+        const urls = calls.map((c) => c.url).sort();
+        expect(urls).toEqual([
+            'https://api.example.com/api/agents/agent-1/usage',
+            'https://api.example.com/api/agents/agent-2/usage',
+        ]);
+        await sink.shutdown();
+    });
+    it('does not throw when fetch rejects (fire-and-forget)', async () => {
+        vi.stubGlobal('fetch', vi.fn(() => Promise.reject(new Error('network down'))));
+        const sink = new HttpUsageSink({
+            url: 'https://api.example.com/usage',
+            getBearerToken: () => 'jwt',
+            maxBatchSize: 1,
+        });
+        sink.log('agent-1', makeReport('user-a'));
+        await expect(sink.flush()).resolves.toBeUndefined();
+        await sink.shutdown();
+    });
+});
+//# sourceMappingURL=http-usage-sink.test.js.map

package/dist/src/sinks/http-usage-sink.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-usage-sink.test.js","sourceRoot":"","sources":["../../../src/sinks/http-usage-sink.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAC,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAC,MAAM,QAAQ,CAAC;AACvE,OAAO,EAAC,aAAa,EAAC,MAAM,sBAAsB,CAAC;AAanD,SAAS,YAAY,CACnB,MAAM,GAAG,GAAG;IAEZ,MAAM,KAAK,GAAgB,EAAE,CAAC;IAC9B,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAmB,EAAE,IAAkB,EAAE,EAAE;QACxE,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,IAAI,GACR,OAAO,IAAI,EAAE,IAAI,KAAK,QAAQ;YAC5B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;YACvB,CAAC,CAAC,EAAC,OAAO,EAAE,EAAE,EAAC,CAAC;QACpB,MAAM,OAAO,GAAG,IAAI,EAAE,OAA6C,CAAC;QACpE,MAAM,aAAa,GAAG,OAAO,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QACvD,KAAK,CAAC,IAAI,CAAC,EAAC,GAAG,EAAE,IAAI,EAAE,aAAa,EAAC,CAAC,CAAC;QACvC,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAC,MAAM,EAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAClC,OAAO,EAAC,SAAS,EAAE,KAAK,EAAC,CAAC;AAC5B,CAAC;AAED,SAAS,UAAU,CAAC,OAAe;IACjC,OAAO;QACL,SAAS,EAAE,WAAW;QACtB,OAAO;QACP,KAAK,EAAE,0BAA0B;QACjC,aAAa,EAAE,CAAC;QAChB,MAAM,EAAE,EAAC,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAC;KAC9D,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,EAAC,KAAK,EAAC,GAAG,YAAY,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC;YAC7B,GAAG,EAAE,oDAAoD;YACzD,cAAc,EAAE,GAAG,EAAE,CAAC,YAAY;YAClC,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;QAC9C,MAAM,EAAE,CAAC,yBAAyB,EAAE,CAAC;QAErC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CACxB,kDAAkD,CACnD,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAE9D,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAC,KAAK,EAAC,GAAG,YAAY,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC;YAC7B,GAAG,EAAE,+BAA+B;YACpC,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,YAAY,EAAE,GAAG;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAEnB,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;YAClE,EAAE;YACF,QAAQ;YACR,QAAQ;SACT,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAC,KAAK,EAAC,GAAG,YAAY,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC;YAC7B,GAAG,EAAE,oDAAoD;YACzD,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,YAAY,EAAE,GAAG;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAEnB,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;YACnB,kDAAkD;YAClD,kDAAkD;SACnD,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,EAAE,CAAC,UAAU,CACX,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CACvD,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC;YAC7B,GAAG,EAAE,+BAA+B;YACpC,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,MAAM,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAEpD,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}