|
| 1 | +/* |
| 2 | + * Copyright OpenSearch Contributors |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
| 6 | +import React from 'react'; |
| 7 | +import { render, waitFor } from '@testing-library/react'; |
| 8 | + |
| 9 | +import { ChatHeaderButton, ChatHeaderButtonInstance } from './chat_header_button'; |
| 10 | +import { ChatService } from '../services/chat_service'; |
| 11 | +import { coreMock } from '../../../../core/public/mocks'; |
| 12 | + |
| 13 | +// Mock dependencies |
| 14 | + |
| 15 | +jest.mock('./chat_window', () => { |
| 16 | + const ActualReact = jest.requireActual('react'); |
| 17 | + return { |
| 18 | + ChatWindow: ActualReact.forwardRef((props: any, ref: any) => { |
| 19 | + ActualReact.useImperativeHandle(ref, () => ({ |
| 20 | + startNewChat: jest.fn(), |
| 21 | + sendMessage: jest.fn().mockResolvedValue(undefined), |
| 22 | + })); |
| 23 | + return null; |
| 24 | + }), |
| 25 | + }; |
| 26 | +}); |
| 27 | + |
| 28 | +describe('ChatHeaderButton', () => { |
| 29 | + let mockCore: ReturnType<typeof coreMock.createStart>; |
| 30 | + let mockChatService: jest.Mocked<ChatService>; |
| 31 | + let mockContextProvider: any; |
| 32 | + |
| 33 | + beforeEach(() => { |
| 34 | + jest.clearAllMocks(); |
| 35 | + mockCore = coreMock.createStart(); |
| 36 | + mockChatService = { |
| 37 | + sendMessage: jest.fn(), |
| 38 | + newThread: jest.fn(), |
| 39 | + } as any; |
| 40 | + mockContextProvider = {}; |
| 41 | + |
| 42 | + // Mock sidecar |
| 43 | + mockCore.overlays.sidecar.open = jest.fn().mockReturnValue({ |
| 44 | + close: jest.fn(), |
| 45 | + }); |
| 46 | + mockCore.overlays.sidecar.setSidecarConfig = jest.fn(); |
| 47 | + }); |
| 48 | + |
| 49 | + describe('ref functionality', () => { |
| 50 | + it('should expose startNewConversation method via ref', async () => { |
| 51 | + const ref = React.createRef<ChatHeaderButtonInstance>(); |
| 52 | + |
| 53 | + render( |
| 54 | + <ChatHeaderButton |
| 55 | + core={mockCore} |
| 56 | + chatService={mockChatService} |
| 57 | + contextProvider={mockContextProvider} |
| 58 | + ref={ref} |
| 59 | + /> |
| 60 | + ); |
| 61 | + |
| 62 | + expect(ref.current).toBeDefined(); |
| 63 | + expect(ref.current?.startNewConversation).toBeDefined(); |
| 64 | + expect(typeof ref.current?.startNewConversation).toBe('function'); |
| 65 | + }); |
| 66 | + |
| 67 | + it('should call startNewChat and sendMessage when startNewConversation is invoked', async () => { |
| 68 | + const ref = React.createRef<ChatHeaderButtonInstance>(); |
| 69 | + |
| 70 | + render( |
| 71 | + <ChatHeaderButton |
| 72 | + core={mockCore} |
| 73 | + chatService={mockChatService} |
| 74 | + contextProvider={mockContextProvider} |
| 75 | + ref={ref} |
| 76 | + /> |
| 77 | + ); |
| 78 | + |
| 79 | + // Call startNewConversation |
| 80 | + await ref.current?.startNewConversation({ content: 'test message' }); |
| 81 | + |
| 82 | + // Verify sidecar was opened |
| 83 | + await waitFor(() => { |
| 84 | + expect(mockCore.overlays.sidecar.open).toHaveBeenCalled(); |
| 85 | + }); |
| 86 | + }); |
| 87 | + }); |
| 88 | +}); |
0 commit comments