// @vitest-environment happy-dom
import { expect, it, vi } from 'vitest'
import { installCodeBlockLabels } from './codeBlockLabels'

it('keeps footer labels in sync when the language changes and stops after disposal', async () => {
  const root = document.createElement('div')
  root.innerHTML = '<div class="milkdown-code-block"><button class="language-button">Python</button></div>'
  const dispose = installCodeBlockLabels(root)
  const block = root.firstElementChild as HTMLElement
  expect(block.dataset.languageLabel).toBe('Python')
  block.querySelector('button')!.textContent = 'TypeScript'
  await new Promise(resolve => setTimeout(resolve, 0))
  expect(block.dataset.languageLabel).toBe('TypeScript')
  dispose()
  block.querySelector('button')!.textContent = 'Rust'
  await new Promise(resolve => setTimeout(resolve, 0))
  expect(block.dataset.languageLabel).toBe('TypeScript')
})

it('ignores code text mutations and discovers newly inserted code blocks', async () => {
  const root = document.createElement('div')
  root.innerHTML = '<div class="milkdown-code-block"><button class="language-button">Python</button><div class="cm-content">old</div></div>'
  const dispose = installCodeBlockLabels(root)
  const scan = vi.spyOn(root, 'querySelectorAll')
  try {
    root.querySelector('.cm-content')!.textContent = 'new code'
    await new Promise(resolve => setTimeout(resolve, 0))
    expect(scan).not.toHaveBeenCalled()
    const block = document.createElement('div')
    block.className = 'milkdown-code-block'
    block.innerHTML = '<button class="language-button">Rust</button>'
    root.append(block)
    await new Promise(resolve => setTimeout(resolve, 0))
    expect(block.dataset.languageLabel).toBe('Rust')
  } finally { dispose(); scan.mockRestore() }
})