kimi-code/packages/agent-core/test/tools/bash-env.test.ts

import { Readable, type Writable } from 'node:stream';

import type { KaosProcess } from '@moonshot-ai/kaos';
import { describe, expect, it, vi } from 'vitest';

import { BashTool } from '../../src/tools/builtin/shell/bash';
import type { Environment } from '../../src/utils/environment';
import { executeTool } from './fixtures/execute-tool';
import { createFakeKaos } from './fixtures/fake-kaos';

const posixEnv: Environment = {
  osKind: 'Linux',
  osArch: 'x86_64',
  osVersion: 'test',
  shellPath: '/bin/bash',
  shellName: 'bash',
};

function fakeProcess(): KaosProcess {
  return {
    stdin: { end: vi.fn(), write: vi.fn() } as unknown as Writable,
    stdout: Readable.from([]),
    stderr: Readable.from([]),
    pid: 401,
    exitCode: 0,
    wait: vi.fn(async () => 0),
    kill: vi.fn(async () => {}),
  };
}

const signal = new AbortController().signal;

async function captureSpawnEnv(): Promise<Record<string, string>> {
  const execWithEnv = vi.fn().mockResolvedValue(fakeProcess());
  const tool = new BashTool(createFakeKaos({ execWithEnv }), '/workspace', posixEnv);
  await executeTool(tool, {
    turnId: '0',
    toolCallId: 'tc_env',
    args: { command: 'true', timeout: 1000 },
    signal,
  });
  return execWithEnv.mock.calls[0]?.[1] as Record<string, string>;
}

describe('BashTool noninteractive env semantics', () => {
  it('preserves an existing GIT_TERMINAL_PROMPT from the ambient env', async () => {
    const previous = process.env['GIT_TERMINAL_PROMPT'];
    process.env['GIT_TERMINAL_PROMPT'] = '1';
    try {
      const env = await captureSpawnEnv();
      expect(env['GIT_TERMINAL_PROMPT']).toBe('1');
    } finally {
      if (previous === undefined) delete process.env['GIT_TERMINAL_PROMPT'];
      else process.env['GIT_TERMINAL_PROMPT'] = previous;
    }
  });

  it('defaults GIT_TERMINAL_PROMPT to "0" when the ambient env does not set it', async () => {
    const previous = process.env['GIT_TERMINAL_PROMPT'];
    delete process.env['GIT_TERMINAL_PROMPT'];
    try {
      const env = await captureSpawnEnv();
      expect(env['GIT_TERMINAL_PROMPT']).toBe('0');
    } finally {
      if (previous !== undefined) process.env['GIT_TERMINAL_PROMPT'] = previous;
    }
  });
});