dify/web/app/components/base/notion-connector/index.spec.tsx

import { render, screen } from '@testing-library/react'
import userEvent from '@testing-library/user-event'
import { describe, expect, it, vi } from 'vitest'
import NotionConnector from './index'

describe('NotionConnector', () => {
  it('should render the layout and actual sub-components (Icons & Button)', () => {
    const { container } = render(<NotionConnector onSetting={vi.fn()} />)

    // Verify Title & Tip translations
    expect(screen.getByText('datasetCreation.stepOne.notionSyncTitle')).toBeInTheDocument()
    expect(screen.getByText('datasetCreation.stepOne.notionSyncTip')).toBeInTheDocument()

    const notionWrapper = container.querySelector('.h-12.w-12')
    const dotsWrapper = container.querySelector('.system-md-semibold')

    expect(notionWrapper?.querySelector('svg')).toBeInTheDocument()
    expect(dotsWrapper?.querySelector('svg')).toBeInTheDocument()

    const button = screen.getByRole('button', {
      name: /datasetcreation.stepone.connect/i,
    })

    expect(button).toBeInTheDocument()
    expect(button).toHaveClass('btn', 'btn-primary')
  })

  it('should trigger the onSetting callback when the real button is clicked', async () => {
    const onSetting = vi.fn()
    const user = userEvent.setup()
    render(<NotionConnector onSetting={onSetting} />)

    const button = screen.getByRole('button', {
      name: /datasetcreation.stepone.connect/i,
    })

    await user.click(button)

    expect(onSetting).toHaveBeenCalledTimes(1)
  })

  it('should maintain the correct visual hierarchy classes', () => {
    const { container } = render(<NotionConnector onSetting={vi.fn()} />)

    // Verify the outer container has the specific workflow-process-bg
    const mainContainer = container.firstChild
    expect(mainContainer).toHaveClass('bg-workflow-process-bg', 'rounded-2xl', 'p-6')
  })
})