import type { Mock } from 'vite-plus/test'
import { expect } from 'vite-plus/test'
import type { Component } from 'vue'

export const assertOpenContextMenu = async (
  openContextMenuMock: Mock,
  menu: Component,
  props: Record<string, any> = {},
) => {
  const firstArg = openContextMenuMock.mock.calls[0][0]
  const secondArg = openContextMenuMock.mock.calls[0][1]
  const thirdArg = openContextMenuMock.mock.calls[0][2]

  if (typeof firstArg.__asyncLoader === 'function') {
    const actualMenu = await firstArg.__asyncLoader()
    expect(actualMenu).toBe(menu)
  } else {
    expect(firstArg).toBe(menu)
  }

  expect(secondArg).toBeInstanceOf(MouseEvent)
  expect(thirdArg).toEqual(props)
}

export const assertOpenModal = async (openModalMock: Mock, modal: Component, props?: Record<string, any>) => {
  expect(openModalMock).toHaveBeenCalled()
  const firstArg = openModalMock.mock.calls[0][0]
  const secondArg = openModalMock.mock.calls[0][1]

  if (typeof firstArg.__asyncLoader === 'function') {
    const actualModal = await firstArg.__asyncLoader()
    expect(actualModal).toBe(modal)
  } else {
    expect(firstArg).toBe(modal)
  }

  if (props !== undefined) {
    expect(secondArg).toEqual(props)
  }
}