ComfyUI_frontend/src/renderer/extensions/vueNodes/widgets/components/WidgetButton.test.ts

import { mount } from '@vue/test-utils'
import Button from 'primevue/button'
import type { ButtonProps } from 'primevue/button'
import PrimeVue from 'primevue/config'
import { describe, expect, it, vi } from 'vitest'
import { ref, watch } from 'vue'

import WidgetButton from '@/renderer/extensions/vueNodes/widgets/components/WidgetButton.vue'
import type { SimplifiedWidget } from '@/types/simplifiedWidget'

describe('WidgetButton Interactions', () => {
  const createMockWidget = (
    options: Partial<ButtonProps> = {},
    callback?: () => void,
    name: string = 'test_button'
  ): SimplifiedWidget<void> => {
    const valueRef = ref()
    if (callback) watch(valueRef, callback)
    return {
      name,
      type: 'button',
      value: () => valueRef,
      options
    }
  }

  const mountComponent = (widget: SimplifiedWidget<void>, readonly = false) => {
    return mount(WidgetButton, {
      global: {
        plugins: [PrimeVue],
        components: { Button }
      },
      props: {
        widget,
        readonly
      }
    })
  }

  const clickButton = async (wrapper: ReturnType<typeof mount>) => {
    const button = wrapper.findComponent({ name: 'Button' })
    await button.trigger('click')
    return button
  }

  describe('Click Handling', () => {
    it('calls callback when button is clicked', async () => {
      const mockCallback = vi.fn()
      const widget = createMockWidget({}, mockCallback)
      const wrapper = mountComponent(widget)

      await clickButton(wrapper)

      expect(mockCallback).toHaveBeenCalledTimes(1)
    })

    it('handles missing callback gracefully', async () => {
      const widget = createMockWidget({}, undefined)
      const wrapper = mountComponent(widget)

      // Should not throw error when clicking without callback
      await expect(clickButton(wrapper)).resolves.toBeDefined()
    })

    it('calls callback multiple times when clicked multiple times', async () => {
      const mockCallback = vi.fn()
      const widget = createMockWidget({}, mockCallback)
      const wrapper = mountComponent(widget)

      const numClicks = 8

      for (let i = 0; i < numClicks; i++) {
        await clickButton(wrapper)
      }

      expect(mockCallback).toHaveBeenCalledTimes(numClicks)
    })
  })

  describe('Component Rendering', () => {
    it('renders button component', () => {
      const widget = createMockWidget()
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.exists()).toBe(true)
    })

    it('renders widget text when name is provided', () => {
      const widget = createMockWidget()
      const wrapper = mountComponent(widget)

      expect(wrapper.text()).toBe('test_button')
    })

    it('sets button size to small', () => {
      const widget = createMockWidget()
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('size')).toBe('small')
    })

    it('passes widget options to button component', () => {
      const buttonOptions = {
        label: 'Custom Label',
        icon: 'pi pi-check',
        severity: 'success' as const
      }
      const widget = createMockWidget(buttonOptions)
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('label')).toBe('Custom Label')
      expect(button.props('icon')).toBe('pi pi-check')
      expect(button.props('severity')).toBe('success')
    })
  })

  describe('Widget Options', () => {
    it('handles button with text only', () => {
      const widget = createMockWidget({ label: 'Click Me' })
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('label')).toBe('Click Me')
      expect(button.props('icon')).toBeNull()
    })

    it('handles button with icon only', () => {
      const widget = createMockWidget({ icon: 'pi pi-star' })
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('icon')).toBe('pi pi-star')
    })

    it('handles button with both text and icon', () => {
      const widget = createMockWidget({
        label: 'Save',
        icon: 'pi pi-save'
      })
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('label')).toBe('Save')
      expect(button.props('icon')).toBe('pi pi-save')
    })

    it.for([
      'secondary',
      'success',
      'info',
      'warning',
      'danger',
      'help',
      'contrast'
    ] as const)('handles button severity: %s', (severity) => {
      const widget = createMockWidget({ severity })
      const wrapper = mountComponent(widget)
      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.props('severity')).toBe(severity)
    })

    it.for(['outlined', 'text'] as const)(
      'handles button variant: %s',
      (variant) => {
        const widget = createMockWidget({ variant })
        const wrapper = mountComponent(widget)
        const button = wrapper.findComponent({ name: 'Button' })
        expect(button.props('variant')).toBe(variant)
      }
    )
  })

  describe('Edge Cases', () => {
    it('handles widget with no options', () => {
      const widget = createMockWidget()
      const wrapper = mountComponent(widget)

      const button = wrapper.findComponent({ name: 'Button' })
      expect(button.exists()).toBe(true)
    })

    it('handles callback that throws error', async () => {
      const mockCallback = vi.fn(() => {
        throw new Error('Callback error')
      })
      const widget = createMockWidget({}, mockCallback)
      const wrapper = mountComponent(widget)

      // Should not break the component when callback throws
      await expect(clickButton(wrapper)).rejects.toThrow('Callback error')
      expect(mockCallback).toHaveBeenCalledTimes(1)
    })

    it('handles rapid consecutive clicks', async () => {
      const mockCallback = vi.fn()
      const widget = createMockWidget({}, mockCallback)
      const wrapper = mountComponent(widget)

      for (let i = 0; i < 16; i++) await clickButton(wrapper)

      expect(mockCallback).toHaveBeenCalledTimes(16)
    })
  })
})