import { describe, expect, test } from 'vitest'
import { cozify } from '../src/utils/sanitizer'

describe('cozify()', async () => {
  const baseUrl = 'https://cozy.pub'

  test('should remove scripts', async () => {
    const html = '<h1>HELLO</h1><script>console.log()</script>'
    const result = await cozify(html, baseUrl)
    expect(result).not.toContain('<script>')
  })

  test('should remove target=_blank from links', async () => {
    const html = "<a href=# target='_blank'>hey</a>"
    const result = await cozify(html, baseUrl)
    expect(result).not.toContain('target')
    console.log(result)
  })

  test('should add base url to href of links', async () => {
    const html = '<a href="#">hey</a>'
    const result = await cozify(html, baseUrl)
    expect(result).toContain('href="https://cozy.pub?url=#"')
  })

  test('should add prefetch=true to links', async () => {
    const html = '<a href=#>hey</a>'
    const result = await cozify(html, baseUrl)
    expect(result).toContain('prefetch="true"')
  })
})