@terminai/a2a-server 0.21.0

package/dist/src/utils/testing_utils.js

@@ -0,0 +1,173 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ApprovalMode, DEFAULT_GEMINI_MODEL, DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES, DEFAULT_TRUNCATE_TOOL_OUTPUT_THRESHOLD, GeminiClient, HookSystem, } from '@terminai/core';
+import { createMockMessageBus } from '@terminai/core/src/test-utils/mock-message-bus.js';
+import { expect, vi } from 'vitest';
+import crypto from 'node:crypto';
+import net from 'node:net';
+import { buildSignaturePayload, computeBodyHash } from '../http/replay.js';
+export const TEST_REMOTE_TOKEN = 'test-remote-token';
+let canListenCache;
+export async function canListenOnLocalhost() {
+    if (!canListenCache) {
+        canListenCache = new Promise((resolve) => {
+            try {
+                const server = net.createServer();
+                server.once('error', () => resolve(false));
+                server.listen(0, '127.0.0.1', () => {
+                    server.close(() => resolve(true));
+                });
+            }
+            catch (_error) {
+                resolve(false);
+            }
+        });
+    }
+    return canListenCache;
+}
+export async function listenOnLocalhost(app) {
+    return new Promise((resolve, reject) => {
+        const server = app.listen(0, '127.0.0.1', () => resolve(server));
+        server.on('error', (err) => reject(err));
+    });
+}
+export async function closeServer(server) {
+    await new Promise((resolve, reject) => {
+        server.close((err) => {
+            if (err) {
+                reject(err);
+                return;
+            }
+            resolve();
+        });
+    });
+}
+export function createMockConfig(overrides = {}) {
+    const mockConfig = {
+        getToolRegistry: vi.fn().mockReturnValue({
+            getTool: vi.fn(),
+            getAllToolNames: vi.fn().mockReturnValue([]),
+            getAllTools: vi.fn().mockReturnValue([]),
+            getToolsByServer: vi.fn().mockReturnValue([]),
+        }),
+        getApprovalMode: vi.fn().mockReturnValue(ApprovalMode.DEFAULT),
+        getIdeMode: vi.fn().mockReturnValue(false),
+        isInteractive: () => true,
+        getAllowedTools: vi.fn().mockReturnValue([]),
+        getWorkspaceContext: vi.fn().mockReturnValue({
+            isPathWithinWorkspace: () => true,
+        }),
+        getTargetDir: () => '/test',
+        getCheckpointingEnabled: vi.fn().mockReturnValue(false),
+        storage: {
+            getProjectTempDir: () => '/tmp',
+            getProjectTempCheckpointsDir: () => '/tmp/checkpoints',
+        },
+        getTruncateToolOutputThreshold: () => DEFAULT_TRUNCATE_TOOL_OUTPUT_THRESHOLD,
+        getTruncateToolOutputLines: () => DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES,
+        getActiveModel: vi.fn().mockReturnValue(DEFAULT_GEMINI_MODEL),
+        getDebugMode: vi.fn().mockReturnValue(false),
+        getContentGeneratorConfig: vi.fn().mockReturnValue({ model: 'gemini-pro' }),
+        getModel: vi.fn().mockReturnValue('gemini-pro'),
+        getUsageStatisticsEnabled: vi.fn().mockReturnValue(false),
+        setFallbackModelHandler: vi.fn(),
+        initialize: vi.fn().mockResolvedValue(undefined),
+        getProxy: vi.fn().mockReturnValue(undefined),
+        getHistory: vi.fn().mockReturnValue([]),
+        getEmbeddingModel: vi.fn().mockReturnValue('text-embedding-004'),
+        getSessionId: vi.fn().mockReturnValue('test-session-id'),
+        getUserTier: vi.fn(),
+        getEnableMessageBusIntegration: vi.fn().mockReturnValue(false),
+        getMessageBus: vi.fn(),
+        getPolicyEngine: vi.fn(),
+        getEnableExtensionReloading: vi.fn().mockReturnValue(false),
+        getEnableHooks: vi.fn().mockReturnValue(false),
+        getMcpClientManager: vi.fn().mockReturnValue({
+            getMcpServers: vi.fn().mockReturnValue({}),
+        }),
+        getGitService: vi.fn(),
+        ...overrides,
+    };
+    mockConfig.getMessageBus = vi.fn().mockReturnValue(createMockMessageBus());
+    mockConfig.getHookSystem = vi
+        .fn()
+        .mockReturnValue(new HookSystem(mockConfig));
+    mockConfig.getGeminiClient = vi
+        .fn()
+        .mockReturnValue(new GeminiClient(mockConfig));
+    return mockConfig;
+}
+export function createStreamMessageRequest(text, messageId, taskId) {
+    const request = {
+        jsonrpc: '2.0',
+        id: '1',
+        method: 'message/stream',
+        params: {
+            message: {
+                kind: 'message',
+                role: 'user',
+                parts: [{ kind: 'text', text }],
+                messageId,
+            },
+            metadata: {
+                coderAgent: {
+                    kind: 'agent-settings',
+                    workspacePath: '/tmp',
+                },
+            },
+        },
+    };
+    if (taskId) {
+        request.params.taskId = taskId;
+    }
+    return request;
+}
+export function createAuthHeader(token = TEST_REMOTE_TOKEN) {
+    return { Authorization: `Bearer ${token}` };
+}
+export function createSignedHeaders(method, path, body, token = TEST_REMOTE_TOKEN, nonce = crypto.randomUUID()) {
+    const rawBody = body ? JSON.stringify(body) : '';
+    const bodyHash = computeBodyHash(rawBody);
+    const payload = buildSignaturePayload({
+        method,
+        path,
+        bodyHash,
+        nonce,
+    });
+    const signature = crypto
+        .createHmac('sha256', token)
+        .update(payload)
+        .digest('hex');
+    return {
+        Authorization: `Bearer ${token}`,
+        'X-Gemini-Nonce': nonce,
+        'X-Gemini-Signature': signature,
+    };
+}
+export function assertUniqueFinalEventIsLast(events) {
+    // Final event is input-required & final
+    const finalEvent = events[events.length - 1].result;
+    expect(finalEvent.metadata?.['coderAgent']).toMatchObject({
+        kind: 'state-change',
+    });
+    expect(finalEvent.status?.state).toBe('input-required');
+    expect(finalEvent.final).toBe(true);
+    // There is only one event with final and its the last
+    expect(events.filter((e) => e.result.final).length).toBe(1);
+    expect(events.findIndex((e) => e.result.final)).toBe(events.length - 1);
+}
+export function assertTaskCreationAndWorkingStatus(events) {
+    // Initial task creation event
+    const taskEvent = events[0].result;
+    expect(taskEvent.kind).toBe('task');
+    expect(taskEvent.status.state).toBe('submitted');
+    // Status update: working
+    const workingEvent = events[1].result;
+    expect(workingEvent.kind).toBe('status-update');
+    expect(workingEvent.status.state).toBe('working');
+}
+//# sourceMappingURL=testing_utils.js.map

