def test_get_host_nonstandard_https_port(hostchecker):
request = requestMock(b"/foo",
host=b'example.com',
port=8443,
isSecure=True)
request.requestHeaders.removeHeader(
b'host') # force hostchecker to use transport attributes
assert hostchecker.get_host(request) == 'example.com:8443'
def test_get_host_forwarded_multiple(hostchecker):
request = requestMock(
b"/foo", headers={'X-Forwarded-Host': ['example.net, example.com']})
assert hostchecker.get_host(request) == 'example.net'
def test_get_host_header(hostchecker):
# klein's requestMock calls Twisted's Request.setHost, which sets the Host header as well as the
# transport, so we need to override the header value after initialization for this test to work
request = requestMock(b"/foo")
request.requestHeaders.setRawHeaders('Host', ['example.com'])
assert hostchecker.get_host(request) == 'example.com'
def test_get_host_default_http_port(hostchecker):
request = requestMock(b"/foo", host=b'example.com', port=80)
request.requestHeaders.removeHeader(
b'host') # force hostchecker to use transport attributes
assert hostchecker.get_host(request) == 'example.com'
