MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-03-10 22:21:22 -07:00
Code Issues Packages Projects Releases Wiki Activity

test: decouple remote background test from shell service

Browse Source
This commit is contained in:
Adam Weidman
2026-03-09 00:18:38 -04:00
parent 020fcc9674
commit bc76cbeb50
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packages/core/src/agents/remote-invocation.test.ts
View File

@@ -22,7 +22,7 @@ import type { RemoteAgentDefinition } from './types.js';
import { createMockMessageBus } from '../test-utils/mock-message-bus.js';
import { A2AAuthProviderFactory } from './auth-provider/factory.js';
import type { A2AAuthProvider } from './auth-provider/types.js';
import { ShellExecutionService } from '../services/shellExecutionService.js';
import { ExecutionLifecycleService } from '../services/executionLifecycleService.js';
// Mock A2AClientManager
vi.mock('./a2a-client-manager.js', () => ({
@@ -59,6 +59,7 @@ describe('RemoteAgentInvocation', () => {
beforeEach(() => {
vi.clearAllMocks();
ExecutionLifecycleService.resetForTest();
(A2AClientManager.getInstance as Mock).mockReturnValue(mockClientManager);
(
RemoteAgentInvocation as unknown as {
@@ -619,8 +620,8 @@ describe('RemoteAgentInvocation', () => {
const updateOutput = vi.fn((output: unknown) => {
if (output === 'Chunk 1' && executionId !== undefined) {
ShellExecutionService.background(executionId);
unsubscribeStream = ShellExecutionService.subscribe(
ExecutionLifecycleService.background(executionId);
unsubscribeStream = ExecutionLifecycleService.subscribe(
executionId,
(event) => {
if (event.type === 'data' && typeof event.chunk === 'string') {
@@ -643,7 +644,7 @@ describe('RemoteAgentInvocation', () => {
undefined,
(newExecutionId) => {
executionId = newExecutionId;
unsubscribeOnExit = ShellExecutionService.onExit(
unsubscribeOnExit = ExecutionLifecycleService.onExit(
newExecutionId,
onExit,
);