dify/web/hooks/use-timestamp.spec.ts

import type { ReactNode } from 'react'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import { renderHook } from '@testing-library/react'
import { createElement } from 'react'
import { userProfileQueryOptions } from '@/features/account-profile/client'
import { createAccountProfileQueryWrapper } from '@/test/account-profile-query'
import useTimestamp from './use-timestamp'

vi.mock('@/context/app-context', () => ({
  useAppContext: vi.fn(() => ({
    userProfile: {
      id: '8b18e24b-1ac8-4262-aa5c-e9aa95c76846',
      name: 'test',
      avatar: null,
      avatar_url: null,
      email: 'test@dify.ai',
      is_password_set: false,
      interface_language: 'zh-Hans',
      interface_theme: 'light',
      timezone: 'Asia/Shanghai',
      last_login_at: 1744188761,
      last_login_ip: '127.0.0.1',
      created_at: 1728444483,
    },
  })),
}))

const createEmptyQueryWrapper = () => {
  const queryClient = new QueryClient({
    defaultOptions: {
      queries: {
        retry: false,
      },
    },
  })

  function EmptyQueryWrapper({ children }: { children: ReactNode }) {
    return createElement(QueryClientProvider, { client: queryClient }, children)
  }

  return { queryClient, wrapper: EmptyQueryWrapper }
}

describe('useTimestamp', () => {
  describe('formatTime', () => {
    it('should format unix timestamp correctly', () => {
      const { result } = renderHook(() => useTimestamp(), { wrapper: createAccountProfileQueryWrapper() })
      const timestamp = 1704132000

      expect(result.current.formatTime(timestamp, 'YYYY-MM-DD HH:mm:ss'))
        .toBe('2024-01-02 02:00:00')
    })

    it('should format with different patterns', () => {
      const { result } = renderHook(() => useTimestamp(), { wrapper: createAccountProfileQueryWrapper() })
      const timestamp = 1704132000

      expect(result.current.formatTime(timestamp, 'MM/DD/YYYY'))
        .toBe('01/02/2024')

      expect(result.current.formatTime(timestamp, 'HH:mm'))
        .toBe('02:00')
    })
  })

  describe('formatDate', () => {
    it('should format date string correctly', () => {
      const { result } = renderHook(() => useTimestamp(), { wrapper: createAccountProfileQueryWrapper() })
      const dateString = '2024-01-01T12:00:00Z'

      expect(result.current.formatDate(dateString, 'YYYY-MM-DD HH:mm:ss'))
        .toBe('2024-01-01 20:00:00')
    })

    it('should format with different patterns', () => {
      const { result } = renderHook(() => useTimestamp(), { wrapper: createAccountProfileQueryWrapper() })
      const dateString = '2024-01-01T12:00:00Z'

      expect(result.current.formatDate(dateString, 'MM/DD/YYYY'))
        .toBe('01/01/2024')

      expect(result.current.formatDate(dateString, 'HH:mm'))
        .toBe('20:00')
    })
  })

  it('should not request account profile when timezone is provided', () => {
    const { queryClient, wrapper } = createEmptyQueryWrapper()

    const { result } = renderHook(() => useTimestamp({ timezone: 'UTC' }), { wrapper })

    expect(result.current.formatTime(1704132000, 'YYYY-MM-DD HH:mm')).toBe('2024-01-01 18:00')
    expect(queryClient.isFetching({ queryKey: userProfileQueryOptions().queryKey })).toBe(0)
    expect(queryClient.getQueryState(userProfileQueryOptions().queryKey)?.fetchStatus).toBe('idle')
  })
})