Should not blow up on modules that export primitives

jhnns · Nov 7, 2015 · a23e60e · a23e60e
1 parent ac0c8a3
commit a23e60e
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/lib/getDefinePropertySrc.js b/lib/getDefinePropertySrc.js
@@ -11,9 +11,10 @@ var srcs = {
 };
 
 function getDefinePropertySrc() {
-    var src;
+    var src = "if (typeof(module.exports) === 'function' || \n" +
+            "typeof(module.exports) === 'object') {\n";
 
-    src = Object.keys(srcs).reduce(function forEachSrc(preValue, value) {
+    src += Object.keys(srcs).reduce(function forEachSrc(preValue, value) {
         return preValue += "Object.defineProperty(module.exports, '" +
             value +
             "', {enumerable: false, value: " +
@@ -22,6 +23,8 @@ function getDefinePropertySrc() {
             "writable: true}); ";
     }, "");
 
+    src += "\n}";
+
     return src;
 }
 

diff --git a/test/testModules/boolean.js b/test/testModules/boolean.js
@@ -0,0 +1 @@
+module.exports = true;
diff --git a/test/testModules/sealedObject.js b/test/testModules/sealedObject.js
@@ -0,0 +1 @@
+module.exports = {};
diff --git a/test/testModules/sharedTestCases.js b/test/testModules/sharedTestCases.js
@@ -220,6 +220,18 @@ describe("rewire " + (typeof testEnv === "undefined"? "(node)": "(" + testEnv +
         expect(rewired.__get__("someVar")).to.be("hello");
     });
 
+    it("should not be a problem to have a module that exports a primitive", function() {
+        expect(function() {
+            var rewired = rewire("./boolean.js");
+        }).to.not.throwException();
+    });
+
+    it("should not be a problem to have a module that exports a sealed object", function() {
+        expect(function() {
+            var rewired = rewire("./sealedObject.js");
+        }).to.not.throwException();
+    });
+
     it("should not influence the original require if nothing has been required within the rewired module", function () {
         rewire("./emptyModule.js"); // nothing happens here because emptyModule doesn't require anything
         expect(require("./moduleA.js").__set__).to.be(undefined); // if restoring the original node require didn't worked, the module would have a setter