jest snapshot - TS migration

packages/jest-message-util/src/index.ts

@@ -13,6 +13,9 @@ import micromatch from 'micromatch';
 import slash from 'slash';
 import {codeFrameColumns} from '@babel/code-frame';
 import StackUtils from 'stack-utils';
+import {Frame} from './types';
+
+export * from './types';
 
 type Path = Config.Path;
 type AssertionResult = TestResult.AssertionResult;
@@ -227,7 +230,7 @@ export const getStackTraceLines = (
   options: StackTraceOptions = {noStackTrace: false},
 ) => removeInternalStackEntries(stack.split(/\n/), options);
 
-export const getTopFrame = (lines: string[]) => {
+export const getTopFrame = (lines: string[]): Frame | null => {
   for (const line of lines) {
     if (line.includes(PATH_NODE_MODULES) || line.includes(PATH_JEST_PACKAGES)) {
       continue;
@@ -236,7 +239,7 @@ export const getTopFrame = (lines: string[]) => {
     const parsedFrame = stackUtils.parseLine(line.trim());
 
     if (parsedFrame && parsedFrame.file) {
-      return parsedFrame;
+      return parsedFrame as Frame;
     }
   }
 

packages/jest-message-util/src/types.ts

@@ -0,0 +1,5 @@
+import {StackData} from 'stack-utils';
+
+export interface Frame extends StackData {
+  file: string;
+}

packages/jest-snapshot/dts/natural-compare.d.ts

@@ -0,0 +1,4 @@
+declare module 'natural-compare' {
+  const NaturalCompare: (a: string, b: string) => number;
+  export default NaturalCompare;
+}

packages/jest-snapshot/package.json

@@ -10,6 +10,7 @@
   "main": "build/index.js",
   "dependencies": {
     "@babel/types": "^7.0.0",
+    "@jest/types": "^24.1.0",
     "chalk": "^2.0.1",
     "jest-diff": "^24.0.0",
     "jest-matcher-utils": "^24.0.0",
@@ -22,7 +23,9 @@
   },
   "devDependencies": {
     "@types/mkdirp": "^0.5.2",
+    "@types/prettier": "^1.16.1",
     "@types/semver": "^5.5.0",
+    "jest-haste-map": "^24.0.0",
     "prettier": "^1.13.4"
   },
   "engines": {

packages/jest-snapshot/src/State.js → packages/jest-snapshot/src/State.ts

@@ -4,12 +4,11 @@
  * This source code is licensed under the MIT license found in the
  * LICENSE file in the root directory of this source tree.
  *
- * @flow
  */
 
-import type {Path, SnapshotUpdateState} from 'types/Config';
-
 import fs from 'fs';
+import {Config} from '@jest/types';
+
 import {getTopFrame, getStackTraceLines} from 'jest-message-util';
 import {
   saveSnapshotFile,
@@ -19,41 +18,44 @@ import {
   testNameToKey,
   unescape,
 } from './utils';
-import {saveInlineSnapshots, type InlineSnapshot} from './inline_snapshots';
+import {saveInlineSnapshots, InlineSnapshot} from './inline_snapshots';
+import {SnapshotData} from './types';
 
-export type SnapshotStateOptions = {|
-  updateSnapshot: SnapshotUpdateState,
-  getPrettier: () => null | any,
-  getBabelTraverse: () => Function,
-  expand?: boolean,
-|};
+export type SnapshotStateOptions = {
+  updateSnapshot: Config.SnapshotUpdateState;
+  getPrettier: () => null | any;
+  getBabelTraverse: () => Function;
+  expand?: boolean;
+};
 
-export type SnapshotMatchOptions = {|
-  testName: string,
-  received: any,
-  key?: string,
-  inlineSnapshot?: string,
-  error?: Error,
-|};
+export type SnapshotMatchOptions = {
+  testName: string;
+  received: any;
+  key?: string;
+  inlineSnapshot?: string;
+  error?: Error;
+};
 
 export default class SnapshotState {
-  _counters: Map<string, number>;
-  _dirty: boolean;
-  _index: number;
-  _updateSnapshot: SnapshotUpdateState;
-  _snapshotData: {[key: string]: string};
-  _snapshotPath: Path;
-  _inlineSnapshots: Array<InlineSnapshot>;
-  _uncheckedKeys: Set<string>;
-  _getBabelTraverse: () => Function;
-  _getPrettier: () => null | any;
+  private _counters: Map<string, number>;
+  private _dirty: boolean;
+  // @ts-ignore
+  private _index: number;
+  private _updateSnapshot: Config.SnapshotUpdateState;
+  private _snapshotData: SnapshotData;
+  private _snapshotPath: Config.Path;
+  private _inlineSnapshots: Array<InlineSnapshot>;
+  private _uncheckedKeys: Set<string>;
+  private _getBabelTraverse: () => Function;
+  private _getPrettier: () => null | any;
+
   added: number;
   expand: boolean;
   matched: number;
   unmatched: number;
   updated: number;
 
-  constructor(snapshotPath: Path, options: SnapshotStateOptions) {
+  constructor(snapshotPath: Config.Path, options: SnapshotStateOptions) {
     this._snapshotPath = snapshotPath;
     const {data, dirty} = getSnapshotData(
       this._snapshotPath,
@@ -83,15 +85,15 @@ export default class SnapshotState {
     });
   }
 
-  _addSnapshot(
+  private _addSnapshot(
     key: string,
     receivedSerialized: string,
-    options: {isInline: boolean, error?: Error},
+    options: {isInline: boolean; error?: Error},
   ) {
     this._dirty = true;
     if (options.isInline) {
       const error = options.error || new Error();
-      const lines = getStackTraceLines(error.stack);
+      const lines = getStackTraceLines(error.stack || '');
       const frame = getTopFrame(lines);
       if (!frame) {
         throw new Error(
@@ -251,7 +253,7 @@ export default class SnapshotState {
     }
   }
 
-  fail(testName: string, received: any, key?: string) {
+  fail(testName: string, _: any, key?: string) {
     this._counters.set(testName, (this._counters.get(testName) || 0) + 1);
     const count = Number(this._counters.get(testName));
 

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

@@ -4,59 +4,47 @@
  * This source code is licensed under the MIT license found in the
  * LICENSE file in the root directory of this source tree.
  *
- * @flow
  */
 
 jest.mock('fs');
 jest.mock('prettier');
 
-const fs = require('fs');
-const path = require('path');
-const prettier = require('prettier');
-const babelTraverse = require('@babel/traverse').default;
+import fs from 'fs';
+import path from 'path';
+import prettier from 'prettier';
+import babelTraverse from '@babel/traverse';
 
-const {saveInlineSnapshots} = require('../inline_snapshots');
+import {saveInlineSnapshots} from '../inline_snapshots';
 
 const writeFileSync = fs.writeFileSync;
 const readFileSync = fs.readFileSync;
 const existsSync = fs.existsSync;
 const statSync = fs.statSync;
 const readdirSync = fs.readdirSync;
 beforeEach(() => {
-  // $FlowFixMe mock
   fs.writeFileSync = jest.fn();
-  // $FlowFixMe mock
   fs.readFileSync = jest.fn();
-  // $FlowFixMe mock
   fs.existsSync = jest.fn(() => true);
-  // $FlowFixMe mock
-  fs.statSync = jest.fn(filePath => ({
+  (fs.statSync as jest.Mock).mockImplementation(filePath => ({
     isDirectory: () => !filePath.endsWith('.js'),
   }));
-  // $FlowFixMe mock
   fs.readdirSync = jest.fn(() => []);
 
-  prettier.resolveConfig.sync.mockReset();
+  (prettier.resolveConfig.sync as jest.Mock).mockReset();
 });
 afterEach(() => {
-  // $FlowFixMe mock
   fs.writeFileSync = writeFileSync;
-  // $FlowFixMe mock
   fs.readFileSync = readFileSync;
-  // $FlowFixMe mock
   fs.existsSync = existsSync;
-  // $FlowFixMe mock
   fs.statSync = statSync;
-  // $FlowFixMe mock
   fs.readdirSync = readdirSync;
 });
 
 test('saveInlineSnapshots() replaces empty function call with a template literal', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => `expect(1).toMatchInlineSnapshot();\n`,
-  ): any);
+  );
 
   saveInlineSnapshots(
     [
@@ -79,12 +67,11 @@ test.each([['babylon'], ['flow'], ['typescript']])(
   'saveInlineSnapshots() replaces existing template literal - %s parser',
   parser => {
     const filename = path.join(__dirname, 'my.test.js');
-    // $FlowFixMe mock
-    fs.readFileSync = (jest.fn(
+    (fs.readFileSync as jest.Mock).mockImplementation(
       () => 'expect(1).toMatchInlineSnapshot(`2`);\n',
-    ): any);
+    );
 
-    prettier.resolveConfig.sync.mockReturnValue({parser});
+    (prettier.resolveConfig.sync as jest.Mock).mockReturnValue({parser});
 
     saveInlineSnapshots(
       [
@@ -97,7 +84,9 @@ test.each([['babylon'], ['flow'], ['typescript']])(
       babelTraverse,
     );
 
-    expect(prettier.resolveConfig.sync.mock.results[0].value).toEqual({parser});
+    expect(
+      (prettier.resolveConfig.sync as jest.Mock).mock.results[0].value,
+    ).toEqual({parser});
 
     expect(fs.writeFileSync).toHaveBeenCalledWith(
       filename,
@@ -108,10 +97,9 @@ test.each([['babylon'], ['flow'], ['typescript']])(
 
 test('saveInlineSnapshots() replaces existing template literal with property matchers', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => 'expect(1).toMatchInlineSnapshot({}, `2`);\n',
-  ): any);
+  );
 
   saveInlineSnapshots(
     [
@@ -132,10 +120,9 @@ test('saveInlineSnapshots() replaces existing template literal with property mat
 
 test('saveInlineSnapshots() throws if frame does not match', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => 'expect(1).toMatchInlineSnapshot();\n',
-  ): any);
+  );
 
   const save = () =>
     saveInlineSnapshots(
@@ -154,10 +141,9 @@ test('saveInlineSnapshots() throws if frame does not match', () => {
 
 test('saveInlineSnapshots() throws if multiple calls to to the same location', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => 'expect(1).toMatchInlineSnapshot();\n',
-  ): any);
+  );
 
   const frame = {column: 11, file: filename, line: 1};
   const save = () =>
@@ -174,10 +160,9 @@ test('saveInlineSnapshots() throws if multiple calls to to the same location', (
 
 test('saveInlineSnapshots() uses escaped backticks', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => 'expect("`").toMatchInlineSnapshot();\n',
-  ): any);
+  );
 
   const frame = {column: 13, file: filename, line: 1};
   saveInlineSnapshots([{frame, snapshot: '`'}], prettier, babelTraverse);
@@ -190,11 +175,10 @@ test('saveInlineSnapshots() uses escaped backticks', () => {
 
 test('saveInlineSnapshots() works with non-literals in expect call', () => {
   const filename = path.join(__dirname, 'my.test.js');
-  // $FlowFixMe mock
-  fs.readFileSync = (jest.fn(
+  (fs.readFileSync as jest.Mock).mockImplementation(
     () => `expect({a: 'a'}).toMatchInlineSnapshot();\n`,
-  ): any);
-  prettier.resolveConfig.sync.mockReturnValue({
+  );
+  (prettier.resolveConfig.sync as jest.Mock).mockReturnValue({
     bracketSpacing: false,
     singleQuote: true,
   });

packages/jest-snapshot/src/__tests__/matcher.test.js → packages/jest-snapshot/src/__tests__/matcher.test.ts

@@ -5,15 +5,19 @@
  * LICENSE file in the root directory of this source tree.
  *
  */
-'use strict';
 
-const {toMatchSnapshot} = require('../');
+const jestSnapshot = require('../');
+
+const {toMatchSnapshot} = jestSnapshot;
 
 it(`matcher returns matcher name, expected and actual values`, () => {
   const actual = 'a';
   const expected = 'b';
   const matcher = toMatchSnapshot.bind({
-    snapshotState: {match: (testName, received) => ({actual, expected})},
+    snapshotState: {
+      // @ts-ignore
+      match: (testName: string, received: any) => ({actual, expected}), // eslint-disable-line
+    },
   });
 
   const matcherResult = matcher({a: 1});

packages/jest-snapshot/src/__tests__/mock_serializer.test.js → packages/jest-snapshot/src/__tests__/mock_serializer.test.ts

@@ -5,7 +5,6 @@
  * LICENSE file in the root directory of this source tree.
  *
  */
-'use strict';
 
 import prettyFormat from 'pretty-format';
 

packages/jest-snapshot/src/__tests__/plugins.test.js → packages/jest-snapshot/src/__tests__/plugins.test.ts

@@ -5,11 +5,10 @@
  * LICENSE file in the root directory of this source tree.
  *
  */
-'use strict';
 
 beforeEach(() => jest.resetModules());
 
-const testPath = names => {
+const testPath = (names: string[]) => {
   const {addSerializer, getSerializers} = require('../plugins');
   const prev = getSerializers();
   const added = names.map(name =>