import { describe, it, expect, vi, beforeEach } from 'vitest'; import { render, screen, waitFor, fireEvent } from '@testing-library/react'; import { BottomMenuModeSelection } from './BottomMenuModeSelection'; let mockConfig: Record = {}; const mockUpdateSession = vi.fn().mockResolvedValue({}); const mockGetSession = vi.fn().mockResolvedValue({ data: null }); vi.mock('../ConfigContext', () => ({ useConfig: () => ({ config: mockConfig, }), })); vi.mock('../../utils/analytics', () => ({ trackModeChanged: vi.fn(), })); vi.mock('../../api', () => ({ updateSession: (...args: unknown[]) => mockUpdateSession(...args), getSession: (...args: unknown[]) => mockGetSession(...args), })); // Radix dropdown doesn't open in jsdom — render children directly vi.mock('../ui/dropdown-menu', () => ({ DropdownMenu: ({ children }: { children: React.ReactNode }) =>

{children}

, DropdownMenuTrigger: ({ children }: { children: React.ReactNode }) =>

{children}

, DropdownMenuContent: ({ children }: { children: React.ReactNode }) =>

{children}

, DropdownMenuItem: ({ children }: { children: React.ReactNode }) =>

{children}

, })); describe('BottomMenuModeSelection', () => { beforeEach(() => { vi.clearAllMocks(); mockConfig = {}; }); it('displays mode from config when no session', async () => { mockConfig.GOOSE_MODE = 'approve'; render(); await waitFor(() => { expect(screen.getByText('manual')).toBeInTheDocument(); }); }); it('defaults to auto when config has no mode', async () => { mockConfig.GOOSE_MODE = undefined; render(); await waitFor(() => { expect(screen.getByText('autonomous')).toBeInTheDocument(); }); }); it('fetches mode from session when sessionId is present', async () => { mockConfig.GOOSE_MODE = 'auto'; mockGetSession.mockResolvedValue({ data: { goose_mode: 'approve' } }); render(); await waitFor(() => { expect(screen.getByText('manual')).toBeInTheDocument(); }); expect(mockGetSession).toHaveBeenCalledWith({ path: { session_id: 'test-session-123' }, }); }); it('calls updateSession and does not write global config', async () => { mockConfig.GOOSE_MODE = 'auto'; render(); fireEvent.click(screen.getByText('Manual')); await waitFor(() => { expect(mockUpdateSession).toHaveBeenCalledWith({ body: { session_id: 'test-session-123', goose_mode: 'approve' }, }); }); }); it('does not call updateSession when sessionId is null', async () => { mockConfig.GOOSE_MODE = 'auto'; render(); fireEvent.click(screen.getByText('Manual')); await waitFor(() => { expect(screen.getByText('manual')).toBeInTheDocument(); }); expect(mockUpdateSession).not.toHaveBeenCalled(); }); it('ignores stale session fetch after sessionId changes', async () => { let resolveA: (value: unknown) => void; const promiseA = new Promise((resolve) => { resolveA = resolve; }); mockGetSession .mockImplementationOnce(() => promiseA) .mockResolvedValueOnce({ data: { goose_mode: 'auto' } }); const { rerender } = render(); rerender(); await waitFor(() => { expect(screen.getByText('autonomous')).toBeInTheDocument(); }); resolveA!({ data: { goose_mode: 'approve' } }); await waitFor(() => { expect(screen.getByText('autonomous')).toBeInTheDocument(); }); expect(screen.queryByText('manual')).not.toBeInTheDocument(); }); });