add more test cases

Author: TildaDares

lib/rules/forbid-prop-types.js

@@ -66,11 +66,11 @@ module.exports = {
     function isPropTypesPackage(node) {
       return (
         node.type === 'Identifier'
-        && node.name === propTypesPackageName
+        && (node.name === null || node.name === propTypesPackageName)
       ) || (
         node.type === 'MemberExpression'
         && node.property.name === 'PropTypes'
-        && node.object.name === reactPackageName
+        && (node.object.name === null || node.object.name === reactPackageName)
       );
     }
 
@@ -218,7 +218,6 @@ module.exports = {
           node.callee.object
           && !isPropTypesPackage(node.callee.object)
           && !propsUtil.isPropTypesDeclaration(node.callee)
-          && node.callee.object.name !== 'PropTypes'
         ) {
           return;
         }

tests/lib/rules/forbid-prop-types.js

@@ -654,6 +654,27 @@ ruleTester.run('forbid-prop-types', rule, {
           })
         )
       `,
+    },
+    {
+      code: `
+        import CustomPropTypes from "prop-types";
+        class Component extends React.Component {};
+        Component.propTypes = {
+          a: CustomPropTypes.shape({
+            b: CustomPropTypes.String,
+            c: CustomPropTypes.number.isRequired,
+          })
+        }
+      `,
+    },
+    {
+      code: `
+        import CustomReact from "react"
+        class Component extends React.Component {};
+        Component.propTypes = {
+          b: CustomReact.PropTypes.string,
+        }
+      `,
     }
   )),