fixed restore not effect on http(s)

node-modules · Dec 1, 2012 · 2692bd5 · 2692bd5
1 parent 4828345
commit 2692bd5
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/lib/mm.js b/lib/mm.js
@@ -110,14 +110,6 @@ exports.empty = function (mod, method, timeout) {
   return exports.datas(mod, method, null, timeout);
 };
 
-/**
- * remove all mock effects.
- */
-exports.restore = function () {
-  muk.restore();
-  return this;
-};
-
 exports.http = {};
 exports.https = {};
 
@@ -312,4 +304,14 @@ function _requestError(mod, url, reqError, resError, delay) {
     return req;
   };
   return this;
-}
+}
+
+/**
+ * remove all mock effects.
+ */
+exports.restore = function () {
+  http.request = http.__sourceRequest;
+  https.request = https.__sourceRequest;
+  muk.restore();
+  return this;
+};