cc-haha/src/server/__tests__/computer-use-python.test.ts

import { describe, expect, test } from 'bun:test'
import { detectPythonRuntime, isPythonVersionAtLeast } from '../api/computer-use-python.js'

describe('isPythonVersionAtLeast', () => {
  test('accepts supported Python 3.9+ versions', () => {
    expect(isPythonVersionAtLeast('3.9.0', 3, 9)).toBe(true)
    expect(isPythonVersionAtLeast('3.12.11', 3, 9)).toBe(true)
  })

  test('rejects missing or older Python versions', () => {
    expect(isPythonVersionAtLeast(null, 3, 9)).toBe(false)
    expect(isPythonVersionAtLeast('3.8.18', 3, 9)).toBe(false)
    expect(isPythonVersionAtLeast('2.7.18', 3, 9)).toBe(false)
  })
})

describe('detectPythonRuntime', () => {
  test('prefers python3 on Windows when available', async () => {
    const calls: string[] = []
    const result = await detectPythonRuntime(
      'win32',
      async (cmd, args) => {
        calls.push(`${cmd} ${args.join(' ')}`.trim())
        if (cmd === 'python3' && args.join(' ') === '--version') {
          return { ok: true, stdout: 'Python 3.12.11', stderr: '', code: 0 }
        }
        if (cmd === 'where' && args[0] === 'python3') {
          return { ok: true, stdout: 'C:\\Python312\\python3.exe', stderr: '', code: 0 }
        }
        return { ok: false, stdout: '', stderr: '', code: 1 }
      },
    )

    expect(result.installed).toBe(true)
    expect(result.version).toBe('3.12.11')
    expect(result.path).toBe('C:\\Python312\\python3.exe')
    expect(result.command).toBe('python3')
    expect(result.prefixArgs).toEqual([])
    expect(result.source).toBe('system')
    expect(result.error).toBeNull()
    expect(calls).toEqual(['python3 --version', 'where python3'])
  })

  test('uses a custom Python path before PATH candidates', async () => {
    const calls: string[] = []
    const customPython = 'C:\\Users\\me\\miniconda3\\envs\\cu\\python.exe'
    const result = await detectPythonRuntime(
      'win32',
      async (cmd, args) => {
        calls.push(`${cmd} ${args.join(' ')}`.trim())
        if (cmd === customPython && args.join(' ') === '--version') {
          return { ok: true, stdout: 'Python 3.11.9', stderr: '', code: 0 }
        }
        return { ok: false, stdout: '', stderr: '', code: 1 }
      },
      undefined,
      customPython,
    )

    expect(result.installed).toBe(true)
    expect(result.version).toBe('3.11.9')
    expect(result.path).toBe(customPython)
    expect(result.command).toBe(customPython)
    expect(result.prefixArgs).toEqual([])
    expect(result.source).toBe('custom')
    expect(result.error).toBeNull()
    expect(calls).toEqual([`${customPython} --version`])
  })

  test('reports an invalid custom Python path without falling back to PATH', async () => {
    const calls: string[] = []
    const result = await detectPythonRuntime(
      'win32',
      async (cmd, args) => {
        calls.push(`${cmd} ${args.join(' ')}`.trim())
        if (cmd === 'C:\\missing\\python.exe') {
          return { ok: false, stdout: '', stderr: 'not found', code: 1 }
        }
        if (cmd === 'python') {
          return { ok: true, stdout: 'Python 3.12.0', stderr: '', code: 0 }
        }
        return { ok: false, stdout: '', stderr: '', code: 1 }
      },
      undefined,
      'C:\\missing\\python.exe',
    )

    expect(result.installed).toBe(false)
    expect(result.path).toBe('C:\\missing\\python.exe')
    expect(result.command).toBe('C:\\missing\\python.exe')
    expect(result.source).toBe('custom')
    expect(result.error).toBe('not found')
    expect(calls).toEqual(['C:\\missing\\python.exe --version'])
  })

  test('falls back to py -3 on Windows', async () => {
    const result = await detectPythonRuntime(
      'win32',
      async (cmd, args) => {
        if (cmd === 'python3' || cmd === 'python') {
          return { ok: false, stdout: '', stderr: '', code: 1 }
        }
        if (cmd === 'py' && args.join(' ') === '-3 --version') {
          return { ok: true, stdout: '', stderr: 'Python 3.12.11', code: 0 }
        }
        if (cmd === 'where' && args[0] === 'py') {
          return { ok: true, stdout: 'C:\\Windows\\py.exe', stderr: '', code: 0 }
        }
        return { ok: false, stdout: '', stderr: '', code: 1 }
      },
    )

    expect(result.installed).toBe(true)
    expect(result.version).toBe('3.12.11')
    expect(result.path).toBe('C:\\Windows\\py.exe')
    expect(result.command).toBe('py')
    expect(result.prefixArgs).toEqual(['-3'])
    expect(result.source).toBe('system')
    expect(result.error).toBeNull()
  })

  test('falls back to venv python when system python is not discoverable', async () => {
    const venvPython = 'C:\\Users\\Relakkes\\.claude\\.runtime\\venv\\Scripts\\python.exe'
    const result = await detectPythonRuntime(
      'win32',
      async (cmd, args) => {
        if (cmd === venvPython && args.join(' ') === '--version') {
          return { ok: true, stdout: 'Python 3.12.11', stderr: '', code: 0 }
        }
        return { ok: false, stdout: '', stderr: '', code: 1 }
      },
      venvPython,
    )

    expect(result.installed).toBe(true)
    expect(result.version).toBe('3.12.11')
    expect(result.path).toBe(venvPython)
    expect(result.command).toBe(venvPython)
    expect(result.prefixArgs).toEqual([])
    expect(result.source).toBe('venv')
    expect(result.error).toBeNull()
  })
})