Support jest snapshots

Author: mohsen1

test/collapse-intersection-interfaces-transform/multiple/output.tsx

@@ -1,8 +1,8 @@
 type Foo = {
-    foo: string;
-    bar: number;
+foo: string,
+bar: number
 };
 type Bar = {
-    foo: number;
-    bar: string;
+foo: number,
+bar: string
 };

test/end-to-end/multiple-components/output.tsx

@@ -1,33 +1,34 @@
 type HelloProps = {
-    message?: string;
+    message?: string
 };
 const Hello: React.SFC<HelloProps> = ({ message }) => {
     return <div>hello {message}</div>;
 };
 type HeyProps = {
-    message?: string;
+    message?: string
 };
 const Hey: React.SFC<HeyProps> = ({ name }) => {
     return <div>hey, {name}</div>;
 };
 type MyComponentState = {
-    foo: number;
-    bar: number;
+    foo: number,
+    bar: number
 };
-export default class MyComponent extends React.Component<{
-    }, MyComponentState> {
+export default class MyComponent extends React.Component<{}, MyComponentState> {
     render() {
-        return <button onClick={this.onclick.bind(this)}/>;
+        return <button onClick={this.onclick.bind(this)} />;
     }
     onclick() {
         this.setState({ foo: 1, bar: 2 });
     }
 }
 type AnotherComponentProps = {
-    foo: string;
+    foo: string
 };
-export class AnotherComponent extends React.Component<AnotherComponentProps, {
-    }> {
+export class AnotherComponent extends React.Component<
+    AnotherComponentProps,
+    {}
+> {
     render() {
         return <div />;
     }

test/react-js-make-props-and-state-transform/multiple-components/output.tsx

@@ -1,21 +1,22 @@
 import * as React from 'react';
-type MyComponentState = { foo: number; bar: number; };
-export default class MyComponent extends React.Component<{
-    }, MyComponentState> {
+type MyComponentState = { foo: number, bar: number };
+export default class MyComponent extends React.Component<{}, MyComponentState> {
     render() {
-        return <button onClick={this.onclick.bind(this)}/>;
+        return <button onClick={this.onclick.bind(this)} />;
     }
     onclick() {
         this.setState({ foo: 1, bar: 2 });
     }
 }
 type AnotherComponentProps = {
-    foo: string;
+    foo: string
 };
-export class AnotherComponent extends React.Component<AnotherComponentProps, {
-    }> {
+export class AnotherComponent extends React.Component<
+    AnotherComponentProps,
+    {}
+> {
     static propTypes = {
-        foo: React.PropTypes.string.isRequired,
+        foo: React.PropTypes.string.isRequired
     };
     render() {
         return <div />;

test/react-move-prop-types-to-class-transform/multiple-components/output.tsx

@@ -1,17 +1,20 @@
-class SomeComponent extends React.Component<{
-        foo: number;
-    }, {
-        bar: string;
-    }> {
+class SomeComponent extends React.Component<
+    {
+        foo: number
+    },
+    {
+        bar: string
+    }
+> {
     static propTypes = { foo: React.PropTypes.string };
     render() {
         return null;
     }
 }
 SomeComponent.propTypes = { foo: React.PropTypes.string };
 class AnotherComponent extends React.Component<{
-        baz: number;
-    }> {
+    baz: number
+}> {
     static propTypes = { baz: React.PropTypes.string };
     render() {
         return null;

test/react-remove-prop-types-assignment-transform/multiple/output.tsx

@@ -1,11 +1,12 @@
-class SomeComponent extends React.Component<{
-        foo: string;
-        baz: string;
-    }, {
-        bar: string;
-    }> {
-}
+class SomeComponent extends React.Component<
+    {
+        foo: string,
+        baz: string
+    },
+    {
+        bar: string
+    }
+> {}
 class AnotherComponent extends React.Component<{
-        lol: number;
-    }> {
-}
+    lol: number
+}> {}

test/react-remove-prop-types-import/from-prop-types/output.tsx

@@ -1,5 +1,4 @@
-import React from 'react';
-
+import React from 'react'
 export const Hello = ({ message }) => {
-    return <div>hello {message}</div>;
-};
+  return <div>hello {message}</div>
+}

test/react-remove-prop-types-import/from-react-multi-named-import/output.tsx

@@ -1,7 +1,6 @@
 import React, { Component } from 'react';
-
 export class MyComponent extends Component {
-    render() {
-        return <div>hello</div>;
-    }
+  render() {
+    return <div>hello</div>;
+  }
 }

test/react-remove-prop-types-import/from-react-simple/output.tsx

@@ -1,5 +1,4 @@
-import React from 'react';
-
+import React from 'react'
 export const Hello = ({ message }) => {
-    return <div>hello {message}</div>;
-};
+  return <div>hello {message}</div>
+}

test/react-remove-static-prop-types-member-transform/multiple-components/output.tsx

@@ -1,12 +1,16 @@
-class SomeComponent extends React.Component<{
-        foo: number;
-    }, {
-        bar: string;
-    }> {
-}
-class AnotherComponent extends React.Component<{
-        foo: number;
-    }, {
-        bar: string;
-    }> {
-}
+class SomeComponent extends React.Component<
+    {
+        foo: number
+    },
+    {
+        bar: string
+    }
+> {}
+class AnotherComponent extends React.Component<
+    {
+        foo: number
+    },
+    {
+        bar: string
+    }
+> {}

test/react-stateless-function-make-props-transform/multiple-components/output.tsx

@@ -1,18 +1,18 @@
 type HelloProps = {
-    message?: string;
+  message?: string
 };
 const Hello: React.SFC<HelloProps> = ({ message }) => {
-    return <div>hello {message}</div>;
+  return <div>hello {message}</div>;
 };
 type HeyProps = {
-    name: string;
+  name: string
 };
 const Hey: React.SFC<HeyProps> = ({ name }) => {
-    return <div>hey, {name}</div>;
+  return <div>hey, {name}</div>;
 };
 Hey.propTypes = {
-    name: React.PropTypes.string.isRequired,
+  name: React.PropTypes.string.isRequired
 };
 Hello.propTypes = {
-    message: React.PropTypes.string,
+  message: React.PropTypes.string
 };

test/transformers.test.ts

@@ -7,6 +7,8 @@ import * as path from 'path';
 import * as fs from 'fs';
 import * as ts from 'typescript';
 import * as chalk from 'chalk';
+import * as _ from 'lodash';
+
 import {
     reactJSMakePropsAndStateInterfaceTransformFactoryFactory,
     reactStatelessFunctionMakePropsTransformFactoryFactory,
@@ -33,15 +35,21 @@ const transformToFolderMap: [string, TransformFactoryFactory[]][] = [
     ['end-to-end', allTransforms],
 ];
 
+const isJestUpdateSnapshotEnabled = !!_.intersection(process.argv, ['-u', '--updateSnapshot']).length;
+
 for (const [testFolderName, getFactory] of transformToFolderMap) {
     describe(testFolderName.replace(/\-/g, ' ').replace('transform', ''), () => {
         for (const folderName of fs.readdirSync(path.join(__dirname, testFolderName))) {
             const folder = path.join(path.join(__dirname, testFolderName, folderName));
             if (fs.statSync(folder).isDirectory()) {
                 it(`${testFolderName} ${folderName}`, async () => {
                     const inputPath = path.join(folder, 'input.tsx');
-                    const output = await readFile(path.join(folder, 'output.tsx'));
+                    const outputPath = path.join(folder, 'output.tsx');
                     const result = compile(inputPath, getFactory);
+                    if (isJestUpdateSnapshotEnabled) {
+                        await writeFile(outputPath, result);
+                    }
+                    const output = await readFile(outputPath);
                     expect(stripEmptyLines(result)).toEqual(stripEmptyLines(output));
                 });
             }
@@ -70,3 +78,17 @@ function readFile(pathToFile: string) {
         });
     });
 }
+
+/**
+ * Read a file
+ * @param pathToFile Path to a string file
+ * @param contents Contents of the file
+ */
+function writeFile(pathToFile: string, contents: string) {
+    return new Promise<string>((resolve, reject) => {
+        fs.writeFile(pathToFile, contents, (error) => {
+            if (error) { return reject(error); }
+            resolve();
+        });
+    });
+}