From f76ba06c42daf01d028d10bf95d80ede0b5abf81 Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Sun, 21 Oct 2018 13:59:29 -0700 Subject: [PATCH] header rewriter: ensure the 'Status' header is prefix-rewritten, update test --- pywb/rewrite/header_rewriter.py | 2 ++ pywb/rewrite/test/test_header_rewriter.py | 7 +++++-- 2 files changed, 7 insertions(+), 2 deletions(-) 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