vitest.setup.ts

import { beforeAll, vi } from 'vitest'
import { setProjectAnnotations } from '@storybook/react-vite'
import * as projectAnnotations from './.storybook/preview'

import type { ReactPortal } from 'react'

setProjectAnnotations(projectAnnotations)

beforeAll(() => {
  vi.stubGlobal(
    'ResizeObserver',
    vi.fn().mockImplementation(() => ({
      observe: vi.fn(),
      unobserve: vi.fn(),
      disconnect: vi.fn(),
    })),
  )
  vi.stubGlobal(
    'IntersectionObserver',
    vi.fn().mockImplementation(() => ({
      observe() {
        return null
      },
      disconnect() {
        return null
      },
    })),
  )
  vi.stubGlobal(
    'matchMedia',
    vi.fn().mockImplementation((query: string) => ({
      matches: false,
      media: query,
      onchange: null,
      addListener: vi.fn(),
      removeListener: vi.fn(),
      addEventListener: vi.fn(),
      removeEventListener: vi.fn(),
      dispatchEvent: vi.fn(),
    })),
  )

  vi.mock('@react-aria/utils', async () => {
    const utils = await vi.importActual('@react-aria/utils')

    return {
      ...utils,
      useId: () => 'test-id',
    }
  })

  vi.mock('@react-aria/overlays', async () => {
    const overlays = await vi.importActual('@react-aria/overlays')

    return {
      ...overlays,
      Overlay: vi.fn(({ children }) => children as ReactPortal),
    }
  })

  vi.mock('react-stately', async () => {
    const stately =
      await vi.importActual<typeof import('react-stately')>('react-stately')

    return {
      ...stately,
      useOverlayTriggerState: (
        ...args: Parameters<typeof stately.useOverlayTriggerState>
      ) => {
        const state = stately.useOverlayTriggerState(...args)

        return { ...state, isOpen: true }
      },
    }
  })
})