mirror of
https://github.com/langgenius/dify.git
synced 2026-04-08 09:16:41 +08:00
101 lines
2.3 KiB
TypeScript
101 lines
2.3 KiB
TypeScript
import type { Mock } from 'vitest'
|
|
import { renderHook } from '@testing-library/react'
|
|
import { useContext } from 'use-context-selector'
|
|
import { useLanguage } from './hooks'
|
|
|
|
vi.mock('@tanstack/react-query', () => ({
|
|
useQuery: vi.fn(),
|
|
useQueryClient: vi.fn(() => ({
|
|
invalidateQueries: vi.fn(),
|
|
})),
|
|
}))
|
|
|
|
// mock use-context-selector
|
|
vi.mock('use-context-selector', () => ({
|
|
useContext: vi.fn(),
|
|
createContext: () => ({
|
|
Provider: ({ children }: any) => children,
|
|
Consumer: ({ children }: any) => children(null),
|
|
}),
|
|
useContextSelector: vi.fn(),
|
|
}))
|
|
|
|
// mock service/common functions
|
|
vi.mock('@/service/common', () => ({
|
|
fetchDefaultModal: vi.fn(),
|
|
fetchModelList: vi.fn(),
|
|
fetchModelProviderCredentials: vi.fn(),
|
|
getPayUrl: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/service/use-common', () => ({
|
|
commonQueryKeys: {
|
|
modelProviders: ['common', 'model-providers'],
|
|
},
|
|
}))
|
|
|
|
// mock context hooks
|
|
vi.mock('@/context/i18n', () => ({
|
|
__esModule: true,
|
|
default: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/context/provider-context', () => ({
|
|
useProviderContext: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/context/modal-context', () => ({
|
|
useModalContextSelector: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/context/event-emitter', () => ({
|
|
useEventEmitterContextContext: vi.fn(),
|
|
}))
|
|
|
|
// mock plugins
|
|
vi.mock('@/app/components/plugins/marketplace/hooks', () => ({
|
|
useMarketplacePlugins: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/app/components/plugins/marketplace/utils', () => ({
|
|
getMarketplacePluginsByCollectionId: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('./provider-added-card', () => ({
|
|
default: vi.fn(),
|
|
}))
|
|
|
|
afterAll(() => {
|
|
vi.resetModules()
|
|
vi.clearAllMocks()
|
|
})
|
|
|
|
describe('useLanguage', () => {
|
|
it('should replace hyphen with underscore in locale', () => {
|
|
(useContext as Mock).mockReturnValue({
|
|
locale: 'en-US',
|
|
})
|
|
const { result } = renderHook(() => useLanguage())
|
|
expect(result.current).toBe('en_US')
|
|
})
|
|
|
|
it('should return locale as is if no hyphen exists', () => {
|
|
(useContext as Mock).mockReturnValue({
|
|
locale: 'enUS',
|
|
})
|
|
|
|
const { result } = renderHook(() => useLanguage())
|
|
expect(result.current).toBe('enUS')
|
|
})
|
|
|
|
it('should handle multiple hyphens', () => {
|
|
// Mock the I18n context return value
|
|
(useContext as Mock).mockReturnValue({
|
|
locale: 'zh-Hans-CN',
|
|
})
|
|
|
|
const { result } = renderHook(() => useLanguage())
|
|
expect(result.current).toBe('zh_Hans-CN')
|
|
})
|
|
})
|