Add importer tests for isReactComponentClass

Author: phated

src/utils/__tests__/isReactComponentClass-test.js

@@ -6,37 +6,55 @@
  *
  */
 
-import { expression, statement, parse } from '../../../tests/utils';
+import {
+  expression,
+  statement,
+  parse,
+  noopImporter,
+  makeMockImporter,
+} from '../../../tests/utils';
 import isReactComponentClass from '../isReactComponentClass';
 
 describe('isReactComponentClass', () => {
+  const mockImporter = makeMockImporter({
+    component: statement(`
+      export default React.Component;
+      import React from 'react';
+    `).get('declaration'),
+
+    pureComponent: statement(`
+      export default React.PureComponent;
+      import React from 'react';
+    `).get('declaration'),
+  });
+
   describe('render method', () => {
     it('ignores class declarations with a render method without superclass', () => {
       const def = statement('class Foo { render() {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('ignores class expression with a render method without superclass', () => {
       const def = expression('class { render() {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('ignores static render methods', () => {
       const def = statement('class Foo { static render() {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('ignores dynamic render methods', () => {
       const def = statement('class Foo { static [render]() {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('ignores getter or setter render methods', () => {
       let def = statement('class Foo { get render() {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
 
       def = statement('class Foo { set render(value) {}}');
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
   });
 
@@ -51,7 +69,7 @@ describe('isReactComponentClass', () => {
         class Foo extends Bar {}
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
   });
 
@@ -62,7 +80,7 @@ describe('isReactComponentClass', () => {
         class Foo extends React.Component {}
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('accepts class expressions extending React.Component', () => {
@@ -71,7 +89,7 @@ describe('isReactComponentClass', () => {
         var Foo = class extends React.Component {}
       `).get('body', 1, 'declarations', 0, 'init');
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('resolves the super class reference', () => {
@@ -81,7 +99,7 @@ describe('isReactComponentClass', () => {
         class Foo extends C {}
       `).get('body', 2);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('does not accept references to other modules', () => {
@@ -90,7 +108,7 @@ describe('isReactComponentClass', () => {
         class Foo extends Component {}
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('does not consider super class if render method is present', () => {
@@ -99,7 +117,16 @@ describe('isReactComponentClass', () => {
         class Foo extends Component { render() {} }
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
+    });
+
+    it('can resolve Component from an intermediate module', () => {
+      const def = parse(`
+        import RC from 'component';
+        class Foo extends RC {}
+      `).get('body', 1);
+
+      expect(isReactComponentClass(def, mockImporter)).toBe(true);
     });
   });
 
@@ -110,7 +137,7 @@ describe('isReactComponentClass', () => {
         class Foo extends React.PureComponent {}
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('accepts class expressions extending React.PureComponent', () => {
@@ -119,7 +146,7 @@ describe('isReactComponentClass', () => {
         var Foo = class extends React.PureComponent {}
       `).get('body', 1, 'declarations', 0, 'init');
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('resolves the super class reference', () => {
@@ -129,7 +156,7 @@ describe('isReactComponentClass', () => {
         class Foo extends C {}
       `).get('body', 2);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
     });
 
     it('does not accept references to other modules', () => {
@@ -138,7 +165,7 @@ describe('isReactComponentClass', () => {
         class Foo extends PureComponent {}
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(false);
+      expect(isReactComponentClass(def, noopImporter)).toBe(false);
     });
 
     it('does not consider super class if render method is present', () => {
@@ -147,7 +174,16 @@ describe('isReactComponentClass', () => {
         class Foo extends PureComponent { render() {} }
       `).get('body', 1);
 
-      expect(isReactComponentClass(def)).toBe(true);
+      expect(isReactComponentClass(def, noopImporter)).toBe(true);
+    });
+
+    it('can resolve PureComponent from an intermediate module', () => {
+      const def = parse(`
+        import PC from 'pureComponent';
+        class Foo extends PC {}
+      `).get('body', 1);
+
+      expect(isReactComponentClass(def, mockImporter)).toBe(true);
     });
   });
 });