1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
|
import pytest
from wsgi_intercept import http_client_intercept, WSGIAppError
from . import wsgi_app
from .install import installer_class, skipnetwork
import http.client as http_lib
HOST = 'some_hopefully_nonexistant_domain'
InstalledApp = installer_class(http_client_intercept)
def test_http():
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80) as app:
http_client = http_lib.HTTPConnection(HOST)
http_client.request('GET', '/')
content = http_client.getresponse().read()
http_client.close()
assert content == b'WSGI intercept successful!\n'
assert app.success()
def test_https():
with InstalledApp(wsgi_app.simple_app, host=HOST, port=443) as app:
http_client = http_lib.HTTPSConnection(HOST)
http_client.request('GET', '/')
content = http_client.getresponse().read()
http_client.close()
assert content == b'WSGI intercept successful!\n'
assert app.success()
def test_other():
with InstalledApp(wsgi_app.simple_app, host=HOST, port=8080) as app:
http_client = http_lib.HTTPConnection(HOST + ':8080')
http_client.request('GET', '/')
content = http_client.getresponse().read()
http_client.close()
assert content == b'WSGI intercept successful!\n'
assert app.success()
def test_proxy_handling():
"""Proxy variable no impact."""
with InstalledApp(wsgi_app.simple_app, host=HOST, port=80,
proxy='some.host:1234') as app:
http_client = http_lib.HTTPConnection(HOST)
http_client.request('GET', '/')
content = http_client.getresponse().read()
http_client.close()
assert content == b'WSGI intercept successful!\n'
assert app.success()
def test_app_error():
with InstalledApp(wsgi_app.raises_app, host=HOST, port=80):
http_client = http_lib.HTTPConnection(HOST)
with pytest.raises(WSGIAppError):
http_client.request('GET', '/')
http_client.getresponse().read()
http_client.close()
@skipnetwork
def test_http_not_intercepted():
with InstalledApp(wsgi_app.raises_app, host=HOST, port=80):
http_client = http_lib.HTTPConnection('google.com')
http_client.request('GET', '/')
response = http_client.getresponse()
http_client.close()
assert 200 <= int(response.status) < 400
@skipnetwork
def test_https_not_intercepted():
with InstalledApp(wsgi_app.raises_app, host=HOST, port=443):
http_client = http_lib.HTTPSConnection('google.com')
http_client.request('GET', '/')
response = http_client.getresponse()
http_client.close()
assert 200 <= int(response.status) < 400
|