[fix]prop-types: fix case with destructuring and defualt param

fixes jsx-eslint#2241
golopot · Apr 20, 2019 · aba172b · aba172b
1 parent f0c0b4d
commit aba172b
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 6 deletions.
diff --git a/lib/util/usedPropTypes.js b/lib/util/usedPropTypes.js
@@ -316,16 +316,17 @@ module.exports = function usedPropTypesInstructions(context, components, utils)
         break;
       case 'ArrowFunctionExpression':
       case 'FunctionDeclaration':
-      case 'FunctionExpression':
+      case 'FunctionExpression': {
         if (node.params.length === 0) {
           break;
         }
         type = 'destructuring';
-        properties = node.params[0].properties;
-        if (inSetStateUpdater()) {
-          properties = node.params[1].properties;
-        }
+        const propParam = inSetStateUpdater() ? node.params[1] : node.params[0];
+        properties = propParam.type === 'AssignmentPattern'
+          ? propParam.left.properties
+          : propParam.properties;
         break;
+      }
       case 'VariableDeclarator':
         for (let i = 0, j = node.id.properties.length; i < j; i++) {
           // let {props: {firstname}} = this
@@ -421,7 +422,15 @@ module.exports = function usedPropTypesInstructions(context, components, utils)
    *   FunctionDeclaration, or FunctionExpression
    */
   function markDestructuredFunctionArgumentsAsUsed(node) {
-    const destructuring = node.params && node.params[0] && node.params[0].type === 'ObjectPattern';
+    const param = inSetStateUpdater()
+      ? node.params && node.params[1]
+      : node.params && node.params[0];
+
+    const destructuring = param && (
+      param.type === 'ObjectPattern' ||
+      param.type === 'AssignmentPattern' && param.left.type === 'ObjectPattern'
+    );
+
     if (destructuring && (components.get(node) || components.get(node.parent))) {
       markPropTypesAsUsed(node);
     }

diff --git a/tests/lib/rules/prop-types.js b/tests/lib/rules/prop-types.js
@@ -3391,6 +3391,16 @@ ruleTester.run('prop-types', rule, {
       errors: [
         {message: '\'bar\' is missing in props validation'}
       ]
+    }, {
+      code: [
+        'function SomeComponent({bar} = baz) {',
+        '  function f({foo}) {}',
+        '  return <div className={f()}>{bar}</div>;',
+        '}'
+      ].join('\n'),
+      errors: [
+        {message: '\'bar\' is missing in props validation'}
+      ]
     }, {
       code: [
         'class Hello extends React.PureComponent {',
@@ -4224,6 +4234,21 @@ ruleTester.run('prop-types', rule, {
         message: '\'bar\' is missing in props validation'
       }]
     },
+    {
+      code: `
+        class Foo extends React.Component {
+          setZoo() {
+            this.setState((state, {zoo}) => ({ zoo }));
+          }
+          render() {
+            return <div />;
+          }
+        }
+      `,
+      errors: [{
+        message: '\'zoo\' is missing in props validation'
+      }]
+    },
     {
       code: `
         class Foo extends React.Component {