From 4b6b5effc744583eb0227d14d0c5e324c50cf074 Mon Sep 17 00:00:00 2001 From: Samuel Fekete Date: Fri, 3 Nov 2017 11:07:23 +0000 Subject: [PATCH] Add headers in proxy server response --- tests/integration/test_proxy.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/integration/test_proxy.py b/tests/integration/test_proxy.py index d3c8df1..a1fc003 100644 --- a/tests/integration/test_proxy.py +++ b/tests/integration/test_proxy.py @@ -19,10 +19,15 @@ class Proxy(SimpleHTTPServer.SimpleHTTPRequestHandler): ''' Simple proxy server. - (from: http://effbot.org/librarybook/simplehttpserver.htm). + (Inspired by: http://effbot.org/librarybook/simplehttpserver.htm). ''' def do_GET(self): - self.copyfile(urlopen(self.path), self.wfile) + upstream_response = urlopen(self.path) + self.send_response(upstream_response.status, upstream_response.msg) + for header in upstream_response.headers.items(): + self.send_header(*header) + self.end_headers() + self.copyfile(upstream_response, self.wfile) @pytest.yield_fixture(scope='session')