diff options
Diffstat (limited to 'tests/test_proxy_headers.py')
| -rw-r--r-- | tests/test_proxy_headers.py | 63 |
1 files changed, 51 insertions, 12 deletions
diff --git a/tests/test_proxy_headers.py b/tests/test_proxy_headers.py index e6f0ed6..9ed131e 100644 --- a/tests/test_proxy_headers.py +++ b/tests/test_proxy_headers.py @@ -23,7 +23,11 @@ class TestProxyHeadersMiddleware(unittest.TestCase): inner = DummyApp() app = self._makeOne(inner) response = self._callFUT( - app, headers={"X_FOO": "BAR", "X_FORWARDED_PROTO": "https",} + app, + headers={ + "X_FOO": "BAR", + "X_FORWARDED_PROTO": "https", + }, ) self.assertEqual(response.status, "200 OK") self.assertEqual(inner.environ["wsgi.url_scheme"], "http") @@ -38,7 +42,10 @@ class TestProxyHeadersMiddleware(unittest.TestCase): response = self._callFUT( app, addr=["192.168.1.1", 8080], - headers={"X_FOO": "BAR", "X_FORWARDED_PROTO": "https",}, + headers={ + "X_FOO": "BAR", + "X_FORWARDED_PROTO": "https", + }, ) environ = inner.environ @@ -113,7 +120,9 @@ class TestProxyHeadersMiddleware(unittest.TestCase): "X_FORWARDED_HOST": "example.org", } response = self._callFUT( - app, addr=["192.168.1.1", 80], headers=headers_orig.copy(), + app, + addr=["192.168.1.1", 80], + headers=headers_orig.copy(), ) self.assertEqual(response.status, "200 OK") environ = inner.environ @@ -178,7 +187,9 @@ class TestProxyHeadersMiddleware(unittest.TestCase): def test_parse_proxy_headers_forwarded_for(self): inner = DummyApp() app = self._makeOne( - inner, trusted_proxy="*", trusted_proxy_headers={"x-forwarded-for"}, + inner, + trusted_proxy="*", + trusted_proxy_headers={"x-forwarded-for"}, ) response = self._callFUT(app, headers={"X_FORWARDED_FOR": "192.0.2.1"}) self.assertEqual(response.status, "200 OK") @@ -189,7 +200,9 @@ class TestProxyHeadersMiddleware(unittest.TestCase): def test_parse_proxy_headers_forwarded_for_v6_missing_brackets(self): inner = DummyApp() app = self._makeOne( - inner, trusted_proxy="*", trusted_proxy_headers={"x-forwarded-for"}, + inner, + trusted_proxy="*", + trusted_proxy_headers={"x-forwarded-for"}, ) response = self._callFUT(app, headers={"X_FORWARDED_FOR": "2001:db8::0"}) self.assertEqual(response.status, "200 OK") @@ -473,7 +486,10 @@ class TestProxyHeadersMiddleware(unittest.TestCase): trusted_proxy_headers={"forwarded"}, ) response = self._callFUT( - app, headers={"FORWARDED": "For=198.51.100.2;;proto=https;by=_unused",} + app, + headers={ + "FORWARDED": "For=198.51.100.2;;proto=https;by=_unused", + }, ) self.assertEqual(response.status, "200 OK") @@ -489,7 +505,10 @@ class TestProxyHeadersMiddleware(unittest.TestCase): trusted_proxy_headers={"forwarded"}, ) response = self._callFUT( - app, headers={"FORWARDED": "For=198.51.100.2; proto =https",} + app, + headers={ + "FORWARDED": "For=198.51.100.2; proto =https", + }, ) self.assertEqual(response.status, "400 Bad Request") self.assertIn(b'Header "Forwarded" malformed', response.body) @@ -503,7 +522,10 @@ class TestProxyHeadersMiddleware(unittest.TestCase): trusted_proxy_headers={"forwarded"}, ) response = self._callFUT( - app, headers={"FORWARDED": 'For= "198.51.100.2"; proto =https',} + app, + headers={ + "FORWARDED": 'For= "198.51.100.2"; proto =https', + }, ) self.assertEqual(response.status, "400 Bad Request") self.assertIn(b'Header "Forwarded" malformed', response.body) @@ -553,7 +575,11 @@ class TestProxyHeadersMiddleware(unittest.TestCase): def test_parse_no_valid_proxy_headers(self): inner = DummyApp() - app = self._makeOne(inner, trusted_proxy="*", trusted_proxy_count=1,) + app = self._makeOne( + inner, + trusted_proxy="*", + trusted_proxy_count=1, + ) response = self._callFUT( app, headers={ @@ -580,7 +606,12 @@ class TestProxyHeadersMiddleware(unittest.TestCase): trusted_proxy_headers={"x-forwarded-proto"}, logger=logger, ) - response = self._callFUT(app, headers={"X_FORWARDED_PROTO": "http, https",}) + response = self._callFUT( + app, + headers={ + "X_FORWARDED_PROTO": "http, https", + }, + ) self.assertEqual(response.status, "400 Bad Request") self.assertIn(b'Header "X-Forwarded-Proto" malformed', response.body) @@ -594,7 +625,12 @@ class TestProxyHeadersMiddleware(unittest.TestCase): trusted_proxy_headers={"x-forwarded-port"}, logger=logger, ) - response = self._callFUT(app, headers={"X_FORWARDED_PORT": "443, 80",}) + response = self._callFUT( + app, + headers={ + "X_FORWARDED_PORT": "443, 80", + }, + ) self.assertEqual(response.status, "400 Bad Request") self.assertIn(b'Header "X-Forwarded-Port" malformed', response.body) @@ -701,7 +737,10 @@ class DummyResponse: def DummyEnviron( - addr=("127.0.0.1", 8080), scheme="http", server="localhost", headers=None, + addr=("127.0.0.1", 8080), + scheme="http", + server="localhost", + headers=None, ): environ = { "REMOTE_ADDR": addr[0], |