mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 09:13:23 +00:00
41da9ddbab
Added support and tests for the Requests library. At least basic support seems to be working. The tests should run fine if you don't have requests installed; it just won't verify the requests- specific tests.
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import httplib
|
|
from contextlib import contextmanager
|
|
from .stubs import VCRHTTPConnection, VCRHTTPSConnection
|
|
|
|
|
|
_HTTPConnection = httplib.HTTPConnection
|
|
_HTTPSConnection = httplib.HTTPSConnection
|
|
|
|
try:
|
|
import requests.packages.urllib3.connectionpool
|
|
_VerifiedHTTPSConnection = requests.packages.urllib3.connectionpool.VerifiedHTTPSConnection
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def install(cassette_path):
|
|
httplib.HTTPConnection = httplib.HTTP._connection_class = VCRHTTPConnection
|
|
httplib.HTTPSConnection = httplib.HTTPS._connection_class = VCRHTTPSConnection
|
|
httplib.HTTPConnection._vcr_cassette_path = cassette_path
|
|
httplib.HTTPSConnection._vcr_cassette_path = cassette_path
|
|
try:
|
|
import requests.packages.urllib3.connectionpool
|
|
from .requests_stubs import VCRVerifiedHTTPSConnection
|
|
requests.packages.urllib3.connectionpool.VerifiedHTTPSConnection = VCRVerifiedHTTPSConnection
|
|
requests.packages.urllib3.connectionpool.VerifiedHTTPSConnection._vcr_cassette_path = cassette_path
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def reset():
|
|
httplib.HTTPConnection = httplib.HTTP._connection_class = _HTTPConnection
|
|
httplib.HTTPSConnection = httplib.HTTPS._connection_class = \
|
|
_HTTPSConnection
|
|
try:
|
|
import requests.packages.urllib3.connectionpool
|
|
requests.packages.urllib3.connectionpool.VerifiedHTTPSConnection = _VerifiedHTTPSConnection
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
@contextmanager
|
|
def use_cassette(cassette_path):
|
|
install(cassette_path)
|
|
yield
|
|
reset()
|