qwen-code/packages/core/src/ide/detect-ide.test.ts

/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { describe, it, expect, afterEach, vi } from 'vitest';
import { detectIde, DetectedIde } from './detect-ide.js';

describe('detectIde', () => {
  afterEach(() => {
    vi.unstubAllEnvs();
  });

  it.each([
    {
      env: {},
      expected: DetectedIde.VSCode,
    },
    {
      env: { __COG_BASHRC_SOURCED: '1' },
      expected: DetectedIde.Devin,
    },
    {
      env: { REPLIT_USER: 'test' },
      expected: DetectedIde.Replit,
    },
    {
      env: { CURSOR_TRACE_ID: 'test' },
      expected: DetectedIde.Cursor,
    },
    {
      env: { CODESPACES: 'true' },
      expected: DetectedIde.Codespaces,
    },
    {
      env: { EDITOR_IN_CLOUD_SHELL: 'true' },
      expected: DetectedIde.CloudShell,
    },
    {
      env: { CLOUD_SHELL: 'true' },
      expected: DetectedIde.CloudShell,
    },
    {
      env: { TERM_PRODUCT: 'Trae' },
      expected: DetectedIde.Trae,
    },
    {
      env: { FIREBASE_DEPLOY_AGENT: 'true' },
      expected: DetectedIde.FirebaseStudio,
    },
    {
      env: { MONOSPACE_ENV: 'true' },
      expected: DetectedIde.FirebaseStudio,
    },
  ])('detects the IDE for $expected', ({ env, expected }) => {
    // Clear all environment variables first
    vi.unstubAllEnvs();

    // Set TERM_PROGRAM to vscode (required for all IDE detection)
    vi.stubEnv('TERM_PROGRAM', 'vscode');

    // Explicitly stub all environment variables that detectIde() checks to undefined
    // This ensures no real environment variables interfere with the tests
    vi.stubEnv('__COG_BASHRC_SOURCED', undefined);
    vi.stubEnv('REPLIT_USER', undefined);
    vi.stubEnv('CURSOR_TRACE_ID', undefined);
    vi.stubEnv('CODESPACES', undefined);
    vi.stubEnv('EDITOR_IN_CLOUD_SHELL', undefined);
    vi.stubEnv('CLOUD_SHELL', undefined);
    vi.stubEnv('TERM_PRODUCT', undefined);
    vi.stubEnv('FIREBASE_DEPLOY_AGENT', undefined);
    vi.stubEnv('MONOSPACE_ENV', undefined);

    // Set only the specific environment variables for this test case
    for (const [key, value] of Object.entries(env)) {
      vi.stubEnv(key, value);
    }

    expect(detectIde()).toBe(expected);
  });

  it('returns undefined for non-vscode', () => {
    // Clear all environment variables first
    vi.unstubAllEnvs();

    // Set TERM_PROGRAM to something other than vscode
    vi.stubEnv('TERM_PROGRAM', 'definitely-not-vscode');

    expect(detectIde()).toBeUndefined();
  });
});