def test_extract_ip_when_second_header_with_different_setting(self):
headers = {'Cf-Connecting-Ip': '1.2.3.4', 'X-Forwarded-For': '1.1.1.1, 1.2.2.2, 1.2.3.5'}
configuration.ip_headers = ["CF-CONNECTING-IP", "X-Forwarded-For"]
self.assertEqual(
ExtractorsIp(headers).call(),
'1.2.3.4'
)
def test_extract_ip_for_spoof_ip_attempt_when_all_trusted_proxies(self):
headers = {'Client-Ip': '6.6.6.6', 'X-Forwarded-For': '6.6.6.6, 2.2.2.3, 192.168.0.7'}
configuration.trusted_proxies = [r'^2.2.2.\d$']
self.assertEqual(
ExtractorsIp(headers).call(),
'6.6.6.6'
)
def test_extract_ip_when_second_header(self):
headers = {'Cf-Connecting-Ip': '1.2.3.4', 'X-Forwarded-For': '1.1.1.1, 1.2.2.2, 1.2.3.5'}
configuration.ip_headers = ["HTTP_CF_CONNECTING_IP", "X-Forwarded-For"]
self.assertEqual(
ExtractorsIp(headers).call(),
'1.2.3.4'
)
def test_extract_ip_when_all_trusted_proxies(self):
xf_header = """
127.0.0.1,10.0.0.1,172.31.0.1,192.168.0.1,::1,fd00::,localhost,unix,unix:/tmp/sock
"""
headers = {'Remote-Addr': '127.0.0.1', 'X-Forwarded-For': xf_header}
self.assertEqual(
ExtractorsIp(headers).call(),
'127.0.0.1'
)
def test_extract_ip_when_trust_proxy_depth(self):
xf_header = """
6.6.6.6,2.2.2.3,6.6.6.5
"""
headers = {'Remote-Addr': '6.6.6.4', 'X-Forwarded-For': xf_header}
configuration.trusted_proxy_depth = 1
self.assertEqual(
ExtractorsIp(headers).call(),
'2.2.2.3'
)
def test_extract_ip_from_wsgi_request_x_forwarded_for(self):
self.assertEqual(
ExtractorsIp(request_without_ip_x_forwarded_for()).call(),
request_ip()
)
def test_extract_ip_from_wsgi_request_remote_addr(self):
self.assertEqual(
ExtractorsIp(request_without_ip_remote_addr()).call(),
request_ip()
)
def test_extract_ip(self):
self.assertEqual(ExtractorsIp(request()).call(), request_ip())
def test_extract_ip_from_wsgi_request_cf_connection_ip(self):
self.assertEqual(
ExtractorsIp(request_with_ip_cf_connecting_ip()).call(),
request_ip_next())
def _ip(self):
return ExtractorsIp(self.pre_headers).call()
def __init__(self, request, cookies):
self.client_id = ExtractorsClientId(request.environ, cookies).call()
self.headers = ExtractorsHeaders(request.environ).call()
self.request_ip = ExtractorsIp(request).call()
def test_extract_ip_for_spoof_ip_attempt(self):
headers = {'Client-Ip': '6.6.6.6', 'X-Forwarded-For': '6.6.6.6, 2.2.2.3, 192.168.0.7'}
self.assertEqual(
ExtractorsIp(headers).call(),
'2.2.2.3'
)
def test_extract_ip(self):
headers = {'X-Forwarded-For': '1.2.3.5'}
self.assertEqual(ExtractorsIp(headers).call(), '1.2.3.5')