Check for import declaration type when getting default values, resolves

#63 (#99)
reactjs · Aug 22, 2016 · b8681fb · b8681fb
1 parent 9325439
commit b8681fb
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/handlers/__tests__/defaultPropsHandler-test.js b/src/handlers/__tests__/defaultPropsHandler-test.js
@@ -86,6 +86,27 @@ describe('defaultPropsHandler', () => {
       `;
       test(parse(src).get('body', 0));
     });
+
+    it('should find prop default values that are imported variables', () => {
+      var src = `
+        import ImportedComponent from './ImportedComponent';
+
+        class Foo {
+          static defaultProps = {
+            foo: ImportedComponent,
+          };
+        }
+      `;
+      defaultPropsHandler(documentation, parse(src).get('body', 1));
+      expect(documentation.descriptors).toEqual({
+        foo: {
+          defaultValue: {
+            value: 'ImportedComponent',
+            computed: true,
+          },
+        },
+      });
+    });
   });
 
   describe('ClassExpression with static defaultProps', () => {

diff --git a/src/handlers/defaultPropsHandler.js b/src/handlers/defaultPropsHandler.js
@@ -27,8 +27,12 @@ function getDefaultValue(path) {
     defaultValue = node.raw;
   } else {
     path = resolveToValue(path);
-    node = path.node;
-    defaultValue = printValue(path);
+    if (types.ImportDeclaration.check(path.node)) {
+      defaultValue = node.name;
+    } else {
+      node = path.node;
+      defaultValue = printValue(path);
+    }
   }
   if (typeof defaultValue !== 'undefined') {
     return {