feat: support custom literal tag on inlineSnapshot

vitest-dev · Feb 10, 2022 · 2623543 · 2623543
1 parent 6772c5f
commit 2623543
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/packages/vitest/src/integrations/snapshot/port/inlineSnapshot.ts b/packages/vitest/src/integrations/snapshot/port/inlineSnapshot.ts
@@ -72,7 +72,7 @@ function prepareSnapString(snap: string, source: string, index: number) {
     : `${quote}\n${lines.map(i => indentNext + i).join('\n').replace(/`/g, '\\`')}\n${indent}${quote}`
 }
 
-const startRegex = /(?:toMatchInlineSnapshot|toThrowErrorMatchingInlineSnapshot)\s*\(\s*(['"`\)])/m
+const startRegex = /(?:toMatchInlineSnapshot|toThrowErrorMatchingInlineSnapshot)\s*\(\s*[\w_$]*(['"`\)])/m
 export function replaceInlineSnap(code: string, s: MagicString, index: number, newSnap: string) {
   const startMatch = startRegex.exec(code.slice(index))
   if (!startMatch)

diff --git a/test/core/test/snapshot-inline.test.ts b/test/core/test/snapshot-inline.test.ts
@@ -87,3 +87,22 @@ test('properties inline snapshot', () => {
     }
   `)
 })
+
+test('literal tag', () => {
+  const html = String.raw
+  const text = `
+<body>
+  <h1>My First Heading.</h1>
+  <p>My first paragraph.</p>
+</body>
+`
+
+  expect(text).toMatchInlineSnapshot(html`
+    "
+    <body>
+      <h1>My First Heading.</h1>
+      <p>My first paragraph.</p>
+    </body>
+    "
+  `)
+})