Monkeypatch requests.request

Before we were only monkeypatching `requests.get`.
We should also monkeypatch `requests.request`
(which is what httpie is using)
This commit is contained in:
Marc Abramowitz
2015-02-04 09:21:05 -08:00
parent 39f4c19546
commit e6e8040e8b

View File

@@ -20,6 +20,8 @@ class monkeypatch(object):
requests = self._get_global_requests_module()
self.orig_requests_get = requests.get
requests.get = self.session.get
self.orig_requests_request = requests.request
requests.request = self.session.request
def _get_global_requests_module(self):
return sys.modules['requests']
@@ -30,3 +32,4 @@ class monkeypatch(object):
def __exit__(self, *args):
requests = self._get_global_requests_module()
requests.get = self.orig_requests_get
requests.request = self.orig_requests_request