Add support for dot notation

jhnns · jhnns · commit 4bc98b98549e · 2015-02-06T18:06:04.000+01:00
jhnns#39
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
+        });
+
     });
 
 });

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@ function __set__() {`
`24`	`24`	`for (arguments.varName in arguments.env) {`
`25`	`25`	`if (arguments.env.hasOwnProperty(arguments.varName)) {`
`26`	`26`	`arguments.varValue = arguments.env[arguments.varName];`
`27`		`- arguments.src += arguments.varName + " = arguments.env." + arguments.varName + "; ";`
	`27`	`+ arguments.src += arguments.varName + " = arguments.env[" + JSON.stringify(arguments.varName) + "]; ";`
`28`	`28`	`arguments.revertArgs[0][arguments.varName] = eval(arguments.varName);`
`29`	`29`	`}`
`30`	`30`	`}`