Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/vitest-dev/vitest. Next generation testing framework powered by Vite. vitest.dev
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275/** * Copyright (c) Meta Platforms, Inc. and affiliates. * * This source code is licensed under the MIT license found in the * LICENSE file in the root directory of this source tree. */
import type { Diff } from './cleanupSemantic'import { DIFF_DELETE, DIFF_EQUAL, DIFF_INSERT } from './cleanupSemantic'import type { DiffOptionsColor, DiffOptionsNormalized } from './types'
function formatTrailingSpaces(line: string, trailingSpaceFormatter: DiffOptionsColor): string { return line.replace(/\s+$/, match => trailingSpaceFormatter(match))}
function printDiffLine(line: string, isFirstOrLast: boolean, color: DiffOptionsColor, indicator: string, trailingSpaceFormatter: DiffOptionsColor, emptyFirstOrLastLinePlaceholder: string): string { return line.length !== 0 ? color( `${indicator} ${formatTrailingSpaces(line, trailingSpaceFormatter)}`, ) : indicator !== ' ' ? color(indicator) : (isFirstOrLast && emptyFirstOrLastLinePlaceholder.length !== 0) ? color(`${indicator} ${emptyFirstOrLastLinePlaceholder}`) : ''}
function printDeleteLine(line: string, isFirstOrLast: boolean, { aColor, aIndicator, changeLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder,}: DiffOptionsNormalized): string { return printDiffLine( line, isFirstOrLast, aColor, aIndicator, changeLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder, )}
function printInsertLine(line: string, isFirstOrLast: boolean, { bColor, bIndicator, changeLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder,}: DiffOptionsNormalized): string { return printDiffLine( line, isFirstOrLast, bColor, bIndicator, changeLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder, )}
function printCommonLine(line: string, isFirstOrLast: boolean, { commonColor, commonIndicator, commonLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder,}: DiffOptionsNormalized): string { return printDiffLine( line, isFirstOrLast, commonColor, commonIndicator, commonLineTrailingSpaceColor, emptyFirstOrLastLinePlaceholder, )}
// In GNU diff format, indexes are one-based instead of zero-based.function createPatchMark(aStart: number, aEnd: number, bStart: number, bEnd: number, { patchColor }: DiffOptionsNormalized): string { return patchColor( `@@ -${aStart + 1},${aEnd - aStart} +${bStart + 1},${bEnd - bStart} @@`, )}
// jest --no-expand//// Given array of aligned strings with inverse highlight formatting,// return joined lines with diff formatting (and patch marks, if needed).export function joinAlignedDiffsNoExpand(diffs: Array<Diff>, options: DiffOptionsNormalized): string { const iLength = diffs.length const nContextLines = options.contextLines const nContextLines2 = nContextLines + nContextLines
// First pass: count output lines and see if it has patches. let jLength = iLength let hasExcessAtStartOrEnd = false let nExcessesBetweenChanges = 0 let i = 0 while (i !== iLength) { const iStart = i while (i !== iLength && diffs[i][0] === DIFF_EQUAL) i += 1
if (iStart !== i) { if (iStart === 0) { // at start if (i > nContextLines) { jLength -= i - nContextLines // subtract excess common lines hasExcessAtStartOrEnd = true } } else if (i === iLength) { // at end const n = i - iStart if (n > nContextLines) { jLength -= n - nContextLines // subtract excess common lines hasExcessAtStartOrEnd = true } } else { // between changes const n = i - iStart if (n > nContextLines2) { jLength -= n - nContextLines2 // subtract excess common lines nExcessesBetweenChanges += 1 } } }
while (i !== iLength && diffs[i][0] !== DIFF_EQUAL) i += 1 }
const hasPatch = nExcessesBetweenChanges !== 0 || hasExcessAtStartOrEnd if (nExcessesBetweenChanges !== 0) jLength += nExcessesBetweenChanges + 1 // add patch lines else if (hasExcessAtStartOrEnd) jLength += 1 // add patch line
const jLast = jLength - 1
const lines: Array<string> = []
let jPatchMark = 0 // index of placeholder line for current patch mark if (hasPatch) lines.push('') // placeholder line for first patch mark
// Indexes of expected or received lines in current patch: let aStart = 0 let bStart = 0 let aEnd = 0 let bEnd = 0
const pushCommonLine = (line: string): void => { const j = lines.length lines.push(printCommonLine(line, j === 0 || j === jLast, options)) aEnd += 1 bEnd += 1 }
const pushDeleteLine = (line: string): void => { const j = lines.length lines.push(printDeleteLine(line, j === 0 || j === jLast, options)) aEnd += 1 }
const pushInsertLine = (line: string): void => { const j = lines.length lines.push(printInsertLine(line, j === 0 || j === jLast, options)) bEnd += 1 }
// Second pass: push lines with diff formatting (and patch marks, if needed). i = 0 while (i !== iLength) { let iStart = i while (i !== iLength && diffs[i][0] === DIFF_EQUAL) i += 1
if (iStart !== i) { if (iStart === 0) { // at beginning if (i > nContextLines) { iStart = i - nContextLines aStart = iStart bStart = iStart aEnd = aStart bEnd = bStart }
for (let iCommon = iStart; iCommon !== i; iCommon += 1) pushCommonLine(diffs[iCommon][1]) } else if (i === iLength) { // at end const iEnd = i - iStart > nContextLines ? iStart + nContextLines : i
for (let iCommon = iStart; iCommon !== iEnd; iCommon += 1) pushCommonLine(diffs[iCommon][1]) } else { // between changes const nCommon = i - iStart
if (nCommon > nContextLines2) { const iEnd = iStart + nContextLines
for (let iCommon = iStart; iCommon !== iEnd; iCommon += 1) pushCommonLine(diffs[iCommon][1])
lines[jPatchMark] = createPatchMark( aStart, aEnd, bStart, bEnd, options, ) jPatchMark = lines.length lines.push('') // placeholder line for next patch mark
const nOmit = nCommon - nContextLines2 aStart = aEnd + nOmit bStart = bEnd + nOmit aEnd = aStart bEnd = bStart
for (let iCommon = i - nContextLines; iCommon !== i; iCommon += 1) pushCommonLine(diffs[iCommon][1]) } else { for (let iCommon = iStart; iCommon !== i; iCommon += 1) pushCommonLine(diffs[iCommon][1]) } } }
while (i !== iLength && diffs[i][0] === DIFF_DELETE) { pushDeleteLine(diffs[i][1]) i += 1 }
while (i !== iLength && diffs[i][0] === DIFF_INSERT) { pushInsertLine(diffs[i][1]) i += 1 } }
if (hasPatch) lines[jPatchMark] = createPatchMark(aStart, aEnd, bStart, bEnd, options)
return lines.join('\n')}
// jest --expand//// Given array of aligned strings with inverse highlight formatting,// return joined lines with diff formatting.export function joinAlignedDiffsExpand(diffs: Array<Diff>, options: DiffOptionsNormalized): string { return diffs .map((diff: Diff, i: number, diffs: Array<Diff>): string => { const line = diff[1] const isFirstOrLast = i === 0 || i === diffs.length - 1
switch (diff[0]) { case DIFF_DELETE: return printDeleteLine(line, isFirstOrLast, options)
case DIFF_INSERT: return printInsertLine(line, isFirstOrLast, options)
default: return printCommonLine(line, isFirstOrLast, options) } }) .join('\n')}