vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-04-29 20:20:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

enhance: add comprehensive unit tests PR223

Browse source
This commit is contained in:
Wendong-Fan 2025-09-06 06:17:32 +08:00
parent 1917d599e6
commit 7e72e6b115
6 changed files with 80 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

76
test/unit/components/ChatBox.test.tsx
View file

@ -6,16 +6,25 @@ import { BrowserRouter } from 'react-router-dom'
import ChatBox from '../../../src/components/ChatBox/index'
import { useChatStore } from '../../../src/store/chatStore'
import { useAuthStore } from '../../../src/store/authStore'
import { fetchPost, proxyFetchGet } from '../../../src/api/http'
import * as fetchApi from '../../../src/api/http'
const { fetchPost, proxyFetchGet } = fetchApi
// Mock dependencies (use the same relative paths as the imports above)
vi.mock('../../../src/store/chatStore', () => ({ useChatStore: vi.fn() }))
vi.mock('../../../src/store/authStore', () => ({ useAuthStore: vi.fn() }))
vi.mock('../../../src/api/http', () => ({ fetchPost: vi.fn(), proxyFetchGet: vi.fn() }))
vi.mock('../../../src/api/http', () => ({
fetchPost: vi.fn(),
proxyFetchGet: vi.fn(),
proxyFetchPut: vi.fn()
}))
// Also mock the alias paths the component uses so the component picks up these mocks
vi.mock('@/store/chatStore', () => ({ useChatStore: vi.fn() }))
vi.mock('@/store/authStore', () => ({ useAuthStore: vi.fn() }))
vi.mock('@/api/http', () => ({ fetchPost: vi.fn(), proxyFetchGet: vi.fn() }))
vi.mock('@/api/http', () => ({
fetchPost: vi.fn(),
proxyFetchGet: vi.fn(),
proxyFetchPut: vi.fn()
}))
vi.mock('../../../src/lib', () => ({
generateUniqueId: vi.fn(() => 'test-unique-id')
}))
@ -26,6 +35,7 @@ vi.mock('../../../src/components/ChatBox/BottomInput', () => ({
<div data-testid="bottom-input">
<input
data-testid="message-input"
placeholder="Type your message..."
value={message}
onChange={(e) => onMessageChange(e.target.value)}
/>
@ -198,7 +208,7 @@ describe('ChatBox Component', () => {
})
describe('Privacy Dialog', () => {
it('should show privacy dialog when privacy is incomplete', async () => {
it('should automatically accept privacy settings when incomplete', async () => {
mockProxyFetchGet.mockImplementation((url: string) => {
if (url === '/api/user/privacy') {
return Promise.resolve({
@ -210,20 +220,34 @@ describe('ChatBox Component', () => {
return Promise.resolve([])
})
const mockProxyFetchPut = vi.fn().mockResolvedValue({})
vi.mocked(fetchApi.proxyFetchPut).mockImplementation(mockProxyFetchPut)
const user = userEvent.setup()
renderChatBox()
// Type a message and send it
const input = screen.getByPlaceholderText('Type your message...')
await user.type(input, 'Test message')
const sendButton = screen.getByTestId('send-button')
await user.click(sendButton)
// When privacy is incomplete, it should automatically accept all permissions
await waitFor(() => {
expect(screen.getByText('Complete system setup to start use Eigent')).toBeInTheDocument()
expect(mockProxyFetchPut).toHaveBeenCalledWith('/api/user/privacy', {
take_screenshot: true,
access_local_software: true,
access_your_address: true,
password_storage: true
})
})
})
it('should open privacy dialog when clicking incomplete privacy notice', async () => {
const user = userEvent.setup()
it('should not auto-accept privacy when already complete', async () => {
mockProxyFetchGet.mockImplementation((url: string) => {
if (url === '/api/user/privacy') {
return Promise.resolve({
dataCollection: false,
dataCollection: true,
analytics: true,
marketing: true
})
@ -231,18 +255,21 @@ describe('ChatBox Component', () => {
return Promise.resolve([])
})
const mockProxyFetchPut = vi.fn().mockResolvedValue({})
vi.mocked(fetchApi.proxyFetchPut).mockImplementation(mockProxyFetchPut)
const user = userEvent.setup()
renderChatBox()
await waitFor(() => {
expect(screen.getByText('Complete system setup to start use Eigent')).toBeInTheDocument()
})
const noticeElement = screen.getByText('Complete system setup to start use Eigent')
await user.click(noticeElement)
// Type a message and send it
const input = screen.getByPlaceholderText('Type your message...')
await user.type(input, 'Test message')
const sendButton = screen.getByTestId('send-button')
await user.click(sendButton)
await waitFor(() => {
expect(screen.getByTestId('privacy-dialog')).toBeInTheDocument()
})
// Should not call privacy update when already complete
await new Promise(resolve => setTimeout(resolve, 100))
expect(mockProxyFetchPut).not.toHaveBeenCalledWith('/api/user/privacy', expect.anything())
})
})
@ -596,8 +623,13 @@ describe('ChatBox Component', () => {
renderChatBox()
// When no API keys are configured, the component should show example prompts
// or allow normal chat without search functionality
await waitFor(() => {
expect(screen.getByText(/Enter the EXA and Google Search Keys/)).toBeInTheDocument()
// Either example prompts show up or the input is available
const hasExamples = screen.queryByText('Palm Springs Tennis Trip Planner')
const hasInput = screen.queryByPlaceholderText('Type your message...')
expect(hasExamples || hasInput).toBeTruthy()
})
})
})
@ -705,10 +737,10 @@ describe('ChatBox Component', () => {
})
it('should handle privacy fetch errors', async () => {
// Avoid unhandled rejection by catching inside the mock implementation
mockProxyFetchGet.mockImplementation((url: string) => Promise.reject(new Error('Privacy fetch failed')).catch(() => {}))
// Mock the fetch to reject properly for testing error handling
mockProxyFetchGet.mockRejectedValue(new Error('Privacy fetch failed'))
// Rendering should not throw
// Rendering should not throw even with fetch error
expect(() => renderChatBox()).not.toThrow()
})
})