dify/web/app/components/datasets/preview/__tests__/container.spec.tsx

import { render, screen } from '@testing-library/react'
import { beforeEach, describe, expect, it, vi } from 'vitest'
import PreviewContainer from '../container'

// Tests for PreviewContainer - a layout wrapper with header and scrollable content area
describe('PreviewContainer', () => {
  beforeEach(() => {
    vi.clearAllMocks()
  })

  describe('Rendering', () => {
    it('should render header content in a header element', () => {
      render(<PreviewContainer header={<span>Header Title</span>}>Body</PreviewContainer>)

      expect(screen.getByText('Header Title'))!.toBeInTheDocument()
      const headerEl = screen.getByText('Header Title').closest('header')
      expect(headerEl)!.toBeInTheDocument()
    })

    it('should render children in the content area', () => {
      render(
        <PreviewContainer header="Header" data-testid="inner-container">
          Main content
        </PreviewContainer>,
      )

      const contentEl = screen.getByTestId('inner-container').lastElementChild
      expect(contentEl)!.toHaveTextContent('Main content')
    })

    it('should render both header and children simultaneously', () => {
      render(
        <PreviewContainer header={<h2>My Header</h2>}>
          <p>Body paragraph</p>
        </PreviewContainer>,
      )

      expect(screen.getByText('My Header'))!.toBeInTheDocument()
      expect(screen.getByText('Body paragraph'))!.toBeInTheDocument()
    })

    it('should render without children', () => {
      render(<PreviewContainer header="Header" data-testid="inner-container" />)

      const contentEl = screen.getByTestId('inner-container').lastElementChild
      expect(contentEl)!.toBeInTheDocument()
      expect(contentEl!.childElementCount).toBe(0)
    })
  })

  describe('Props', () => {
    it('should apply className to the outer wrapper div', () => {
      const { container } = render(
        <PreviewContainer header="Header" className="outer-class">
          Content
        </PreviewContainer>,
      )

      expect(container.firstElementChild)!.toHaveClass('outer-class')
    })

    it('should apply mainClassName to the content area', () => {
      render(
        <PreviewContainer header="Header" mainClassName="custom-main" data-testid="inner-container">
          Content
        </PreviewContainer>,
      )

      const contentEl = screen.getByTestId('inner-container').lastElementChild
      expect(contentEl)!.toHaveClass('custom-main')
      expect(contentEl)!.toHaveClass('w-full', 'grow', 'overflow-y-auto', 'px-6', 'py-5')
    })

    it('should forward ref to the inner container div', () => {
      const ref = vi.fn()
      render(
        <PreviewContainer header="Header" ref={ref}>
          Content
        </PreviewContainer>,
      )

      expect(ref).toHaveBeenCalled()
      const refArg = ref.mock.calls[0]![0]
      expect(refArg).toBeInstanceOf(HTMLDivElement)
    })

    it('should pass rest props to the inner container div', () => {
      render(
        <PreviewContainer header="Header" data-testid="inner-container" id="container-1">
          Content
        </PreviewContainer>,
      )

      const inner = screen.getByTestId('inner-container')
      expect(inner)!.toHaveAttribute('id', 'container-1')
    })

    it('should render ReactNode as header', () => {
      render(
        <PreviewContainer
          header={
            <div data-testid="complex-header">
              <span>Complex</span>
            </div>
          }
        >
          Content
        </PreviewContainer>,
      )

      expect(screen.getByTestId('complex-header'))!.toBeInTheDocument()
      expect(screen.getByText('Complex'))!.toBeInTheDocument()
    })
  })

  // Layout structure tests
  describe('Layout Structure', () => {
    it('should have header with border-b styling', () => {
      render(<PreviewContainer header="Header">Content</PreviewContainer>)

      const headerEl = screen.getByText('Header').closest('header')
      expect(headerEl)!.toHaveClass('border-b', 'border-divider-subtle')
    })

    it('should have inner div with flex column layout', () => {
      render(
        <PreviewContainer header="Header" data-testid="inner">
          Content
        </PreviewContainer>,
      )

      const inner = screen.getByTestId('inner')
      expect(inner)!.toHaveClass('flex', 'h-full', 'w-full', 'flex-col')
    })

    it('should have content area with overflow-y-auto for scrolling', () => {
      render(
        <PreviewContainer header="Header" data-testid="inner-container">
          Content
        </PreviewContainer>,
      )

      expect(screen.getByTestId('inner-container').lastElementChild)!.toHaveClass('overflow-y-auto')
    })
  })

  // DisplayName test
  describe('DisplayName', () => {
    it('should have correct displayName', () => {
      expect(PreviewContainer.displayName).toBe('PreviewContainer')
    })
  })

  describe('Edge Cases', () => {
    it('should render with empty string header', () => {
      render(<PreviewContainer header="">Content</PreviewContainer>)

      const headerEl = screen.getByRole('banner')
      expect(headerEl)!.toBeInTheDocument()
    })

    it('should render with null children', () => {
      render(
        <PreviewContainer header="Header" data-testid="inner-container">
          {null}
        </PreviewContainer>,
      )

      expect(screen.getByTestId('inner-container').lastElementChild)!.toBeInTheDocument()
    })

    it('should render with multiple children', () => {
      render(
        <PreviewContainer header="Header">
          <div>Child 1</div>
          <div>Child 2</div>
          <div>Child 3</div>
        </PreviewContainer>,
      )

      expect(screen.getByText('Child 1'))!.toBeInTheDocument()
      expect(screen.getByText('Child 2'))!.toBeInTheDocument()
      expect(screen.getByText('Child 3'))!.toBeInTheDocument()
    })

    it('should not crash on re-render with different props', () => {
      const { rerender } = render(
        <PreviewContainer header="First" className="a">
          Content A
        </PreviewContainer>,
      )

      rerender(
        <PreviewContainer header="Second" className="b" mainClassName="new-main">
          Content B
        </PreviewContainer>,
      )

      expect(screen.getByText('Second'))!.toBeInTheDocument()
      expect(screen.getByText('Content B'))!.toBeInTheDocument()
    })
  })
})