def test_extend_whitelisted_headers(self):
configuration.whitelisted += ['TEST']
self.assertEqual(
ExtractorsHeaders(environ()).call(),
{'X-Forwarded-For': '1.2.3.4', 'Test': '1'}
)
configuration.whitelisted.remove('Test')
def test_extract_headers(self):
self.assertEqual(
ExtractorsHeaders(formatted_headers()).call(), {
'Accept': 'application/json',
'Authorization': True,
'Cookie': True,
'Content-Length': '0',
'Ok': 'OK',
'User-Agent': 'Mozilla 1234',
'X-Forwarded-For': '1.2.3.4'
})
def test_denylisted_headers(self):
configuration.denylisted = ['Accept']
self.assertEqual(
ExtractorsHeaders(formatted_headers()).call(), {
'Accept': True,
'Authorization': True,
'Cookie': True,
'Content-Length': '0',
'Ok': 'OK',
'User-Agent': 'Mozilla 1234',
'X-Forwarded-For': '1.2.3.4'
})
def test_only_default_allowlisted_headers(self):
configuration.allowlisted = DEFAULT_ALLOWLIST
self.assertEqual(
ExtractorsHeaders(formatted_headers()).call(), {
'Accept': 'application/json',
'Authorization': True,
'Cookie': True,
'Ok': True,
'Content-Length': '0',
'User-Agent': 'Mozilla 1234',
'X-Forwarded-For': True
})
def test_extract_headers(self):
self.assertEqual(
ExtractorsHeaders(environ()).call(),
{'X-Forwarded-For': '1.2.3.4'})
def _headers(self):
return ExtractorsHeaders(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_denylisted_and_allowlisted_headers(self):
configuration.denylisted = ['Accept']
configuration.allowlisted = ['Accept']
self.assertEqual(
ExtractorsHeaders(formatted_headers()).call()['Accept'], True)