Skip to content

Commit

Permalink
override http.get
Browse files Browse the repository at this point in the history
  • Loading branch information
Martin Kuba committed Jun 4, 2017
1 parent c2d605a commit 119fa65
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions lib/common.js
Expand Up @@ -120,7 +120,8 @@ var overrideRequests = function(newRequest) {
http: require('http'),
https: require('https')
}[moduleName],
overriddenRequest = module.request;
overriddenRequest = module.request,
overriddenGet = module.get;

if(requestOverride[moduleName]) {
throw new Error('Module\'s request already overridden for ' + moduleName + ' protocol.');
Expand All @@ -129,14 +130,21 @@ var overrideRequests = function(newRequest) {
// Store the properties of the overridden request so that it can be restored later on.
requestOverride[moduleName] = {
module: module,
request: overriddenRequest
request: overriddenRequest,
get: overriddenGet
};

module.request = function(options, callback) {
// debug('request options:', options);
return newRequest(proto, overriddenRequest.bind(module), options, callback);
};

module.get = function(options, callback) {
var req = newRequest(proto, overriddenRequest.bind(module), options, callback);
req.end();
return req;
}

debug('- overridden request for', proto);
});
};
Expand All @@ -155,6 +163,7 @@ var restoreOverriddenRequests = function() {
var override = requestOverride[proto];
if(override) {
override.module.request = override.request;
override.module.get = override.get;
debug('- restored request for', proto);
}
});
Expand Down

0 comments on commit 119fa65

Please sign in to comment.