index.test.js

import { unified } from "unified"
import remark from "remark-parse"
import remark2rehype from "remark-rehype"
import stringify from "rehype-stringify"
import codeTitle from "./index.js"
import assert from "assert"
import * as vfile from "to-vfile"

function compile(file, opt) {
  return unified()
    .use(remark)
    .use(remark2rehype)
    .use(codeTitle, opt)
    .use(stringify)
    .processSync(vfile.readSync("./__example__/" + file))
    .toString()
}

describe("Rehype-code-title", () => {
  it("No Title", () => {
    const got = compile("1.md")
    assert.ok(got.match(/<h1>/) === null)
  })

  it("No Title: Custom class name", () => {
    const got = compile("1.md", { className: "my-code" })
    assert.ok(
      got.match(/<h1>/) === null && got.match('class="my-code"') === null
    )
  })

  it("Has title", () => {
    const got = compile("2.md")
    assert.ok(got.match("<h1>pages/_app.tsx</h1>") !== null)
  })

  it("Has title: Custom class name", () => {
    const got = compile("2.md", { className: "my-code" })
    assert.ok(
      got.match("<h1>pages/_app.tsx</h1>") !== null &&
        got.match('class="my-code"') !== null
    )
  })
})