Indented inline snapshots.

e2e/__tests__/__snapshots__/toMatchInlineSnapshot.test.ts.snap

@@ -3,40 +3,40 @@
 exports[`basic support: initial write 1`] = `
 test('inline snapshots', () =>
   expect({apple: 'original value'}).toMatchInlineSnapshot(\`
-Object {
-  "apple": "original value",
-}
-\`));
+    Object {
+      "apple": "original value",
+    }
+  \`));
 
 `;
 
 exports[`basic support: snapshot mismatch 1`] = `
 test('inline snapshots', () =>
   expect({apple: 'updated value'}).toMatchInlineSnapshot(\`
-Object {
-  "apple": "original value",
-}
-\`));
+    Object {
+      "apple": "original value",
+    }
+  \`));
 
 `;
 
 exports[`basic support: snapshot passed 1`] = `
 test('inline snapshots', () =>
   expect({apple: 'original value'}).toMatchInlineSnapshot(\`
-Object {
-  "apple": "original value",
-}
-\`));
+    Object {
+      "apple": "original value",
+    }
+  \`));
 
 `;
 
 exports[`basic support: snapshot updated 1`] = `
 test('inline snapshots', () =>
   expect({apple: 'updated value'}).toMatchInlineSnapshot(\`
-Object {
-  "apple": "updated value",
-}
-\`));
+    Object {
+      "apple": "updated value",
+    }
+  \`));
 
 `;
 
