bugCommand.test.ts

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

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { bugCommand } from './bugCommand.js';
import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
import { GIT_COMMIT_INFO } from '../../generated/git-commit.js';
import { AuthType } from '@qwen-code/qwen-code-core';
import * as systemInfoUtils from '../../utils/systemInfo.js';

const mockOpenBrowserSecurely = vi.hoisted(() => vi.fn());

// Mock dependencies
vi.mock('@qwen-code/qwen-code-core', async (importOriginal) => {
  const actual =
    await importOriginal<typeof import('@qwen-code/qwen-code-core')>();
  return {
    ...actual,
    openBrowserSecurely: mockOpenBrowserSecurely,
  };
});
vi.mock('../../utils/systemInfo.js');

describe('bugCommand', () => {
  beforeEach(() => {
    vi.mocked(systemInfoUtils.getExtendedSystemInfo).mockResolvedValue({
      cliVersion: '0.1.0',
      osPlatform: 'test-platform',
      osArch: 'x64',
      osRelease: '22.0.0',
      nodeVersion: 'v20.0.0',
      npmVersion: '10.0.0',
      sandboxEnv: 'test',
      modelVersion: 'qwen3-coder-plus',
      selectedAuthType: '',
      ideClient: 'VSCode',
      sessionId: 'test-session-id',
      memoryUsage: '100 MB',
      gitCommit:
        GIT_COMMIT_INFO && !['N/A'].includes(GIT_COMMIT_INFO)
          ? GIT_COMMIT_INFO
          : undefined,
    });
    mockOpenBrowserSecurely.mockClear();
    mockOpenBrowserSecurely.mockResolvedValue(undefined);
    vi.stubEnv('SANDBOX', 'qwen-test');
  });

  afterEach(() => {
    vi.unstubAllEnvs();
    vi.clearAllMocks();
  });

  it('should generate the default GitHub issue URL', async () => {
    const mockContext = createMockCommandContext({
      services: {
        config: {
          getBugCommand: () => undefined,
        },
      },
    });

    if (!bugCommand.action) throw new Error('Action is not defined');
    await bugCommand.action(mockContext, 'A test bug');

    const qwenCodeLine =
      GIT_COMMIT_INFO && !['N/A'].includes(GIT_COMMIT_INFO)
        ? `Qwen Code: 0.1.0 (${GIT_COMMIT_INFO})`
        : 'Qwen Code: 0.1.0';
    const expectedInfo = `${qwenCodeLine}
Runtime: Node.js v20.0.0 / npm 10.0.0
IDE Client: VSCode
OS: test-platform x64 (22.0.0)
Model: qwen3-coder-plus
Fast Model: qwen3-coder-plus
Session ID: test-session-id
Sandbox: test
Proxy: no proxy
Memory Usage: 100 MB`;
    const expectedUrl =
      'https://github.com/QwenLM/qwen-code/issues/new?template=bug_report.yml&title=A%20test%20bug&info=%0A' +
      encodeURIComponent(expectedInfo) +
      '%0A';

    expect(mockContext.ui.addItem).toHaveBeenCalledWith(
      {
        type: 'info',
        text: 'To submit your bug report, please open the following URL in your browser:',
        linkUrl: expectedUrl,
        linkText: 'Open GitHub bug report form',
      },
      expect.any(Number),
    );
    expect(mockOpenBrowserSecurely).toHaveBeenCalledWith(expectedUrl);
  });

  it('should use a custom URL template from config if provided', async () => {
    const customTemplate =
      'https://internal.bug-tracker.com/new?desc={title}&details={info}';
    const mockContext = createMockCommandContext({
      services: {
        config: {
          getBugCommand: () => ({ urlTemplate: customTemplate }),
        },
      },
    });

    if (!bugCommand.action) throw new Error('Action is not defined');
    await bugCommand.action(mockContext, 'A custom bug');

    const qwenCodeLine =
      GIT_COMMIT_INFO && !['N/A'].includes(GIT_COMMIT_INFO)
        ? `Qwen Code: 0.1.0 (${GIT_COMMIT_INFO})`
        : 'Qwen Code: 0.1.0';
    const expectedInfo = `${qwenCodeLine}
Runtime: Node.js v20.0.0 / npm 10.0.0
IDE Client: VSCode
OS: test-platform x64 (22.0.0)
Model: qwen3-coder-plus
Fast Model: qwen3-coder-plus
Session ID: test-session-id
Sandbox: test
Proxy: no proxy
Memory Usage: 100 MB`;
    const expectedUrl = customTemplate
      .replace('{title}', encodeURIComponent('A custom bug'))
      .replace('{info}', encodeURIComponent(`\n${expectedInfo}\n`));

    expect(mockContext.ui.addItem).toHaveBeenCalledWith(
      {
        type: 'info',
        text: 'To submit your bug report, please open the following URL in your browser:',
        linkUrl: expectedUrl,
        linkText: 'Open GitHub bug report form',
      },
      expect.any(Number),
    );
    expect(mockOpenBrowserSecurely).toHaveBeenCalledWith(expectedUrl);
  });

  it('should include Base URL when auth type is OpenAI', async () => {
    vi.mocked(systemInfoUtils.getExtendedSystemInfo).mockResolvedValue({
      cliVersion: '0.1.0',
      osPlatform: 'test-platform',
      osArch: 'x64',
      osRelease: '22.0.0',
      nodeVersion: 'v20.0.0',
      npmVersion: '10.0.0',
      sandboxEnv: 'test',
      modelVersion: 'qwen3-coder-plus',
      selectedAuthType: AuthType.USE_OPENAI,
      ideClient: 'VSCode',
      sessionId: 'test-session-id',
      memoryUsage: '100 MB',
      baseUrl: 'https://api.openai.com/v1',
      gitCommit:
        GIT_COMMIT_INFO && !['N/A'].includes(GIT_COMMIT_INFO)
          ? GIT_COMMIT_INFO
          : undefined,
    });

    const mockContext = createMockCommandContext({
      services: {
        config: {
          getBugCommand: () => undefined,
        },
      },
    });

    if (!bugCommand.action) throw new Error('Action is not defined');
    await bugCommand.action(mockContext, 'OpenAI bug');

    const qwenCodeLine =
      GIT_COMMIT_INFO && !['N/A'].includes(GIT_COMMIT_INFO)
        ? `Qwen Code: 0.1.0 (${GIT_COMMIT_INFO})`
        : 'Qwen Code: 0.1.0';
    const expectedInfo = `${qwenCodeLine}
Runtime: Node.js v20.0.0 / npm 10.0.0
IDE Client: VSCode
OS: test-platform x64 (22.0.0)
Auth: API Key - ${AuthType.USE_OPENAI}
Base URL: https://api.openai.com/v1
Model: qwen3-coder-plus
Fast Model: qwen3-coder-plus
Session ID: test-session-id
Sandbox: test
Proxy: no proxy
Memory Usage: 100 MB`;
    const expectedUrl =
      'https://github.com/QwenLM/qwen-code/issues/new?template=bug_report.yml&title=OpenAI%20bug&info=%0A' +
      encodeURIComponent(expectedInfo) +
      '%0A';

    expect(mockContext.ui.addItem).toHaveBeenCalledWith(
      {
        type: 'info',
        text: 'To submit your bug report, please open the following URL in your browser:',
        linkUrl: expectedUrl,
        linkText: 'Open GitHub bug report form',
      },
      expect.any(Number),
    );
    expect(mockOpenBrowserSecurely).toHaveBeenCalledWith(expectedUrl);
  });
});