import { expect } from '@playwright/test'
import { comfyPageFixture as test } from './ComfyPage'

test.describe('Copy Paste', () => {
  test('Can copy and paste node', async ({ comfyPage }) => {
    await comfyPage.clickEmptyLatentNode()
    await comfyPage.page.mouse.move(10, 10)
    await comfyPage.ctrlC()
    await comfyPage.ctrlV()
    await expect(comfyPage.canvas).toHaveScreenshot('copied-node.png')
  })

  test('Can copy and paste text', async ({ comfyPage }) => {
    const textBox = comfyPage.widgetTextBox
    await textBox.click()
    const originalString = await textBox.inputValue()
    await textBox.selectText()
    await comfyPage.ctrlC()
    await comfyPage.ctrlV()
    await comfyPage.ctrlV()
    const resultString = await textBox.inputValue()
    expect(resultString).toBe(originalString + originalString)
  })

  /**
   * https://github.com/Comfy-Org/ComfyUI_frontend/issues/98
   */
  test('Paste in text area with node previously copied', async ({
    comfyPage
  }) => {
    await comfyPage.clickEmptyLatentNode()
    await comfyPage.ctrlC()
    const textBox = comfyPage.widgetTextBox
    await textBox.click()
    await textBox.inputValue()
    await textBox.selectText()
    await comfyPage.ctrlC()
    await comfyPage.ctrlV()
    await comfyPage.ctrlV()
    await expect(comfyPage.canvas).toHaveScreenshot(
      'paste-in-text-area-with-node-previously-copied.png'
    )
  })

  test('Copy text area does not copy node', async ({ comfyPage }) => {
    const textBox = comfyPage.widgetTextBox
    await textBox.click()
    await textBox.inputValue()
    await textBox.selectText()
    await comfyPage.ctrlC()
    // Unfocus textbox.
    await comfyPage.page.mouse.click(10, 10)
    await comfyPage.ctrlV()
    await expect(comfyPage.canvas).toHaveScreenshot('no-node-copied.png')
  })
})