dify/web/app/components/header/account-setting/model-provider-page/hooks.spec.ts

import { renderHook } from '@testing-library/react'
import { useLanguage } from './hooks'
import { useContext } from 'use-context-selector'
import { after } from 'node:test'

jest.mock('swr', () => ({
  __esModule: true,
  default: jest.fn(), // mock useSWR
  useSWRConfig: jest.fn(),
}))

// mock use-context-selector
jest.mock('use-context-selector', () => ({
  useContext: jest.fn(),
}))

// mock service/common functions
jest.mock('@/service/common', () => ({
  fetchDefaultModal: jest.fn(),
  fetchModelList: jest.fn(),
  fetchModelProviderCredentials: jest.fn(),
  fetchModelProviders: jest.fn(),
  getPayUrl: jest.fn(),
}))

// mock context hooks
jest.mock('@/context/i18n', () => ({
  __esModule: true,
  default: jest.fn(),
}))

jest.mock('@/context/provider-context', () => ({
  useProviderContext: jest.fn(),
}))

jest.mock('@/context/modal-context', () => ({
  useModalContextSelector: jest.fn(),
}))

jest.mock('@/context/event-emitter', () => ({
  useEventEmitterContextContext: jest.fn(),
}))

// mock plugins
jest.mock('@/app/components/plugins/marketplace/hooks', () => ({
  useMarketplacePlugins: jest.fn(),
}))

jest.mock('@/app/components/plugins/marketplace/utils', () => ({
  getMarketplacePluginsByCollectionId: jest.fn(),
}))

jest.mock('./provider-added-card', () => jest.fn())

after(() => {
  jest.resetModules()
  jest.clearAllMocks()
})

describe('useLanguage', () => {
  it('should replace hyphen with underscore in locale', () => {
    (useContext as jest.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 jest.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 jest.Mock).mockReturnValue({
      locale: 'zh-Hans-CN',
    })

    const { result } = renderHook(() => useLanguage())
    expect(result.current).toBe('zh_Hans-CN')
  })
})