mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''Test using a proxy.'''
|
|
|
|
# External imports
|
|
import multiprocessing
|
|
import pytest
|
|
|
|
from six.moves import socketserver, SimpleHTTPServer
|
|
from six.moves.urllib.request import urlopen
|
|
|
|
# Internal imports
|
|
import vcr
|
|
|
|
# Conditional imports
|
|
requests = pytest.importorskip("requests")
|
|
|
|
|
|
class Proxy(SimpleHTTPServer.SimpleHTTPRequestHandler):
|
|
'''
|
|
Simple proxy server.
|
|
|
|
(Inspired by: http://effbot.org/librarybook/simplehttpserver.htm).
|
|
'''
|
|
def do_GET(self):
|
|
upstream_response = urlopen(self.path)
|
|
try:
|
|
status = upstream_response.status
|
|
headers = upstream_response.headers.items()
|
|
except AttributeError:
|
|
# In Python 2 the response is an addinfourl instance.
|
|
status = upstream_response.code
|
|
headers = upstream_response.info().items()
|
|
self.send_response(status, upstream_response.msg)
|
|
for header in headers:
|
|
self.send_header(*header)
|
|
self.end_headers()
|
|
self.copyfile(upstream_response, self.wfile)
|
|
|
|
|
|
@pytest.yield_fixture(scope='session')
|
|
def proxy_server():
|
|
httpd = socketserver.ThreadingTCPServer(('', 0), Proxy)
|
|
proxy_process = multiprocessing.Process(
|
|
target=httpd.serve_forever,
|
|
)
|
|
proxy_process.start()
|
|
yield 'http://{0}:{1}'.format(*httpd.server_address)
|
|
proxy_process.terminate()
|
|
|
|
|
|
def test_use_proxy(tmpdir, httpbin, proxy_server):
|
|
'''Ensure that it works with a proxy.'''
|
|
with vcr.use_cassette(str(tmpdir.join('proxy.yaml'))):
|
|
response = requests.get(httpbin.url, proxies={'http': proxy_server})
|
|
|
|
with vcr.use_cassette(str(tmpdir.join('proxy.yaml'))) as cassette:
|
|
cassette_response = requests.get(httpbin.url, proxies={'http': proxy_server})
|
|
|
|
assert cassette_response.headers == response.headers
|
|
assert cassette.play_count == 1
|