Usememo fixer

__tests__/require-memo.ts

@@ -1,5 +1,6 @@
 import { RuleTester } from "eslint";
 import rule from "../require-memo";
+import { normalizeIndent } from '../utils';
 
 const ruleTester = new RuleTester({
   parser: require.resolve("@typescript-eslint/parser"),
@@ -11,72 +12,130 @@ const ruleTester = new RuleTester({
 ruleTester.run("memo", rule, {
   valid: [
     {
-      code: `const Component = React.memo(() => <div />)`,
+      code: normalizeIndent`
+        const Component = memo(() => <div />)
+      `,
     },
     {
-      code: `const Component = memo(() => <div />)`,
+      code: normalizeIndent`
+        const Component = memo(() => <div />)
+      `,
     },
     {
-      code: `const Component = memo(useRef(() => <div />))`,
+      code: normalizeIndent`
+        const Component = memo(useRef(() => <div />))
+      `,
     },
     {
-      code: `const Component = React.useRef(React.memo(() => <div />))`,
+      code: normalizeIndent`
+        const Component = useRef(memo(() => <div />))
+      `,
     },
     {
-      code: `const myFunction = () => <div />`,
+      code: normalizeIndent`
+        const myFunction = () => <div />
+      `,
     },
     {
-      code: `const myFunction = wrapper(() => <div />)`,
+      code: normalizeIndent`
+        const myFunction = wrapper(() => <div />)
+      `,
     },
     {
-      code: `const Component = React.memo(function() { return <div />; });`,
+      code: normalizeIndent`
+        const Component = memo(function() { return <div />; });
+      `,
     },
     {
-      code: `const Component = memo(function Component() { return <div />; });`,
+      code: normalizeIndent`
+        const Component = memo(function Component() { return <div />; });
+      `,
     },
     {
-      code: `const myFunction = () => <div />`,
+      code: normalizeIndent`
+        const myFunction = () => <div />
+      `,
     },
     {
-      code: `const myFunction = wrapper(() => <div />)`,
+      code: normalizeIndent`
+        const myFunction = wrapper(() => <div />)
+      `,
     },
     {
-      code: `function myFunction() { return <div />; }`,
+      code: normalizeIndent`
+        function myFunction() { return <div />; }
+      `,
     },
     {
-      code: `const myFunction = wrapper(function() { return <div /> })`,
+      code: normalizeIndent`
+        const myFunction = wrapper(function() { return <div /> })
+      `,
     },
     {
       filename: "dir/myFunction.js",
       parserOptions: { ecmaVersion: 6, sourceType: "module" },
-      code: `export default function() { return <div /> };`,
+      code: normalizeIndent`
+        export default function() { return <div /> };
+      `,
     },
   ],
   invalid: [
     {
-      code: `const Component = () => <div />`,
+      code: `import { memo } from 'react'
+const Component = () => <div />`,
       errors: [{ messageId: "memo-required" }],
+      output: `import { memo } from 'react'
+const Component = memo(() => <div />)`
     },
     {
-      code: `const Component = useRef(() => <div />)`,
+      code: `// @flow
+      import { map } from 'lodash'
+const Component = useRef(() => <div />)`,
       errors: [{ messageId: "memo-required" }],
-    },
-    {
-      code: `const Component = function Component() { return <div />; }`,
-      errors: [{ messageId: "memo-required" }],
-    },
-    {
-      code: `const Component = useRef(function() { return <div />; })`,
+      output: `// @flow
+import { memo } from 'react'
+      import { map } from 'lodash'
+const Component = memo(useRef(() => <div />))`
+    },
+//     {
+//       code: `const Component = function Component() { return <div />; }`,
+//       errors: [{ messageId: "memo-required" }],
+//       output: `import { memo } from 'react'
+// const Component = memo(function Component() { return <div />; })`
+//     },
+//     {
+//       code: `const Component = useRef(function() { return <div />; })`,
+//       errors: [{ messageId: "memo-required" }],
+//       output: `import { memo } from 'react'
+// const Component = useRef(memo(function() { return <div />; }))`
+//     },
+    {
+      code: `// @flow
+import { map } from 'lodash'
+
+console.warn('Hello World')
+function Component() { return <div />; }`,
       errors: [{ messageId: "memo-required" }],
+      output: `// @flow
+import { memo } from 'react'
+import { map } from 'lodash'
+
+console.warn('Hello World')
+memo(function Component() { return <div />; })`
     },
     {
-      code: `function Component() { return <div />; }`,
+      code: `import { memo } from 'react'
+
+function Component() { return <div />; }`,
       errors: [{ messageId: "memo-required" }],
+      output: `import { memo } from 'react'
+
+memo(function Component() { return <div />; })`
     },
     // {
     //   filename: "dir/Component.js",
     //   parserOptions: { ecmaVersion: 6, sourceType: "module" },
-    //   code: `export default function() { return <div /> };`,
+    //   code: export default function() { return <div /> };,
     //   errors: [{ messageId: "memo-required" }],
     // },
   ],