backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

rewrite: add basic test for X-Forwarded-Proto #57

Browse Source
This commit is contained in:
Ilya Kreymer 2015-02-04 21:44:18 -08:00
parent 78812c8085
commit cc144fdead
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pywb/rewrite/rewrite_live.py
View File

@ -93,7 +93,7 @@ class LiveRewriter(object):
elif name == 'REL_REFERER': elif name == 'REL_REFERER':
name = 'Referer' name = 'Referer'
else: else:
continue value = None
if value: if value:
headers[name] = value headers[name] = value

8
pywb/rewrite/test/test_rewrite_live.py
View File

@ -26,6 +26,14 @@ def test_csrf_token_headers():
assert req_headers == {'X-CSRFToken': 'foobar', 'Cookie': 'csrftoken=foobar'} assert req_headers == {'X-CSRFToken': 'foobar', 'Cookie': 'csrftoken=foobar'}
def test_forwarded_scheme():
rewriter = LiveRewriter()
env = {'HTTP_X_FORWARDED_PROTO': 'https', 'Other': 'Value'}
req_headers = rewriter.translate_headers('http://example.com/', 'com,example)/', env)
assert req_headers == {'X-Forwarded-Proto': 'http'}
def test_req_cookie_rewrite_1(): def test_req_cookie_rewrite_1():
rewriter = LiveRewriter() rewriter = LiveRewriter()
env = {'HTTP_COOKIE': 'A=B'} env = {'HTTP_COOKIE': 'A=B'}