agent-orchestrator/tests/e2e/specs/features.test.ts

import { browser, expect } from '@wdio/globals';

/** Reset UI to home state (close any open panels/overlays). */
async function resetToHomeState(): Promise<void> {
  const settingsPanel = await browser.$('.settings-panel');
  if (await settingsPanel.isExisting()) {
    const closeBtn = await browser.$('.settings-close');
    if (await closeBtn.isExisting()) await closeBtn.click();
  }
  const overlay = await browser.$('.search-overlay');
  if (await overlay.isExisting()) await browser.keys('Escape');
}

/** Open the settings panel, waiting for content to render. */
async function openSettings(): Promise<void> {
  const panel = await browser.$('.sidebar-panel');
  const isOpen = await panel.isDisplayed().catch(() => false);
  if (!isOpen) {
    await browser.execute(() => {
      const btn = document.querySelector('[data-testid="settings-btn"]');
      if (btn) (btn as HTMLElement).click();
    });
    await panel.waitForDisplayed({ timeout: 5000 });
  }
  await browser.waitUntil(
    async () => {
      const count = await browser.execute(() =>
        document.querySelectorAll('.settings-tab .settings-section').length,
      );
      return (count as number) >= 1;
    },
    { timeout: 5000, timeoutMsg: 'Settings sections did not render within 5s' },
  );
  await browser.pause(200);
}

/** Close the settings panel if open. */
async function closeSettings(): Promise<void> {
  const panel = await browser.$('.sidebar-panel');
  if (await panel.isDisplayed().catch(() => false)) {
    await browser.execute(() => {
      const btn = document.querySelector('.panel-close');
      if (btn) (btn as HTMLElement).click();
    });
    await browser.pause(500);
  }
}

/** Open command palette — idempotent (won't toggle-close if already open). */
async function openCommandPalette(): Promise<void> {
  // Ensure sidebar is closed first (it can intercept keyboard events)
  await closeSettings();

  // Check if already open
  const alreadyOpen = await browser.execute(() => {
    const p = document.querySelector('.palette');
    return p !== null && getComputedStyle(p).display !== 'none';
  });
  if (alreadyOpen) return;

  // Dispatch Ctrl+K via JS for reliability with WebKit2GTK/tauri-driver
  await browser.execute(() => {
    document.dispatchEvent(new KeyboardEvent('keydown', {
      key: 'k', code: 'KeyK', ctrlKey: true, bubbles: true, cancelable: true,
    }));
  });
  await browser.pause(300);

  const palette = await browser.$('.palette');
  await palette.waitForDisplayed({ timeout: 5000 });
}

/** Close command palette if open — uses backdrop click (more reliable than Escape). */
async function closeCommandPalette(): Promise<void> {
  const isOpen = await browser.execute(() => {
    const p = document.querySelector('.palette');
    return p !== null && getComputedStyle(p).display !== 'none';
  });
  if (!isOpen) return;

  // Click backdrop to close (more reliable than dispatching Escape)
  await browser.execute(() => {
    const backdrop = document.querySelector('.palette-backdrop');
    if (backdrop) (backdrop as HTMLElement).click();
  });
  await browser.pause(500);
}

describe('BTerminal — Command Palette', () => {
  beforeEach(async () => {
    await closeCommandPalette();
  });

  it('should show palette input', async () => {
    await openCommandPalette();

    const input = await browser.$('.palette-input');
    await expect(input).toBeDisplayed();

    // Verify input accepts text (functional focus test, not activeElement check
    // which is unreliable in WebKit2GTK/tauri-driver)
    const canType = await browser.execute(() => {
      const el = document.querySelector('.palette-input') as HTMLInputElement | null;
      if (!el) return false;
      el.focus();
      return el === document.activeElement;
    });
    expect(canType).toBe(true);

    await closeCommandPalette();
  });

  it('should show palette items with command labels and categories', async () => {
    await openCommandPalette();

    const items = await browser.$$('.palette-item');
    expect(items.length).toBeGreaterThanOrEqual(1);

    // Each command item should have a label
    const cmdLabel = await browser.$('.palette-item .cmd-label');
    await expect(cmdLabel).toBeDisplayed();
    const labelText = await cmdLabel.getText();
    expect(labelText.length).toBeGreaterThan(0);

    // Commands should be grouped under category headers
    const categories = await browser.$$('.palette-category');
    expect(categories.length).toBeGreaterThanOrEqual(1);

    await closeCommandPalette();
  });

  it('should highlight selected item in palette', async () => {
    await openCommandPalette();

    // First item should be selected by default
    const selectedItem = await browser.$('.palette-item.selected');
    await expect(selectedItem).toBeExisting();

    await closeCommandPalette();
  });

  it('should filter palette items by typing', async () => {
    await openCommandPalette();

    const itemsBefore = await browser.$$('.palette-item');
    const countBefore = itemsBefore.length;

    // Type a nonsense string that won't match any group name
    const input = await browser.$('.palette-input');
    await input.setValue('zzz_nonexistent_group_xyz');
    await browser.pause(300);

    // Should show no results or fewer items
    const noResults = await browser.$('.no-results');
    const itemsAfter = await browser.$$('.palette-item');
    // Either no-results message appears OR item count decreased
    const filtered = (await noResults.isExisting()) || itemsAfter.length < countBefore;
    expect(filtered).toBe(true);

    await closeCommandPalette();
  });

  it('should close palette by clicking backdrop', async () => {
    await openCommandPalette();
    const palette = await browser.$('.palette');

    // Click the backdrop (outside the palette)
    await browser.execute(() => {
      const backdrop = document.querySelector('.palette-backdrop');
      if (backdrop) (backdrop as HTMLElement).click();
    });
    await browser.pause(500);

    await expect(palette).not.toBeDisplayed();
  });
});

