From 5a848c0e778f901bbbc45cfddcd97d3c5373ff7a Mon Sep 17 00:00:00 2001
From: poyoho <907415276@qq.com>
Date: Wed, 9 Feb 2022 21:44:10 +0800
Subject: [PATCH] fix(snapshot): received add prepareExpected (#711)

---
 .../src/integrations/snapshot/port/state.ts      |  2 +-
 .../test/__snapshots__/shapshots.test.ts.snap    | 12 ++++++++++++
 test/snapshots/test/shapshots.test.ts            | 16 ++++++++++++++++
 3 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 test/snapshots/test/__snapshots__/shapshots.test.ts.snap

diff --git a/packages/vitest/src/integrations/snapshot/port/state.ts b/packages/vitest/src/integrations/snapshot/port/state.ts
index f07f19a77e1f..7025cd90bba5 100644
--- a/packages/vitest/src/integrations/snapshot/port/state.ts
+++ b/packages/vitest/src/integrations/snapshot/port/state.ts
@@ -193,7 +193,7 @@ export default class SnapshotState {
     const receivedSerialized = addExtraLineBreaks(serialize(received, undefined, this._snapshotFormat))
     const expected = isInline ? inlineSnapshot : this._snapshotData[key]
     const expectedTrimmed = prepareExpected(expected)
-    const pass = expectedTrimmed === receivedSerialized?.trim()
+    const pass = expectedTrimmed === prepareExpected(receivedSerialized)
     const hasSnapshot = expected !== undefined
     const snapshotIsPersisted = isInline || fs.existsSync(this._snapshotPath)
 
diff --git a/test/snapshots/test/__snapshots__/shapshots.test.ts.snap b/test/snapshots/test/__snapshots__/shapshots.test.ts.snap
new file mode 100644
index 000000000000..a3bdd937d41c
--- /dev/null
+++ b/test/snapshots/test/__snapshots__/shapshots.test.ts.snap
@@ -0,0 +1,12 @@
+// Vitest Snapshot v1
+
+exports[`multiline strings  1`] = `
+"
+export default function () {
+  function Foo() {
+  }
+
+  return Foo;
+}
+"
+`;
diff --git a/test/snapshots/test/shapshots.test.ts b/test/snapshots/test/shapshots.test.ts
index 220512dbb146..a10c2b8fdecf 100644
--- a/test/snapshots/test/shapshots.test.ts
+++ b/test/snapshots/test/shapshots.test.ts
@@ -1,3 +1,15 @@
+const println = () => {
+  const message = `
+export default function () {
+  function Foo() {
+  }
+
+  return Foo;
+}
+`
+  return message
+}
+
 test('non default snapshot format', () => {
   expect({ foo: ['bar'] }).toMatchInlineSnapshot(`
   Object {
@@ -7,3 +19,7 @@ test('non default snapshot format', () => {
   }
   `)
 })
+
+test('multiline strings ', () => {
+  expect(println()).toMatchSnapshot()
+})