diff --git a/pywb/rewrite/rewrite_live.py b/pywb/rewrite/rewrite_live.py index ca2c04a0..dc43d51e 100644 --- a/pywb/rewrite/rewrite_live.py +++ b/pywb/rewrite/rewrite_live.py @@ -93,7 +93,7 @@ class LiveRewriter(object): elif name == 'REL_REFERER': name = 'Referer' else: - continue + value = None if value: headers[name] = value diff --git a/pywb/rewrite/test/test_rewrite_live.py b/pywb/rewrite/test/test_rewrite_live.py index 99fb6074..eb4d51d4 100644 --- a/pywb/rewrite/test/test_rewrite_live.py +++ b/pywb/rewrite/test/test_rewrite_live.py @@ -26,6 +26,14 @@ def test_csrf_token_headers(): 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(): rewriter = LiveRewriter() env = {'HTTP_COOKIE': 'A=B'}