describe('BTerminal — Terminal Tabs', () => {
  before(async () => {
    await resetToHomeState();
    // Ensure Claude tab is active so terminal section is visible
    await browser.execute(() => {
      const tab = document.querySelector('.project-box .ptab');
      if (tab) (tab as HTMLElement).click();
    });
    await browser.pause(300);
  });

  it('should show terminal toggle on Claude tab', async () => {
    const toggle = await browser.$('.terminal-toggle');
    await expect(toggle).toBeDisplayed();

    const label = await browser.$('.toggle-label');
    const text = await label.getText();
    expect(text.toLowerCase()).toContain('terminal');
  });

  it('should expand terminal area on toggle click', async () => {
    // Click terminal toggle via JS
    await browser.execute(() => {
      const toggle = document.querySelector('.terminal-toggle');
      if (toggle) (toggle as HTMLElement).click();
    });
    await browser.pause(500);

    const termArea = await browser.$('.project-terminal-area');
    await expect(termArea).toBeDisplayed();

    // Chevron should have expanded class
    const chevron = await browser.$('.toggle-chevron.expanded');
    await expect(chevron).toBeExisting();
  });

  it('should show add tab button when terminal expanded', async () => {
    const addBtn = await browser.$('.tab-add');
    await expect(addBtn).toBeDisplayed();
  });

  it('should add a shell tab', async () => {
    // Click add tab button via JS (Svelte onclick)
    await browser.execute(() => {
      const btn = document.querySelector('.tab-bar .tab-add');
      if (btn) (btn as HTMLElement).click();
    });
    await browser.pause(500);

    // Verify tab title via JS to avoid stale element issues
    const title = await browser.execute(() => {
      const el = document.querySelector('.tab-bar .tab-title');
      return el ? el.textContent : '';
    });
    expect((title as string).toLowerCase()).toContain('shell');
  });

  it('should show active tab styling', async () => {
    const activeTab = await browser.$('.tab.active');
    await expect(activeTab).toBeExisting();
  });

  it('should add a second shell tab and switch between them', async () => {
    // Add second tab via JS
    await browser.execute(() => {
      const btn = document.querySelector('.tab-bar .tab-add');
      if (btn) (btn as HTMLElement).click();
    });
    await browser.pause(500);

    const tabCount = await browser.execute(() => {
      return document.querySelectorAll('.tab-bar .tab').length;
    });
    expect(tabCount as number).toBeGreaterThanOrEqual(2);

    // Click first tab and verify it becomes active with Shell title
    await browser.execute(() => {
      const tabs = document.querySelectorAll('.tab-bar .tab');
      if (tabs[0]) (tabs[0] as HTMLElement).click();
    });
    await browser.pause(300);

    const activeTitle = await browser.execute(() => {
      const active = document.querySelector('.tab-bar .tab.active .tab-title');
      return active ? active.textContent : '';
    });
    expect(activeTitle as string).toContain('Shell');
  });

  it('should close a tab', async () => {
    const tabsBefore = await browser.$$('.tab');
    const countBefore = tabsBefore.length;

    // Close the last tab
    await browser.execute(() => {
      const closeBtns = document.querySelectorAll('.tab-close');
      if (closeBtns.length > 0) {
        (closeBtns[closeBtns.length - 1] as HTMLElement).click();
      }
    });
    await browser.pause(500);

    const tabsAfter = await browser.$$('.tab');
    expect(tabsAfter.length).toBe(Number(countBefore) - 1);
  });

  after(async () => {
    // Clean up: close remaining tabs and collapse terminal
    await browser.execute(() => {
      // Close all tabs
      const closeBtns = document.querySelectorAll('.tab-close');
      closeBtns.forEach(btn => (btn as HTMLElement).click());
    });
    await browser.pause(300);

    // Collapse terminal
    await browser.execute(() => {
      const toggle = document.querySelector('.terminal-toggle');
      if (toggle) {
        const chevron = toggle.querySelector('.toggle-chevron.expanded');
        if (chevron) (toggle as HTMLElement).click();
      }
    });
    await browser.pause(300);
  });
});

