mirror of https://github.com/langgenius/dify.git
91 lines
2.6 KiB
TypeScript
91 lines
2.6 KiB
TypeScript
import * as React from 'react'
|
|
import { vi } from 'vitest'
|
|
|
|
type TranslationMap = Record<string, string | string[]>
|
|
|
|
/**
|
|
* Create a t function with optional custom translations
|
|
* Checks translations[key] first, then translations[ns.key], then returns ns.key as fallback
|
|
*/
|
|
export function createTFunction(translations: TranslationMap, defaultNs?: string) {
|
|
return (key: string, options?: Record<string, unknown>) => {
|
|
// Check custom translations first (without namespace)
|
|
if (translations[key] !== undefined)
|
|
return translations[key]
|
|
|
|
const ns = (options?.ns as string | undefined) ?? defaultNs
|
|
const fullKey = ns ? `${ns}.${key}` : key
|
|
|
|
// Check custom translations with namespace
|
|
if (translations[fullKey] !== undefined)
|
|
return translations[fullKey]
|
|
|
|
// Serialize params (excluding ns) for test assertions
|
|
const params = { ...options }
|
|
delete params.ns
|
|
const suffix = Object.keys(params).length > 0 ? `:${JSON.stringify(params)}` : ''
|
|
return `${fullKey}${suffix}`
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Create useTranslation mock with optional custom translations
|
|
*
|
|
* Caches t functions by defaultNs so the same reference is returned
|
|
* across renders, preventing infinite re-render loops when components
|
|
* include t in useEffect/useMemo dependency arrays.
|
|
*
|
|
* @example
|
|
* vi.mock('react-i18next', () => createUseTranslationMock({
|
|
* 'operation.confirm': 'Confirm',
|
|
* }))
|
|
*/
|
|
export function createUseTranslationMock(translations: TranslationMap = {}) {
|
|
const tCache = new Map<string, ReturnType<typeof createTFunction>>()
|
|
const i18n = {
|
|
language: 'en',
|
|
changeLanguage: vi.fn(),
|
|
}
|
|
return {
|
|
useTranslation: (defaultNs?: string) => {
|
|
const cacheKey = defaultNs ?? ''
|
|
if (!tCache.has(cacheKey))
|
|
tCache.set(cacheKey, createTFunction(translations, defaultNs))
|
|
return {
|
|
t: tCache.get(cacheKey)!,
|
|
i18n,
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Create Trans component mock with optional custom translations
|
|
*/
|
|
export function createTransMock(translations: TranslationMap = {}) {
|
|
return {
|
|
Trans: ({ i18nKey, children }: {
|
|
i18nKey: string
|
|
children?: React.ReactNode
|
|
}) => {
|
|
const text = translations[i18nKey] ?? i18nKey
|
|
return React.createElement('span', { 'data-i18n-key': i18nKey }, children ?? text)
|
|
},
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Create complete react-i18next mock (useTranslation + Trans)
|
|
*
|
|
* @example
|
|
* vi.mock('react-i18next', () => createReactI18nextMock({
|
|
* 'modal.title': 'My Modal',
|
|
* }))
|
|
*/
|
|
export function createReactI18nextMock(translations: TranslationMap = {}) {
|
|
return {
|
|
...createUseTranslationMock(translations),
|
|
...createTransMock(translations),
|
|
}
|
|
}
|