diff --git a/pywb/rewrite/header_rewriter.py b/pywb/rewrite/header_rewriter.py index d11b0405..57d4f34f 100644 --- a/pywb/rewrite/header_rewriter.py +++ b/pywb/rewrite/header_rewriter.py @@ -58,6 +58,8 @@ class DefaultHeaderRewriter(object): 'set-cookie': 'cookie', + 'status': 'prefix', + 'strict-transport-security': 'prefix', 'trailer': 'prefix', diff --git a/pywb/rewrite/test/test_header_rewriter.py b/pywb/rewrite/test/test_header_rewriter.py index 551a5441..98870221 100644 --- a/pywb/rewrite/test/test_header_rewriter.py +++ b/pywb/rewrite/test/test_header_rewriter.py @@ -111,7 +111,8 @@ X-Archive-Orig-Transfer-Encoding: chunked\r\n\ ('Set-Cookie', 'foo=bar; Path=/; abc=123; Path=/path.html'), ('Content-Encoding', 'gzip'), ('Transfer-Encoding', 'chunked'), - ('X-Custom', 'test')] + ('X-Custom', 'test'), + ('Status', '200')] rwinfo = self.do_rewrite('200 OK', headers) http_headers = DefaultHeaderRewriter(rwinfo)() @@ -126,7 +127,9 @@ X-Archive-Orig-Transfer-Encoding: chunked\r\n\ assert(('X-Archive-Orig-Transfer-Encoding', 'chunked') in http_headers.headers) assert(('X-Custom', 'test') in http_headers.headers) - assert(len(http_headers.headers) == 7) + assert(('X-Archive-Orig-Status', '200') in http_headers.headers) + + assert(len(http_headers.headers) == 8) assert rwinfo.text_type == None assert rwinfo.charset == None