Merge pull request #17 from asantos00/fix-nested-shape-objects

Fix bug with nested shape objects

Author: typicode

__tests__/__snapshots__/propTypes.js.snap

@@ -50,3 +50,15 @@ Object {
   "requiredFunc": [Function],
 }
 `;
+
+exports[`fakeProps should work with nested shape objects 1`] = `
+Object {
+  "requiredShape": Object {
+    "nestedRequiredShape": Object {
+      "optionalString": "requiredShape.nestedRequiredShape.optionalString",
+    },
+    "optionalString": "requiredShape.optionalString",
+  },
+  "requiredString": "requiredString",
+}
+`;

__tests__/propTypes.js

@@ -8,6 +8,7 @@ const ComponentFile = path.join(
 const SimpleFile = path.join(__dirname, '../fixtures/propTypes/Simple.jsx')
 const NoProps = require.resolve('../fixtures/propTypes/NoProps.jsx')
 const Multiple = require.resolve('../fixtures/propTypes/Multiple.jsx')
+const NestedShape = require.resolve('../fixtures/propTypes/NestedShape.jsx')
 
 describe('fakeProps', () => {
   it('should return an object with no props (snapshot)', () => {
@@ -22,6 +23,10 @@ describe('fakeProps', () => {
     expect(fakeProps(ComponentFile, { optional: true })).toMatchSnapshot()
   })
 
+  it('should work with nested shape objects', () => {
+    expect(fakeProps(NestedShape, { optional: true })).toMatchSnapshot()
+  })
+
   it('should return more props with optional', () => {
     const allProps = fakeProps(ComponentFile, { optional: true })
     const requiredProps = fakeProps(ComponentFile)

fixtures/propTypes/NestedShape.jsx

@@ -0,0 +1,17 @@
+import PropTypes from 'prop-types'
+
+function NestedShape () {
+  return <div />
+}
+
+NestedShape.propTypes = {
+  requiredString: PropTypes.string.isRequired,
+  requiredShape: PropTypes.shape({
+    optionalString: PropTypes.string,
+    nestedRequiredShape: PropTypes.shape({
+      optionalString: PropTypes.string
+    })
+  }).isRequired
+}
+
+export default NestedShape

src/index.js

@@ -29,7 +29,7 @@ function getShape (prefix, object, opts) {
   Object.keys(object).forEach(key => {
     const type = object[key]
     if (type.required || opts.optional) {
-      res[key] = getFakeProp(`${prefix}.${key}`, { type })
+      res[key] = getFakeProp(`${prefix}.${key}`, { type }, opts)
     }
   })