bundling required python dependencies, to make it easier on Tails users

lib/werkzeug/testsuite/contrib/fixers.py

@@ -0,0 +1,193 @@
+# -*- coding: utf-8 -*-
+"""
+    werkzeug.testsuite.fixers
+    ~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Server / Browser fixers.
+
+    :copyright: (c) 2013 by Armin Ronacher.
+    :license: BSD, see LICENSE for more details.
+"""
+import unittest
+
+from werkzeug.testsuite import WerkzeugTestCase
+from werkzeug.datastructures import ResponseCacheControl
+from werkzeug.http import parse_cache_control_header
+
+from werkzeug.test import create_environ, Client
+from werkzeug.wrappers import Request, Response
+from werkzeug.contrib import fixers
+from werkzeug.utils import redirect
+
+
+@Request.application
+def path_check_app(request):
+    return Response('PATH_INFO: %s\nSCRIPT_NAME: %s' % (
+        request.environ.get('PATH_INFO', ''),
+        request.environ.get('SCRIPT_NAME', '')
+    ))
+
+
+class ServerFixerTestCase(WerkzeugTestCase):
+
+    def test_cgi_root_fix(self):
+        app = fixers.CGIRootFix(path_check_app)
+        response = Response.from_app(app, dict(create_environ(),
+            SCRIPT_NAME='/foo',
+            PATH_INFO='/bar',
+            SERVER_SOFTWARE='lighttpd/1.4.27'
+        ))
+        self.assert_equal(response.get_data(),
+                          b'PATH_INFO: /foo/bar\nSCRIPT_NAME: ')
+
+    def test_cgi_root_fix_custom_app_root(self):
+        app = fixers.CGIRootFix(path_check_app, app_root='/baz/poop/')
+        response = Response.from_app(app, dict(create_environ(),
+            SCRIPT_NAME='/foo',
+            PATH_INFO='/bar'
+        ))
+        self.assert_equal(response.get_data(), b'PATH_INFO: /foo/bar\nSCRIPT_NAME: baz/poop')
+
+    def test_path_info_from_request_uri_fix(self):
+        app = fixers.PathInfoFromRequestUriFix(path_check_app)
+        for key in 'REQUEST_URI', 'REQUEST_URL', 'UNENCODED_URL':
+            env = dict(create_environ(), SCRIPT_NAME='/test', PATH_INFO='/?????')
+            env[key] = '/test/foo%25bar?drop=this'
+            response = Response.from_app(app, env)
+            self.assert_equal(response.get_data(), b'PATH_INFO: /foo%bar\nSCRIPT_NAME: /test')
+
+    def test_proxy_fix(self):
+        @Request.application
+        def app(request):
+            return Response('%s|%s' % (
+                request.remote_addr,
+                # do not use request.host as this fixes too :)
+                request.environ['HTTP_HOST']
+            ))
+        app = fixers.ProxyFix(app, num_proxies=2)
+        environ = dict(create_environ(),
+            HTTP_X_FORWARDED_PROTO="https",
+            HTTP_X_FORWARDED_HOST='example.com',
+            HTTP_X_FORWARDED_FOR='1.2.3.4, 5.6.7.8',
+            REMOTE_ADDR='127.0.0.1',
+            HTTP_HOST='fake'
+        )
+
+        response = Response.from_app(app, environ)
+
+        self.assert_equal(response.get_data(), b'1.2.3.4|example.com')
+
+        # And we must check that if it is a redirection it is
+        # correctly done:
+
+        redirect_app = redirect('/foo/bar.hml')
+        response = Response.from_app(redirect_app, environ)
+
+        wsgi_headers = response.get_wsgi_headers(environ)
+        assert wsgi_headers['Location'] == 'https://example.com/foo/bar.hml'
+
+    def test_proxy_fix_weird_enum(self):
+        @fixers.ProxyFix
+        @Request.application
+        def app(request):
+            return Response(request.remote_addr)
+        environ = dict(create_environ(),
+            HTTP_X_FORWARDED_FOR=',',
+            REMOTE_ADDR='127.0.0.1',
+        )
+
+        response = Response.from_app(app, environ)
+        self.assert_strict_equal(response.get_data(), b'127.0.0.1')
+
+    def test_header_rewriter_fix(self):
+        @Request.application
+        def application(request):
+            return Response("", headers=[
+                ('X-Foo', 'bar')
+            ])
+        application = fixers.HeaderRewriterFix(application, ('X-Foo',), (('X-Bar', '42'),))
+        response = Response.from_app(application, create_environ())
+        assert response.headers['Content-Type'] == 'text/plain; charset=utf-8'
+        assert 'X-Foo' not in response.headers
+        assert response.headers['X-Bar'] == '42'
+
+
+class BrowserFixerTestCase(WerkzeugTestCase):
+
+    def test_ie_fixes(self):
+        @fixers.InternetExplorerFix
+        @Request.application
+        def application(request):
+            response = Response('binary data here', mimetype='application/vnd.ms-excel')
+            response.headers['Vary'] = 'Cookie'
+            response.headers['Content-Disposition'] = 'attachment; filename=foo.xls'
+            return response
+
+        c = Client(application, Response)
+        response = c.get('/', headers=[
+            ('User-Agent', 'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)')
+        ])
+
+        # IE gets no vary
+        self.assert_equal(response.get_data(), b'binary data here')
+        assert 'vary' not in response.headers
+        assert response.headers['content-disposition'] == 'attachment; filename=foo.xls'
+        assert response.headers['content-type'] == 'application/vnd.ms-excel'
+
+        # other browsers do
+        c = Client(application, Response)
+        response = c.get('/')
+        self.assert_equal(response.get_data(), b'binary data here')
+        assert 'vary' in response.headers
+
+        cc = ResponseCacheControl()
+        cc.no_cache = True
+
+        @fixers.InternetExplorerFix
+        @Request.application
+        def application(request):
+            response = Response('binary data here', mimetype='application/vnd.ms-excel')
+            response.headers['Pragma'] = ', '.join(pragma)
+            response.headers['Cache-Control'] = cc.to_header()
+            response.headers['Content-Disposition'] = 'attachment; filename=foo.xls'
+            return response
+
+
+        # IE has no pragma or cache control
+        pragma = ('no-cache',)
+        c = Client(application, Response)
+        response = c.get('/', headers=[
+            ('User-Agent', 'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)')
+        ])
+        self.assert_equal(response.get_data(), b'binary data here')
+        assert 'pragma' not in response.headers
+        assert 'cache-control' not in response.headers
+        assert response.headers['content-disposition'] == 'attachment; filename=foo.xls'
+
+        # IE has simplified pragma
+        pragma = ('no-cache', 'x-foo')
+        cc.proxy_revalidate = True
+        response = c.get('/', headers=[
+            ('User-Agent', 'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)')
+        ])
+        self.assert_equal(response.get_data(), b'binary data here')
+        assert response.headers['pragma'] == 'x-foo'
+        assert response.headers['cache-control'] == 'proxy-revalidate'
+        assert response.headers['content-disposition'] == 'attachment; filename=foo.xls'
+
+        # regular browsers get everything
+        response = c.get('/')
+        self.assert_equal(response.get_data(), b'binary data here')
+        assert response.headers['pragma'] == 'no-cache, x-foo'
+        cc = parse_cache_control_header(response.headers['cache-control'],
+                                        cls=ResponseCacheControl)
+        assert cc.no_cache
+        assert cc.proxy_revalidate
+        assert response.headers['content-disposition'] == 'attachment; filename=foo.xls'
+
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(ServerFixerTestCase))
+    suite.addTest(unittest.makeSuite(BrowserFixerTestCase))
+    return suite