package/dist/src/utils/testing_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testing_utils.js","sourceRoot":"","sources":["../../../src/utils/testing_utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH,OAAO,EACL,YAAY,EACZ,oBAAoB,EACpB,kCAAkC,EAClC,sCAAsC,EACtC,YAAY,EACZ,UAAU,GACX,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,mDAAmD,CAAC;AAIzF,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,GAAG,MAAM,UAAU,CAAC;AAC3B,OAAO,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAE3E,MAAM,CAAC,MAAM,iBAAiB,GAAG,mBAAmB,CAAC;AAErD,IAAI,cAA4C,CAAC;AAEjD,MAAM,CAAC,KAAK,UAAU,oBAAoB;IACxC,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACvC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;oBACjC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpC,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC;gBAChB,OAAO,CAAC,KAAK,CAAC,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,GAAoB;IAC1D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,MAAc;IAC9C,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACnB,IAAI,GAAG,EAAE,CAAC;gBACR,MAAM,CAAC,GAAG,CAAC,CAAC;gBACZ,OAAO;YACT,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,YAA6B,EAAE;IAE/B,MAAM,UAAU,GAAG;QACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACvC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC5C,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YACxC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;SAC9C,CAAC;QACF,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC;QAC9D,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC1C,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;QACzB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QAC5C,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC3C,qBAAqB,EAAE,GAAG,EAAE,CAAC,IAAI;SAClC,CAAC;QACF,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO;QAC3B,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QACvD,OAAO,EAAE;YACP,iBAAiB,EAAE,GAAG,EAAE,CAAC,MAAM;YAC/B,4BAA4B,EAAE,GAAG,EAAE,CAAC,kBAAkB;SAC5C;QACZ,8BAA8B,EAAE,GAAG,EAAE,CACnC,sCAAsC;QACxC,0BAA0B,EAAE,GAAG,EAAE,CAAC,kCAAkC;QACpE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,CAAC;QAC7D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC5C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QAC3E,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC;QAC/C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QACzD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;QAChC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAChD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;QAC5C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QACvC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,CAAC;QAChE,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;QACxD,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;QACpB,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC9D,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC3D,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC9C,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC3C,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;SAC3C,CAAC;QACF,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,GAAG,SAAS;KACQ,CAAC;IACvB,UAAU,CAAC,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,EAAE,CAAC,CAAC;IAC3E,UAAU,CAAC,aAAa,GAAG,EAAE;SAC1B,EAAE,EAAE;SACJ,eAAe,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;IAE/C,UAAU,CAAC,eAAe,GAAG,EAAE;SAC5B,EAAE,EAAE;SACJ,eAAe,CAAC,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC;IACjD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,IAAY,EACZ,SAAiB,EACjB,MAAe;IAEf,MAAM,OAAO,GAmBT;QACF,OAAO,EAAE,KAAK;QACd,EAAE,EAAE,GAAG;QACP,MAAM,EAAE,gBAAgB;QACxB,MAAM,EAAE;YACN,OAAO,EAAE;gBACP,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBAC/B,SAAS;aACV;YACD,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV,IAAI,EAAE,gBAAgB;oBACtB,aAAa,EAAE,MAAM;iBACtB;aACF;SACF;KACF,CAAC;IAEF,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;IACjC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAgB,iBAAiB;IAChE,OAAO,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,IAAY,EACZ,IAAa,EACb,QAAgB,iBAAiB,EACjC,QAAgB,MAAM,CAAC,UAAU,EAAE;IAEnC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACjD,MAAM,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAG,qBAAqB,CAAC;QACpC,MAAM;QACN,IAAI;QACJ,QAAQ;QACR,KAAK;KACN,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,MAAM;SACrB,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC;SAC3B,MAAM,CAAC,OAAO,CAAC;SACf,MAAM,CAAC,KAAK,CAAC,CAAC;IACjB,OAAO;QACL,aAAa,EAAE,UAAU,KAAK,EAAE;QAChC,gBAAgB,EAAE,KAAK;QACvB,oBAAoB,EAAE,SAAS;KAChC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,MAA6C;IAE7C,wCAAwC;IACxC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAA+B,CAAC;IAC7E,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC;QACxD,IAAI,EAAE,cAAc;KACrB,CAAC,CAAC;IACH,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACxD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEpC,sDAAsD;IACtD,MAAM,CACJ,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,MAAgC,CAAC,KAAK,CAAC,CAAC,MAAM,CACvE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACV,MAAM,CACJ,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,MAAgC,CAAC,KAAK,CAAC,CACnE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,kCAAkC,CAChD,MAA6C;IAE7C,8BAA8B;IAC9B,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAiB,CAAC;IAC9C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAEjD,yBAAyB;IACzB,MAAM,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAA+B,CAAC;IAC/D,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACpD,CAAC"}