describe('BTerminal — Theme Switching', () => {
  before(async () => {
    await resetToHomeState();
    await openSettings();
    // Scroll to top for theme dropdown
    await browser.execute(() => {
      const content = document.querySelector('.panel-content') || document.querySelector('.sidebar-panel');
      if (content) content.scrollTop = 0;
    });
    await browser.pause(300);
  });

  after(async () => {
    await closeSettings();
  });

  it('should show theme dropdown with group labels', async () => {
    // Close any open dropdowns first
    await browser.execute(() => {
      const openMenu = document.querySelector('.dropdown-menu');
      if (openMenu) {
        const trigger = openMenu.closest('.custom-dropdown')?.querySelector('.dropdown-trigger');
        if (trigger) (trigger as HTMLElement).click();
      }
    });
    await browser.pause(200);

    // Click the first dropdown trigger (theme dropdown)
    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(500);

    const menu = await browser.$('.dropdown-menu');
    await menu.waitForExist({ timeout: 5000 });

    // Should have group labels (Catppuccin, Editor, Deep Dark)
    const groupLabels = await browser.$$('.dropdown-group-label');
    expect(groupLabels.length).toBeGreaterThanOrEqual(2);

    // Close dropdown
    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(300);
  });

  it('should switch theme and update CSS variables', async () => {
    // Get current base color
    const baseBefore = await browser.execute(() => {
      return getComputedStyle(document.documentElement).getPropertyValue('--ctp-base').trim();
    });

    // Open theme dropdown (first custom-dropdown in settings)
    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(500);

    // Wait for dropdown menu
    const menu = await browser.$('.dropdown-menu');
    await menu.waitForExist({ timeout: 5000 });

    // Click the first non-active theme option
    const changed = await browser.execute(() => {
      const options = document.querySelectorAll('.dropdown-menu .dropdown-option:not(.active)');
      if (options.length > 0) {
        (options[0] as HTMLElement).click();
        return true;
      }
      return false;
    });
    expect(changed).toBe(true);
    await browser.pause(500);

    // Verify CSS variable changed
    const baseAfter = await browser.execute(() => {
      return getComputedStyle(document.documentElement).getPropertyValue('--ctp-base').trim();
    });
    expect(baseAfter).not.toBe(baseBefore);

    // Switch back to Catppuccin Mocha (first option) to restore state
    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(500);
    await browser.execute(() => {
      const options = document.querySelectorAll('.dropdown-menu .dropdown-option');
      if (options.length > 0) (options[0] as HTMLElement).click();
    });
    await browser.pause(300);
  });

  it('should show active theme option', async () => {
    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(500);

    const menu = await browser.$('.dropdown-menu');
    await menu.waitForExist({ timeout: 5000 });

    const activeOption = await browser.$('.dropdown-option.active');
    await expect(activeOption).toBeExisting();

    await browser.execute(() => {
      const trigger = document.querySelector('.settings-tab .custom-dropdown .dropdown-trigger');
      if (trigger) (trigger as HTMLElement).click();
    });
    await browser.pause(300);
  });
});

describe('BTerminal — Keyboard Shortcuts', () => {
  before(async () => {
    await resetToHomeState();
    await closeSettings();
    await closeCommandPalette();
  });

  it('should open command palette with Ctrl+K', async () => {
    await openCommandPalette();

    const input = await browser.$('.palette-input');
    await expect(input).toBeDisplayed();

    // Close with Escape
    await closeCommandPalette();
    const palette = await browser.$('.palette');
    const isGone = !(await palette.isDisplayed().catch(() => false));
    expect(isGone).toBe(true);
  });

  it('should toggle settings with Ctrl+,', async () => {
    await browser.keys(['Control', ',']);

    const panel = await browser.$('.sidebar-panel');
    await panel.waitForDisplayed({ timeout: 3000 });

    // Close with Ctrl+,
    await browser.keys(['Control', ',']);
    await panel.waitForDisplayed({ timeout: 3000, reverse: true });
  });

  it('should toggle sidebar with Ctrl+B', async () => {
    // Open sidebar first
    await browser.keys(['Control', ',']);
    const panel = await browser.$('.sidebar-panel');
    await panel.waitForDisplayed({ timeout: 3000 });

    // Toggle off with Ctrl+B
    await browser.keys(['Control', 'b']);
    await panel.waitForDisplayed({ timeout: 3000, reverse: true });
  });

  it('should close sidebar with Escape', async () => {
    // Open sidebar
    await browser.keys(['Control', ',']);
    const panel = await browser.$('.sidebar-panel');
    await panel.waitForDisplayed({ timeout: 3000 });

    // Close with Escape
    await browser.keys('Escape');
    await panel.waitForDisplayed({ timeout: 3000, reverse: true });
  });

  it('should show command palette with categorized commands', async () => {
    await openCommandPalette();

    const items = await browser.$$('.palette-item');
    expect(items.length).toBeGreaterThanOrEqual(1);

    // Commands should have labels
    const cmdLabel = await browser.$('.palette-item .cmd-label');
    await expect(cmdLabel).toBeDisplayed();

    await closeCommandPalette();
  });
});