fix: test

This commit is contained in:
Louis 2025-08-21 12:01:45 +07:00
parent cfbc6b9150
commit 8de5c1709b
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
3 changed files with 71 additions and 150 deletions

View File

@ -5,7 +5,6 @@ import SettingsMenu from '../SettingsMenu'
import { useNavigate, useMatches } from '@tanstack/react-router' import { useNavigate, useMatches } from '@tanstack/react-router'
import { useGeneralSetting } from '@/hooks/useGeneralSetting' import { useGeneralSetting } from '@/hooks/useGeneralSetting'
import { useModelProvider } from '@/hooks/useModelProvider' import { useModelProvider } from '@/hooks/useModelProvider'
import { useAppState } from '@/hooks/useAppState'
// Mock dependencies // Mock dependencies
vi.mock('@tanstack/react-router', () => ({ vi.mock('@tanstack/react-router', () => ({
@ -25,9 +24,7 @@ vi.mock('@/i18n/react-i18next-compat', () => ({
})) }))
vi.mock('@/hooks/useGeneralSetting', () => ({ vi.mock('@/hooks/useGeneralSetting', () => ({
useGeneralSetting: vi.fn(() => ({ useGeneralSetting: vi.fn(() => ({})),
experimentalFeatures: false,
})),
})) }))
vi.mock('@/hooks/useModelProvider', () => ({ vi.mock('@/hooks/useModelProvider', () => ({
@ -71,14 +68,14 @@ describe('SettingsMenu', () => {
beforeEach(() => { beforeEach(() => {
vi.clearAllMocks() vi.clearAllMocks()
vi.mocked(useNavigate).mockReturnValue(mockNavigate) vi.mocked(useNavigate).mockReturnValue(mockNavigate)
vi.mocked(useMatches).mockReturnValue(mockMatches) vi.mocked(useMatches).mockReturnValue(mockMatches)
}) })
it('renders all menu items', () => { it('renders all menu items', () => {
render(<SettingsMenu />) render(<SettingsMenu />)
expect(screen.getByText('common:general')).toBeInTheDocument() expect(screen.getByText('common:general')).toBeInTheDocument()
expect(screen.getByText('common:appearance')).toBeInTheDocument() expect(screen.getByText('common:appearance')).toBeInTheDocument()
expect(screen.getByText('common:privacy')).toBeInTheDocument() expect(screen.getByText('common:privacy')).toBeInTheDocument()
@ -88,29 +85,14 @@ describe('SettingsMenu', () => {
expect(screen.getByText('common:local_api_server')).toBeInTheDocument() expect(screen.getByText('common:local_api_server')).toBeInTheDocument()
expect(screen.getByText('common:https_proxy')).toBeInTheDocument() expect(screen.getByText('common:https_proxy')).toBeInTheDocument()
expect(screen.getByText('common:extensions')).toBeInTheDocument() expect(screen.getByText('common:extensions')).toBeInTheDocument()
})
it('does not show MCP Servers when experimental features disabled', () => {
render(<SettingsMenu />)
expect(screen.queryByText('common:mcp-servers')).not.toBeInTheDocument()
})
it('shows MCP Servers when experimental features enabled', () => {
vi.mocked(useGeneralSetting).mockReturnValue({
experimentalFeatures: true,
})
render(<SettingsMenu />)
expect(screen.getByText('common:mcp-servers')).toBeInTheDocument() expect(screen.getByText('common:mcp-servers')).toBeInTheDocument()
}) })
it('shows provider expansion chevron when providers are active', () => { it('shows provider expansion chevron when providers are active', () => {
render(<SettingsMenu />) render(<SettingsMenu />)
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
expect(chevron).toBeInTheDocument() expect(chevron).toBeInTheDocument()
@ -119,14 +101,14 @@ describe('SettingsMenu', () => {
it('expands providers submenu when chevron is clicked', async () => { it('expands providers submenu when chevron is clicked', async () => {
const user = userEvent.setup() const user = userEvent.setup()
render(<SettingsMenu />) render(<SettingsMenu />)
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
if (!chevron) throw new Error('Chevron button not found') if (!chevron) throw new Error('Chevron button not found')
await user.click(chevron) await user.click(chevron)
expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument() expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument()
expect(screen.getByTestId('provider-avatar-llama.cpp')).toBeInTheDocument() expect(screen.getByTestId('provider-avatar-llama.cpp')).toBeInTheDocument()
}) })
@ -138,52 +120,56 @@ describe('SettingsMenu', () => {
params: { providerName: 'openai' }, params: { providerName: 'openai' },
}, },
]) ])
render(<SettingsMenu />) render(<SettingsMenu />)
expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument() expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument()
expect(screen.getByTestId('provider-avatar-llama.cpp')).toBeInTheDocument() expect(screen.getByTestId('provider-avatar-llama.cpp')).toBeInTheDocument()
}) })
it('highlights active provider in submenu', async () => { it('highlights active provider in submenu', async () => {
const user = userEvent.setup() const user = userEvent.setup()
vi.mocked(useMatches).mockReturnValue([ vi.mocked(useMatches).mockReturnValue([
{ {
routeId: '/settings/providers/$providerName', routeId: '/settings/providers/$providerName',
params: { providerName: 'openai' }, params: { providerName: 'openai' },
}, },
]) ])
render(<SettingsMenu />) render(<SettingsMenu />)
// First expand the providers submenu // First expand the providers submenu
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
if (chevron) await user.click(chevron) if (chevron) await user.click(chevron)
const openaiProvider = screen.getByTestId('provider-avatar-openai').closest('div') const openaiProvider = screen
.getByTestId('provider-avatar-openai')
.closest('div')
expect(openaiProvider).toBeInTheDocument() expect(openaiProvider).toBeInTheDocument()
}) })
it('navigates to provider when provider is clicked', async () => { it('navigates to provider when provider is clicked', async () => {
const user = userEvent.setup() const user = userEvent.setup()
render(<SettingsMenu />) render(<SettingsMenu />)
// First expand the providers // First expand the providers
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
if (!chevron) throw new Error('Chevron button not found') if (!chevron) throw new Error('Chevron button not found')
await user.click(chevron) await user.click(chevron)
// Then click on a provider // Then click on a provider
const openaiProvider = screen.getByTestId('provider-avatar-openai').closest('div') const openaiProvider = screen
.getByTestId('provider-avatar-openai')
.closest('div')
await user.click(openaiProvider!) await user.click(openaiProvider!)
expect(mockNavigate).toHaveBeenCalledWith({ expect(mockNavigate).toHaveBeenCalledWith({
to: '/settings/providers/$providerName', to: '/settings/providers/$providerName',
params: { providerName: 'openai' }, params: { providerName: 'openai' },
@ -192,18 +178,22 @@ describe('SettingsMenu', () => {
it('shows mobile menu toggle button', () => { it('shows mobile menu toggle button', () => {
render(<SettingsMenu />) render(<SettingsMenu />)
const menuToggle = screen.getByRole('button', { name: 'Toggle settings menu' }) const menuToggle = screen.getByRole('button', {
name: 'Toggle settings menu',
})
expect(menuToggle).toBeInTheDocument() expect(menuToggle).toBeInTheDocument()
}) })
it('opens mobile menu when toggle is clicked', async () => { it('opens mobile menu when toggle is clicked', async () => {
const user = userEvent.setup() const user = userEvent.setup()
render(<SettingsMenu />) render(<SettingsMenu />)
const menuToggle = screen.getByRole('button', { name: 'Toggle settings menu' }) const menuToggle = screen.getByRole('button', {
name: 'Toggle settings menu',
})
await user.click(menuToggle) await user.click(menuToggle)
// Menu should now be visible // Menu should now be visible
const menu = screen.getByText('common:general').closest('div') const menu = screen.getByText('common:general').closest('div')
expect(menu).toHaveClass('flex') expect(menu).toHaveClass('flex')
@ -212,21 +202,23 @@ describe('SettingsMenu', () => {
it('closes mobile menu when X is clicked', async () => { it('closes mobile menu when X is clicked', async () => {
const user = userEvent.setup() const user = userEvent.setup()
render(<SettingsMenu />) render(<SettingsMenu />)
// Open menu first // Open menu first
const menuToggle = screen.getByRole('button', { name: 'Toggle settings menu' }) const menuToggle = screen.getByRole('button', {
name: 'Toggle settings menu',
})
await user.click(menuToggle) await user.click(menuToggle)
// Then close it // Then close it
await user.click(menuToggle) await user.click(menuToggle)
// Just verify the toggle button is still there after clicking twice // Just verify the toggle button is still there after clicking twice
expect(menuToggle).toBeInTheDocument() expect(menuToggle).toBeInTheDocument()
}) })
it('hides llamacpp provider during setup remote provider step', async () => { it('hides llamacpp provider during setup remote provider step', async () => {
const user = userEvent.setup() const user = userEvent.setup()
vi.mocked(useMatches).mockReturnValue([ vi.mocked(useMatches).mockReturnValue([
{ {
routeId: '/settings/providers/', routeId: '/settings/providers/',
@ -234,16 +226,16 @@ describe('SettingsMenu', () => {
search: { step: 'setup_remote_provider' }, search: { step: 'setup_remote_provider' },
}, },
]) ])
render(<SettingsMenu />) render(<SettingsMenu />)
// First expand the providers submenu // First expand the providers submenu
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
if (chevron) await user.click(chevron) if (chevron) await user.click(chevron)
// llamacpp provider div should have hidden class // llamacpp provider div should have hidden class
const llamacppElement = screen.getByTestId('provider-avatar-llama.cpp') const llamacppElement = screen.getByTestId('provider-avatar-llama.cpp')
expect(llamacppElement.parentElement).toHaveClass('hidden') expect(llamacppElement.parentElement).toHaveClass('hidden')
@ -253,7 +245,7 @@ describe('SettingsMenu', () => {
it('filters out inactive providers from submenu', async () => { it('filters out inactive providers from submenu', async () => {
const user = userEvent.setup() const user = userEvent.setup()
vi.mocked(useModelProvider).mockReturnValue({ vi.mocked(useModelProvider).mockReturnValue({
providers: [ providers: [
{ {
@ -268,17 +260,19 @@ describe('SettingsMenu', () => {
}, },
], ],
}) })
render(<SettingsMenu />) render(<SettingsMenu />)
// Expand providers // Expand providers
const chevronButtons = screen.getAllByRole('button') const chevronButtons = screen.getAllByRole('button')
const chevron = chevronButtons.find(button => const chevron = chevronButtons.find((button) =>
button.querySelector('svg.tabler-icon-chevron-right') button.querySelector('svg.tabler-icon-chevron-right')
) )
if (chevron) await user.click(chevron) if (chevron) await user.click(chevron)
expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument() expect(screen.getByTestId('provider-avatar-openai')).toBeInTheDocument()
expect(screen.queryByTestId('provider-avatar-anthropic')).not.toBeInTheDocument() expect(
screen.queryByTestId('provider-avatar-anthropic')
).not.toBeInTheDocument()
}) })
}) })

View File

@ -31,16 +31,15 @@ describe('useGeneralSetting', () => {
beforeEach(async () => { beforeEach(async () => {
vi.clearAllMocks() vi.clearAllMocks()
// Get the mocked ExtensionManager // Get the mocked ExtensionManager
const { ExtensionManager } = await import('@/lib/extension') const { ExtensionManager } = await import('@/lib/extension')
mockExtensionManager = ExtensionManager mockExtensionManager = ExtensionManager
// Reset store state to defaults // Reset store state to defaults
useGeneralSetting.setState({ useGeneralSetting.setState({
currentLanguage: 'en', currentLanguage: 'en',
spellCheckChatInput: true, spellCheckChatInput: true,
experimentalFeatures: false,
huggingfaceToken: undefined, huggingfaceToken: undefined,
}) })
@ -49,7 +48,7 @@ describe('useGeneralSetting', () => {
getSettings: vi.fn().mockResolvedValue(null), getSettings: vi.fn().mockResolvedValue(null),
updateSettings: vi.fn(), updateSettings: vi.fn(),
}) })
mockExtensionManager.getInstance.mockReturnValue({ mockExtensionManager.getInstance.mockReturnValue({
getByName: mockGetByName, getByName: mockGetByName,
}) })
@ -60,11 +59,9 @@ describe('useGeneralSetting', () => {
expect(result.current.currentLanguage).toBe('en') expect(result.current.currentLanguage).toBe('en')
expect(result.current.spellCheckChatInput).toBe(true) expect(result.current.spellCheckChatInput).toBe(true)
expect(result.current.experimentalFeatures).toBe(false)
expect(result.current.huggingfaceToken).toBeUndefined() expect(result.current.huggingfaceToken).toBeUndefined()
expect(typeof result.current.setCurrentLanguage).toBe('function') expect(typeof result.current.setCurrentLanguage).toBe('function')
expect(typeof result.current.setSpellCheckChatInput).toBe('function') expect(typeof result.current.setSpellCheckChatInput).toBe('function')
expect(typeof result.current.setExperimentalFeatures).toBe('function')
expect(typeof result.current.setHuggingfaceToken).toBe('function') expect(typeof result.current.setHuggingfaceToken).toBe('function')
}) })
@ -155,42 +152,6 @@ describe('useGeneralSetting', () => {
}) })
}) })
describe('setExperimentalFeatures', () => {
it('should enable experimental features', () => {
const { result } = renderHook(() => useGeneralSetting())
act(() => {
result.current.setExperimentalFeatures(true)
})
expect(result.current.experimentalFeatures).toBe(true)
})
it('should disable experimental features', () => {
const { result } = renderHook(() => useGeneralSetting())
act(() => {
result.current.setExperimentalFeatures(false)
})
expect(result.current.experimentalFeatures).toBe(false)
})
it('should toggle experimental features multiple times', () => {
const { result } = renderHook(() => useGeneralSetting())
act(() => {
result.current.setExperimentalFeatures(true)
})
expect(result.current.experimentalFeatures).toBe(true)
act(() => {
result.current.setExperimentalFeatures(false)
})
expect(result.current.experimentalFeatures).toBe(false)
})
})
describe('setHuggingfaceToken', () => { describe('setHuggingfaceToken', () => {
it('should set huggingface token', () => { it('should set huggingface token', () => {
const { result } = renderHook(() => useGeneralSetting()) const { result } = renderHook(() => useGeneralSetting())
@ -235,7 +196,7 @@ describe('useGeneralSetting', () => {
const mockGetByName = vi.fn() const mockGetByName = vi.fn()
const mockGetSettings = vi.fn().mockResolvedValue(mockSettings) const mockGetSettings = vi.fn().mockResolvedValue(mockSettings)
const mockUpdateSettings = vi.fn() const mockUpdateSettings = vi.fn()
mockExtensionManager.getInstance.mockReturnValue({ mockExtensionManager.getInstance.mockReturnValue({
getByName: mockGetByName, getByName: mockGetByName,
}) })
@ -252,9 +213,9 @@ describe('useGeneralSetting', () => {
expect(mockExtensionManager.getInstance).toHaveBeenCalled() expect(mockExtensionManager.getInstance).toHaveBeenCalled()
expect(mockGetByName).toHaveBeenCalledWith('@janhq/download-extension') expect(mockGetByName).toHaveBeenCalledWith('@janhq/download-extension')
// Wait for async operations // Wait for async operations
await new Promise(resolve => setTimeout(resolve, 0)) await new Promise((resolve) => setTimeout(resolve, 0))
expect(mockGetSettings).toHaveBeenCalled() expect(mockGetSettings).toHaveBeenCalled()
expect(mockUpdateSettings).toHaveBeenCalledWith([ expect(mockUpdateSettings).toHaveBeenCalledWith([
@ -272,13 +233,11 @@ describe('useGeneralSetting', () => {
act(() => { act(() => {
result1.current.setCurrentLanguage('id') result1.current.setCurrentLanguage('id')
result1.current.setSpellCheckChatInput(false) result1.current.setSpellCheckChatInput(false)
result1.current.setExperimentalFeatures(true)
result1.current.setHuggingfaceToken('shared-token') result1.current.setHuggingfaceToken('shared-token')
}) })
expect(result2.current.currentLanguage).toBe('id') expect(result2.current.currentLanguage).toBe('id')
expect(result2.current.spellCheckChatInput).toBe(false) expect(result2.current.spellCheckChatInput).toBe(false)
expect(result2.current.experimentalFeatures).toBe(true)
expect(result2.current.huggingfaceToken).toBe('shared-token') expect(result2.current.huggingfaceToken).toBe('shared-token')
}) })
}) })
@ -290,13 +249,11 @@ describe('useGeneralSetting', () => {
act(() => { act(() => {
result.current.setCurrentLanguage('vn') result.current.setCurrentLanguage('vn')
result.current.setSpellCheckChatInput(false) result.current.setSpellCheckChatInput(false)
result.current.setExperimentalFeatures(true)
result.current.setHuggingfaceToken('complex-token-123') result.current.setHuggingfaceToken('complex-token-123')
}) })
expect(result.current.currentLanguage).toBe('vn') expect(result.current.currentLanguage).toBe('vn')
expect(result.current.spellCheckChatInput).toBe(false) expect(result.current.spellCheckChatInput).toBe(false)
expect(result.current.experimentalFeatures).toBe(true)
expect(result.current.huggingfaceToken).toBe('complex-token-123') expect(result.current.huggingfaceToken).toBe('complex-token-123')
}) })
@ -314,11 +271,9 @@ describe('useGeneralSetting', () => {
// Second update // Second update
act(() => { act(() => {
result.current.setExperimentalFeatures(true)
result.current.setHuggingfaceToken('sequential-token') result.current.setHuggingfaceToken('sequential-token')
}) })
expect(result.current.experimentalFeatures).toBe(true)
expect(result.current.huggingfaceToken).toBe('sequential-token') expect(result.current.huggingfaceToken).toBe('sequential-token')
// Third update // Third update
@ -331,4 +286,4 @@ describe('useGeneralSetting', () => {
expect(result.current.spellCheckChatInput).toBe(true) expect(result.current.spellCheckChatInput).toBe(true)
}) })
}) })
}) })

View File

@ -61,8 +61,6 @@ vi.mock('@/hooks/useGeneralSetting', () => ({
useGeneralSetting: () => ({ useGeneralSetting: () => ({
spellCheckChatInput: true, spellCheckChatInput: true,
setSpellCheckChatInput: vi.fn(), setSpellCheckChatInput: vi.fn(),
experimentalFeatures: false,
setExperimentalFeatures: vi.fn(),
huggingfaceToken: 'test-token', huggingfaceToken: 'test-token',
setHuggingfaceToken: vi.fn(), setHuggingfaceToken: vi.fn(),
}), }),
@ -188,12 +186,14 @@ vi.mock('@tauri-apps/plugin-opener', () => ({
})) }))
vi.mock('@tauri-apps/api/webviewWindow', () => { vi.mock('@tauri-apps/api/webviewWindow', () => {
const MockWebviewWindow = vi.fn().mockImplementation((label: string, options: any) => ({ const MockWebviewWindow = vi
once: vi.fn(), .fn()
setFocus: vi.fn(), .mockImplementation((label: string, options: any) => ({
})) once: vi.fn(),
setFocus: vi.fn(),
}))
MockWebviewWindow.getByLabel = vi.fn().mockReturnValue(null) MockWebviewWindow.getByLabel = vi.fn().mockReturnValue(null)
return { return {
WebviewWindow: MockWebviewWindow, WebviewWindow: MockWebviewWindow,
} }
@ -299,16 +299,6 @@ describe('General Settings Route', () => {
// expect(screen.getByTestId('language-switcher')).toBeInTheDocument() // expect(screen.getByTestId('language-switcher')).toBeInTheDocument()
// }) // })
it('should render switches for experimental features and spell check', async () => {
const Component = GeneralRoute.component as React.ComponentType
await act(async () => {
render(<Component />)
})
const switches = screen.getAllByTestId('switch')
expect(switches.length).toBeGreaterThanOrEqual(2)
})
it('should render huggingface token input', async () => { it('should render huggingface token input', async () => {
const Component = GeneralRoute.component as React.ComponentType const Component = GeneralRoute.component as React.ComponentType
await act(async () => { await act(async () => {
@ -336,24 +326,6 @@ describe('General Settings Route', () => {
expect(switches[0]).toBeInTheDocument() expect(switches[0]).toBeInTheDocument()
}) })
it('should handle experimental features toggle', async () => {
const Component = GeneralRoute.component as React.ComponentType
await act(async () => {
render(<Component />)
})
const switches = screen.getAllByTestId('switch')
expect(switches.length).toBeGreaterThan(0)
// Test that switches are interactive
if (switches.length > 1) {
await act(async () => {
fireEvent.click(switches[1])
})
expect(switches[1]).toBeInTheDocument()
}
})
it('should handle huggingface token change', async () => { it('should handle huggingface token change', async () => {
const Component = GeneralRoute.component as React.ComponentType const Component = GeneralRoute.component as React.ComponentType
await act(async () => { await act(async () => {
@ -514,16 +486,16 @@ describe('General Settings Route', () => {
act(() => { act(() => {
fireEvent.click(checkUpdateButton) fireEvent.click(checkUpdateButton)
}) })
// Now the button should be disabled while checking // Now the button should be disabled while checking
expect(checkUpdateButton).toBeDisabled() expect(checkUpdateButton).toBeDisabled()
// Resolve the promise to finish the update check // Resolve the promise to finish the update check
await act(async () => { await act(async () => {
resolveUpdate!(null) resolveUpdate!(null)
await updatePromise await updatePromise
}) })
// Button should be enabled again // Button should be enabled again
expect(checkUpdateButton).not.toBeDisabled() expect(checkUpdateButton).not.toBeDisabled()
} }