packages/desktop/test/e2e/source-math.spec.ts - develop - githubmirror/marktext

packages/desktop/test/e2e/source-math.spec.ts
95 строк3 KB
Ran Luo
chore: convert repo to pnpm monorepo (packages/desktop, muyajs, website) (#4302)
29 май 2026, 05:25
29 май 2026, 05:25565bfcd
import { expect, test } from '@playwright/test'
import type { ElectronApplication, Page } from 'playwright'
import { launchWithMarkdown, enterSourceMode } from './helpers'

// Regression test for https://github.com/marktext/marktext/issues/4121
// Underscores inside inline math (`$...$`) and block math (`$$...$$`) must
// not be highlighted as Markdown emphasis in the source view: in math mode
// they are subscript operators, not italic delimiters.
const FIXTURE = [
  '$\\text{F}_\\text{A} = \\text{F}_\\text{B}$ vs. $F_A = F_B$',
  '',
  '$$',
  '\\sum_{i=1}^{n} a_{i} = b_{i}',
  '$$',
  '',
  'I owe $5 and you owe $10 only.',
  ''
].join('\n')

interface CMInstance {
  lastLine(): number
  getLine(n: number): string
  getTokenAt(pos: { line: number; ch: number }, precise?: boolean): unknown
}

const readSourceState = async(page: Page) => {
  await page.waitForFunction(
    () => {
      const root = document.querySelector('.source-code .CodeMirror') as
        | (Element & { CodeMirror?: CMInstance })
        | null
      if (!root || !root.CodeMirror) return false
      const cm = root.CodeMirror
      const last = cm.lastLine()
      cm.getTokenAt({ line: last, ch: cm.getLine(last).length }, true)
      return true
    },
    null,
    { timeout: 5000 }
  )

  return page.evaluate(() => {
    const emTexts = Array.from(document.querySelectorAll('.source-code .CodeMirror .cm-em')).map(
      (s) => s.textContent
    )
    const mathInline = document.querySelectorAll('.source-code .CodeMirror .cm-math-inline').length
    const mathBlock = document.querySelectorAll('.source-code .CodeMirror .cm-math-block').length
    return { emTexts, mathInline, mathBlock }
  })
}

test.describe('Source view: math tokenization (#4121)', () => {
  let app: ElectronApplication
  let page: Page

  test.beforeAll(async() => {
    const launched = await launchWithMarkdown(FIXTURE)
    app = launched.app
    page = launched.page
    await enterSourceMode(page, app)
  })

  test.afterAll(async() => {
    if (app) await app.close()
  })

  test('underscores inside $...$ are not styled as Markdown emphasis', async() => {
    const { emTexts, mathInline } = await readSourceState(page)
    // There is no Markdown emphasis (`_word_` or `*word*`) outside math in the
    // fixture, so any `.cm-em` span proves the bug.
    expect(emTexts).toEqual([])
    // And the inline-math regions must actually be classified as math.
    expect(mathInline).toBeGreaterThan(0)
  })

  test('underscores inside $$...$$ block math are not styled as emphasis', async() => {
    const { mathBlock } = await readSourceState(page)
    expect(mathBlock).toBeGreaterThan(0)
  })

  test('a lone $ followed by no closing $ does not enter math mode', async() => {
    // The fixture's last paragraph has `$5 ... $10 only.` (no third $),
    // so any `$` after the second one must not start an unbounded math span.
    const lastLineHasMath = await page.evaluate(() => {
      const root = document.querySelector('.source-code .CodeMirror')
      if (!root) return false
      const spans = root.querySelectorAll('.cm-math-inline, .cm-math-block')
      for (const span of spans) {
        if (span.textContent && span.textContent.includes('only')) return true
      }
      return false
    })
    expect(lastLineHasMath).toBe(false)
  })
})