Add restore method for stubbed property descriptors

lucasfcosta · lucasfcosta · commit e852f4fdf537 · 2017-03-11T13:33:15.000-03:00
diff --git a/lib/sinon/stub.js b/lib/sinon/stub.js
@@ -44,12 +44,11 @@ function stub(object, property, descriptor) {
     var s = stub.create(arity);
     s.rootObj = object;
     s.propName = property;
+    s.restore = function restore() {
+        Object.defineProperty(object, property, actualDescriptor);
+    };
 
-    if (isStubbingNonFuncProperty) {
-        return s;
-    }
-
-    return wrapMethod(object, property, s);
+    return isStubbingNonFuncProperty ? s : wrapMethod(object, property, s);
 }
 
 stub.createStubInstance = function (constructor) {
diff --git a/test/stub-test.js b/test/stub-test.js
@@ -2332,6 +2332,44 @@ describe("stub", function () {
 
             assert.equals(myObj.prop, "bar");
         });
+
+        it("can restore stubbed setters for functions", function () {
+            var propFn = function () {
+                return "bar";
+            };
+
+            var myObj = {
+                prop: propFn
+            };
+
+            var stub = createStub(myObj, "prop");
+
+            stub.get(function () {
+                return "baz";
+            });
+
+            stub.restore();
+
+            assert.equals(myObj.prop, propFn);
+        });
+
+        it("can restore stubbed getters for properties", function () {
+            var myObj = {
+                get prop() {
+                    return "bar";
+                }
+            };
+
+            var stub = createStub(myObj, "prop");
+
+            stub.get(function () {
+                return "baz";
+            });
+
+            stub.restore();
+
+            assert.equals(myObj.prop, "bar");
+        });
     });
 
     describe(".set", function () {
@@ -2392,5 +2430,45 @@ describe("stub", function () {
 
             assert.equals(myObj.example, "bar");
         });
+
+        it("can restore stubbed setters for functions", function () {
+            var propFn = function () {
+                return "bar";
+            };
+
+            var myObj = {
+                prop: propFn
+            };
+
+            var stub = createStub(myObj, "prop");
+
+            stub.set(function () {
+                myObj.otherProp = "baz";
+            });
+
+            stub.restore();
+
+            assert.equals(myObj.prop, propFn);
+        });
+
+        it("can restore stubbed setters for properties", function () {
+            var myObj = { // eslint-disable-line accessor-pairs
+                set prop(val) {
+                    this.otherProp = "bar";
+                    return "bar";
+                }
+            };
+
+            var stub = createStub(myObj, "prop");
+
+            stub.set(function () {
+                myObj.otherProp = "baz";
+            });
+
+            stub.restore();
+
+            myObj.prop = "foo";
+            assert.equals(myObj.otherProp, "bar");
+        });
     });
 });