@@ -45,10 +45,10 @@ test('handles property matchers', () => {
   expect({createdAt: new Date()}).toMatchInlineSnapshot(
     {createdAt: expect.any(Date)},
     \`
-Object {
-  "createdAt": Any<Date>,
-}
-\`
+    Object {
+      "createdAt": Any<Date>,
+    }
+  \`
   );
 });
 
@@ -59,10 +59,10 @@ test('handles property matchers', () => {
   expect({createdAt: "string"}).toMatchInlineSnapshot(
     {createdAt: expect.any(Date)},
     \`
-Object {
-  "createdAt": Any<Date>,
-}
-\`
+    Object {
+      "createdAt": Any<Date>,
+    }
+  \`
   );
 });
 
@@ -73,10 +73,10 @@ test('handles property matchers', () => {
   expect({createdAt: new Date()}).toMatchInlineSnapshot(
     {createdAt: expect.any(Date)},
     \`
-Object {
-  "createdAt": Any<Date>,
-}
-\`
+    Object {
+      "createdAt": Any<Date>,
+    }
+  \`
   );
 });
 
@@ -87,10 +87,10 @@ test('handles property matchers', () => {
   expect({createdAt: 'string'}).toMatchInlineSnapshot(
     {createdAt: expect.any(String)},
     \`
-Object {
-  "createdAt": Any<String>,
-}
-\`
+    Object {
+      "createdAt": Any<String>,
+    }
+  \`
   );
 });
 
@@ -139,10 +139,10 @@ test('inline snapshots', async () => {
 exports[`writes snapshots with non-literals in expect(...) 1`] = `
 it('works with inline snapshots', () => {
   expect({a: 1}).toMatchInlineSnapshot(\`
-Object {
-  "a": 1,
-}
-\`);
+    Object {
+      "a": 1,
+    }
+  \`);
 });
 
 `;

packages/jest-snapshot/src/__tests__/inline_snapshots.test.ts

@@ -199,3 +199,108 @@ test('saveInlineSnapshots() works with non-literals in expect call', () => {
     "expect({a: 'a'}).toMatchInlineSnapshot(`{a: 'a'}`);\n",
   );
 });
+
+test('saveInlineSnapshots() indents multi-line snapshots with spaces', () => {
+  const filename = path.join(__dirname, 'my.test.js');
+  (fs.readFileSync as jest.Mock).mockImplementation(
+    () =>
+      "it('is a test', () => {\n" +
+      "  expect({a: 'a'}).toMatchInlineSnapshot();\n" +
+      '});\n',
+  );
+  (prettier.resolveConfig.sync as jest.Mock).mockReturnValue({
+    bracketSpacing: false,
+    singleQuote: true,
+  });
+
+  saveInlineSnapshots(
+    [
+      {
+        frame: {column: 20, file: filename, line: 2} as Frame,
+        snapshot: `\nObject {\n  a: 'a'\n}\n`,
+      },
+    ],
+    prettier,
+    babelTraverse,
+  );
+
+  expect(fs.writeFileSync).toHaveBeenCalledWith(
+    filename,
+    "it('is a test', () => {\n" +
+      "  expect({a: 'a'}).toMatchInlineSnapshot(`\n" +
+      '    Object {\n' +
+      "      a: 'a'\n" +
+      '    }\n' +
+      '  `);\n' +
+      '});\n',
+  );
+});
+
+test('saveInlineSnapshots() indents multi-line snapshots with tabs', () => {
+  const filename = path.join(__dirname, 'my.test.js');
+  (fs.readFileSync as jest.Mock).mockImplementation(
+    () =>
+      "it('is a test', () => {\n" +
+      "  expect({a: 'a'}).toMatchInlineSnapshot();\n" +
+      '});\n',
+  );
+  (prettier.resolveConfig.sync as jest.Mock).mockReturnValue({
+    bracketSpacing: false,
+    singleQuote: true,
+    useTabs: true,
+  });
+
+  saveInlineSnapshots(
+    [
+      {
+        frame: {column: 20, file: filename, line: 2} as Frame,
+        snapshot: `\nObject {\n  a: 'a'\n}\n`,
+      },
+    ],
+    prettier,
+    babelTraverse,
+  );
+
+  expect(fs.writeFileSync).toHaveBeenCalledWith(
+    filename,
+    "it('is a test', () => {\n" +
+      "\texpect({a: 'a'}).toMatchInlineSnapshot(`\n" +
+      '\t\tObject {\n' +
+      "\t\t  a: 'a'\n" +
+      '\t\t}\n' +
+      '\t`);\n' +
+      '});\n',
+  );
+});
+
+test('saveInlineSnapshots() indents snapshots after prettier reformats', () => {
+  const filename = path.join(__dirname, 'my.test.js');
+  (fs.readFileSync as jest.Mock).mockImplementation(
+    () => "it('is a test', () => expect({a: 'a'}).toMatchInlineSnapshot());\n",
+  );
+  (prettier.resolveConfig.sync as jest.Mock).mockReturnValue({
+    bracketSpacing: false,
+    singleQuote: true,
+  });
+
+  saveInlineSnapshots(
+    [
+      {
+        frame: {column: 40, file: filename, line: 1} as Frame,
+        snapshot: `\nObject {\n  a: 'a'\n}\n`,
+      },
+    ],
+    prettier,
+    babelTraverse,
+  );
+
+  expect(fs.writeFileSync).toHaveBeenCalledWith(
+    filename,
+    "it('is a test', () =>\n" +
+      "  expect({a: 'a'}).toMatchInlineSnapshot(`\n" +
+      '    Object {\n' +
+      "      a: 'a'\n" +
+      '    }\n' +
+      '  `));\n',
+  );
+});

packages/jest-snapshot/src/index.ts

@@ -73,6 +73,44 @@ const printName = (
   );
 };
 
+function stripAddedIndentation(inlineSnapshot: string) {
+  const match = inlineSnapshot.match(/^([^\S\n]*)\S/m);
+  if (!match || !match[1]) {
+    // No indentation.
+    return inlineSnapshot;
+  }
+
+  const indentation = match[1];
+  const lines = inlineSnapshot.split('\n');
+  if (lines.length <= 2) {
+    // Must be at least 3 lines.
+    return inlineSnapshot;
+  }
+
+  if (lines[0].trim() !== '' || lines[lines.length - 1].trim() !== '') {
+    // If not blank first and last lines, abort.
+    return inlineSnapshot;
+  }
+
+  for (let i = 1; i < lines.length - 1; i++) {
+    if (lines[i].indexOf(indentation) !== 0) {
+      // All lines except first and last should have the same indent as the
+      // first line (or more). If this isn't the case we don't want to touch it.
+      return inlineSnapshot;
+    }
+
+    lines[i] = lines[i].substr(indentation.length);
+  }
+
+  // Last line is a special case because it won't have the same indent as others
+  // but may still have been given some indent to line up.
+  lines[lines.length - 1] = '';
+
+  // Return inline snapshot, now at indent 0.
+  inlineSnapshot = lines.join('\n');
+  return inlineSnapshot;
+}
+
 const fileExists = (filePath: Config.Path, hasteFS: HasteFS): boolean =>
   hasteFS.exists(filePath) || fs.existsSync(filePath);
 
@@ -182,7 +220,7 @@ const toMatchInlineSnapshot = function(
   return _toMatchSnapshot({
     context: this,
     expectedArgument,
-    inlineSnapshot: inlineSnapshot || '',
+    inlineSnapshot: stripAddedIndentation(inlineSnapshot || ''),
     matcherName,
     options,
     propertyMatchers,