Add support for dot notation

#39
jhnns · Feb 6, 2015 · 4bc98b9 · 4bc98b9
1 parent 1e06fb0
commit 4bc98b9
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/__set__.js b/lib/__set__.js
@@ -24,7 +24,7 @@ function __set__() {
         for (arguments.varName in arguments.env) {
             if (arguments.env.hasOwnProperty(arguments.varName)) {
                 arguments.varValue = arguments.env[arguments.varName];
-                arguments.src += arguments.varName + " = arguments.env." + arguments.varName + "; ";
+                arguments.src += arguments.varName + " = arguments.env[" + JSON.stringify(arguments.varName) + "]; ";
                 arguments.revertArgs[0][arguments.varName] = eval(arguments.varName);
             }
         }

diff --git a/test/testModules/sharedTestCases.js b/test/testModules/sharedTestCases.js
@@ -284,6 +284,18 @@ describe("rewire " + (typeof testEnv === "undefined"? "(node)": "(" + testEnv +
         expect(rewiredModuleA.getMyObj()).to.eql({
             test: undefined
         });
+
+        revert = rewiredModuleA.__set__({
+            "myObj.test": true
+        });
+        expect(rewiredModuleA.getMyObj()).to.eql({
+            test: true
+        });
+        revert();
+        expect(rewiredModuleA.getMyObj()).to.eql({
+            test: undefined
+        });
+
     });
 
 });