905ef41cb0
requests and urllib will get upset of socks proxy settings are present in the environment. In those interceptors, an exception will now be raised if $http_proxy or $https_proxy are set. Tests are added to cover the variable being set for all four of the interceptor types. This is the quick and dirty solution to the problem. The way this has been done shows some clear opportunities for refactoring down the road. Fixes #30
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
import os
|
|
import py.test
|
|
from wsgi_intercept import requests_intercept, WSGIAppError
|
|
from test import wsgi_app
|
|
from test.install import installer_class
|
|
import requests
|
|
from requests.exceptions import ConnectionError
|
|
|
|
HOST = 'some_hopefully_nonexistant_domain'
|
|
|
|
InstalledApp = installer_class(requests_intercept)
|
|
|
|
|
|
def test_http():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80) as app:
|
|
resp = requests.get('http://some_hopefully_nonexistant_domain:80/')
|
|
assert resp.content == b'WSGI intercept successful!\n'
|
|
assert app.success()
|
|
|
|
|
|
def test_http_default_port():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80) as app:
|
|
resp = requests.get('http://some_hopefully_nonexistant_domain/')
|
|
assert resp.content == b'WSGI intercept successful!\n'
|
|
assert app.success()
|
|
|
|
|
|
def test_http_other_port():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=8080) as app:
|
|
resp = requests.get('http://some_hopefully_nonexistant_domain:8080/')
|
|
assert resp.content == b'WSGI intercept successful!\n'
|
|
assert app.success()
|
|
environ = app.get_internals()
|
|
assert environ['wsgi.url_scheme'] == 'http'
|
|
|
|
|
|
def test_bogus_domain():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80):
|
|
py.test.raises(
|
|
ConnectionError,
|
|
'requests.get("http://_nonexistant_domain_")')
|
|
|
|
|
|
def test_proxy_handling():
|
|
with py.test.raises(RuntimeError) as exc:
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80,
|
|
proxy='some_proxy.com:1234'):
|
|
requests.get('http://some_hopefully_nonexistant_domain:80/')
|
|
assert 'http_proxy or https_proxy set in environment' in str(exc.value)
|
|
# We need to do this by hand because the exception was raised
|
|
# during the entry of the context manager, so the exit handler
|
|
# wasn't reached.
|
|
del os.environ['http_proxy']
|
|
|
|
|
|
def test_https():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=443) as app:
|
|
resp = requests.get('https://some_hopefully_nonexistant_domain:443/')
|
|
assert resp.content == b'WSGI intercept successful!\n'
|
|
assert app.success()
|
|
|
|
|
|
def test_https_default_port():
|
|
with InstalledApp(wsgi_app.simple_app, host=HOST, port=443) as app:
|
|
resp = requests.get('https://some_hopefully_nonexistant_domain/')
|
|
assert resp.content == b'WSGI intercept successful!\n'
|
|
assert app.success()
|
|
environ = app.get_internals()
|
|
assert environ['wsgi.url_scheme'] == 'https'
|
|
|
|
|
|
def test_app_error():
|
|
with InstalledApp(wsgi_app.raises_app, host=HOST, port=80):
|
|
with py.test.raises(WSGIAppError):
|
|
requests.get('http://some_hopefully_nonexistant_domain/')
|
|
|
|
|
|
def test_http_not_intercepted():
|
|
with InstalledApp(wsgi_app.raises_app, host=HOST, port=80):
|
|
resp = requests.get("http://google.com")
|
|
assert resp.status_code >= 200 and resp.status_code < 300
|
|
|
|
|
|
def test_https_not_intercepted():
|
|
with InstalledApp(wsgi_app.raises_app, host=HOST, port=80):
|
|
resp = requests.get("https://google.com")
|
|
assert resp.status_code >= 200 and resp.status_code < 300
|