Merge pull request jestjs#1 from jamedranoa/bug/mocking-babel6

add special slot '__esModule'
mthadley · Jan 22, 2016 · 4456d43 · 4456d43
2 parents b9b4bb0 + f70e60e
commit 4456d43
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/lib/__tests__/moduleMocker-test.js b/src/lib/__tests__/moduleMocker-test.js
@@ -75,6 +75,21 @@ describe('moduleMocker', function() {
       expect(fooMock.nonEnumGetter).toBeUndefined();
     });
 
+    it('does not mock __esModule special property (babel6)', function () {
+      var ClassFoo = function() {};
+
+      var fooModule = {
+        __esModule: true,
+        default: ClassFoo
+      };
+
+      var fooModuleMock = moduleMocker.generateFromMetadata(
+        moduleMocker.getMetadata(fooModule)
+      );
+
+      expect(fooModuleMock.__esModule).toEqual(fooModule.__esModule);
+    });
+
     it('mocks ES2015 non-enumerable methods', function() {
       class ClassFoo {
         foo() {}

diff --git a/src/lib/moduleMocker.js b/src/lib/moduleMocker.js
@@ -287,6 +287,12 @@ function generateMock(metadata, callbacks, refs) {
 
   getSlots(metadata.members).forEach(slot => {
     const slotMetadata = metadata.members[slot];
+    // special slot used by babel 6
+    if (slot === '__esModule') {
+      mock[slot] =  slotMetadata;
+      return;
+    }
+
     if (slotMetadata.ref != null) {
       callbacks.push(() => mock[slot] = refs[slotMetadata.ref]);
     } else {
@@ -349,6 +355,13 @@ function getMetadata(component, _refs) {
   if (type !== 'array') {
     if (type !== 'undefined') {
       getSlots(component).forEach(slot => {
+        // special slot used by babel 6
+        if (slot === '__esModule')  {
+          members = members || {};
+          members[slot] = component[slot];
+          return;
+        }
+
         if (
           slot.charAt(0) === '_' ||
           (