summaryrefslogtreecommitdiffhomepage
path: root/test/test_http_header.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--test/test_http_header.py476
1 files changed, 299 insertions, 177 deletions
diff --git a/test/test_http_header.py b/test/test_http_header.py
index f2294371..002a8826 100644
--- a/test/test_http_header.py
+++ b/test/test_http_header.py
@@ -1,216 +1,302 @@
import unittest
import unit
-class TestUnitHTTPHeader(unit.TestUnitApplicationPython):
+class TestUnitHTTPHeader(unit.TestUnitApplicationPython):
def setUpClass():
unit.TestUnit().check_modules('python')
def test_http_header_value_leading_sp(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': ' ,',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': ' ,',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value leading sp status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value leading sp custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value leading sp custom header',
+ )
def test_http_header_value_leading_htab(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': '\t,',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': '\t,',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value leading htab status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value leading htab custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value leading htab custom header',
+ )
def test_http_header_value_trailing_sp(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': ', ',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': ', ',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value trailing sp status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value trailing sp custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value trailing sp custom header',
+ )
def test_http_header_value_trailing_htab(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': ',\t',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': ',\t',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value trailing htab status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value trailing htab custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value trailing htab custom header',
+ )
def test_http_header_value_both_sp(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': ' , ',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': ' , ',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value both sp status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value both sp custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value both sp custom header',
+ )
def test_http_header_value_both_htab(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': '\t,\t',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': '\t,\t',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value both htab status')
- self.assertEqual(resp['headers']['Custom-Header'], ',',
- 'value both htab custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ ',',
+ 'value both htab custom header',
+ )
def test_http_header_value_chars(self):
self.load('custom_header')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header': '(),/:;<=>?@[\]{}\t !#$%&\'*+-.^_`|~',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header': '(),/:;<=>?@[\]{}\t !#$%&\'*+-.^_`|~',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 200, 'value chars status')
- self.assertEqual(resp['headers']['Custom-Header'],
- '(),/:;<=>?@[\]{}\t !#$%&\'*+-.^_`|~', 'value chars custom header')
+ self.assertEqual(
+ resp['headers']['Custom-Header'],
+ '(),/:;<=>?@[\]{}\t !#$%&\'*+-.^_`|~',
+ 'value chars custom header',
+ )
def test_http_header_value_chars_edge(self):
self.load('custom_header')
- resp = self.http(b"""GET / HTTP/1.1
+ resp = self.http(
+ b"""GET / HTTP/1.1
Host: localhost
Custom-Header: \x20\xFF
Connection: close
-""", raw=True, encoding='latin1')
+""",
+ raw=True,
+ encoding='latin1',
+ )
self.assertEqual(resp['status'], 200, 'value chars edge status')
- self.assertEqual(resp['headers']['Custom-Header'], '\xFF',
- 'value chars edge')
+ self.assertEqual(
+ resp['headers']['Custom-Header'], '\xFF', 'value chars edge'
+ )
def test_http_header_value_chars_below(self):
self.load('custom_header')
- resp = self.http(b"""GET / HTTP/1.1
+ resp = self.http(
+ b"""GET / HTTP/1.1
Host: localhost
Custom-Header: \x1F
Connection: close
-""", raw=True)
+""",
+ raw=True,
+ )
self.assertEqual(resp['status'], 400, 'value chars below')
def test_http_header_field_leading_sp(self):
self.load('empty')
- resp = self.get(headers={
- 'Host': 'localhost',
- ' Custom-Header': 'blah',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ ' Custom-Header': 'blah',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 400, 'field leading sp')
def test_http_header_field_leading_htab(self):
self.load('empty')
- resp = self.get(headers={
- 'Host': 'localhost',
- '\tCustom-Header': 'blah',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ '\tCustom-Header': 'blah',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 400, 'field leading htab')
def test_http_header_field_trailing_sp(self):
self.load('empty')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header ': 'blah',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header ': 'blah',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 400, 'field trailing sp')
def test_http_header_field_trailing_htab(self):
self.load('empty')
- resp = self.get(headers={
- 'Host': 'localhost',
- 'Custom-Header\t': 'blah',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={
+ 'Host': 'localhost',
+ 'Custom-Header\t': 'blah',
+ 'Connection': 'close',
+ }
+ )
self.assertEqual(resp['status'], 400, 'field trailing htab')
def test_http_header_content_length_big(self):
self.load('empty')
- self.assertEqual(self.post(headers={
- 'Host': 'localhost',
- 'Content-Length': str(2 ** 64),
- 'Connection': 'close'
- }, body='X' * 1000)['status'], 400, 'Content-Length big')
+ self.assertEqual(
+ self.post(
+ headers={
+ 'Host': 'localhost',
+ 'Content-Length': str(2 ** 64),
+ 'Connection': 'close',
+ },
+ body='X' * 1000,
+ )['status'],
+ 400,
+ 'Content-Length big',
+ )
def test_http_header_content_length_negative(self):
self.load('empty')
- self.assertEqual(self.post(headers={
- 'Host': 'localhost',
- 'Content-Length': '-100',
- 'Connection': 'close'
- }, body='X' * 1000)['status'], 400, 'Content-Length negative')
+ self.assertEqual(
+ self.post(
+ headers={
+ 'Host': 'localhost',
+ 'Content-Length': '-100',
+ 'Connection': 'close',
+ },
+ body='X' * 1000,
+ )['status'],
+ 400,
+ 'Content-Length negative',
+ )
def test_http_header_content_length_text(self):
self.load('empty')
- self.assertEqual(self.post(headers={
- 'Host': 'localhost',
- 'Content-Length': 'blah',
- 'Connection': 'close'
- }, body='X' * 1000)['status'], 400, 'Content-Length text')
+ self.assertEqual(
+ self.post(
+ headers={
+ 'Host': 'localhost',
+ 'Content-Length': 'blah',
+ 'Connection': 'close',
+ },
+ body='X' * 1000,
+ )['status'],
+ 400,
+ 'Content-Length text',
+ )
def test_http_header_content_length_multiple_values(self):
self.load('empty')
- self.assertEqual(self.post(headers={
- 'Host': 'localhost',
- 'Content-Length': '41, 42',
- 'Connection': 'close'
- }, body='X' * 1000)['status'], 400, 'Content-Length multiple value')
+ self.assertEqual(
+ self.post(
+ headers={
+ 'Host': 'localhost',
+ 'Content-Length': '41, 42',
+ 'Connection': 'close',
+ },
+ body='X' * 1000,
+ )['status'],
+ 400,
+ 'Content-Length multiple value',
+ )
def test_http_header_content_length_multiple_fields(self):
self.load('empty')
- self.assertEqual(self.post(headers={
- 'Host': 'localhost',
- 'Content-Length': ['41', '42'],
- 'Connection': 'close'
- }, body='X' * 1000)['status'], 400, 'Content-Length multiple fields')
+ self.assertEqual(
+ self.post(
+ headers={
+ 'Host': 'localhost',
+ 'Content-Length': ['41', '42'],
+ 'Connection': 'close',
+ },
+ body='X' * 1000,
+ )['status'],
+ 400,
+ 'Content-Length multiple fields',
+ )
def test_http_header_host_absent(self):
self.load('host')
@@ -218,146 +304,182 @@ Connection: close
resp = self.get(headers={'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host absent status')
- self.assertNotEqual(resp['headers']['X-Server-Name'], '',
- 'Host absent SERVER_NAME')
+ self.assertNotEqual(
+ resp['headers']['X-Server-Name'], '', 'Host absent SERVER_NAME'
+ )
def test_http_header_host_empty(self):
self.load('host')
- resp = self.get(headers={
- 'Host': '',
- 'Connection': 'close'
- })
+ resp = self.get(headers={'Host': '', 'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host empty status')
- self.assertNotEqual(resp['headers']['X-Server-Name'], '',
- 'Host empty SERVER_NAME')
+ self.assertNotEqual(
+ resp['headers']['X-Server-Name'], '', 'Host empty SERVER_NAME'
+ )
def test_http_header_host_big(self):
self.load('empty')
- self.assertEqual(self.get(headers={
- 'Host': 'X' * 10000,
- 'Connection': 'close'
- })['status'], 431, 'Host big')
+ self.assertEqual(
+ self.get(headers={'Host': 'X' * 10000, 'Connection': 'close'})[
+ 'status'
+ ],
+ 431,
+ 'Host big',
+ )
def test_http_header_host_port(self):
self.load('host')
- resp = self.get(headers={
- 'Host': 'exmaple.com:7080',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={'Host': 'exmaple.com:7080', 'Connection': 'close'}
+ )
self.assertEqual(resp['status'], 200, 'Host port status')
- self.assertEqual(resp['headers']['X-Server-Name'], 'exmaple.com',
- 'Host port SERVER_NAME')
- self.assertEqual(resp['headers']['X-Http-Host'], 'exmaple.com:7080',
- 'Host port HTTP_HOST')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ 'exmaple.com',
+ 'Host port SERVER_NAME',
+ )
+ self.assertEqual(
+ resp['headers']['X-Http-Host'],
+ 'exmaple.com:7080',
+ 'Host port HTTP_HOST',
+ )
def test_http_header_host_port_empty(self):
self.load('host')
- resp = self.get(headers={
- 'Host': 'exmaple.com:',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={'Host': 'exmaple.com:', 'Connection': 'close'}
+ )
self.assertEqual(resp['status'], 200, 'Host port empty status')
- self.assertEqual(resp['headers']['X-Server-Name'], 'exmaple.com',
- 'Host port empty SERVER_NAME')
- self.assertEqual(resp['headers']['X-Http-Host'], 'exmaple.com:',
- 'Host port empty HTTP_HOST')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ 'exmaple.com',
+ 'Host port empty SERVER_NAME',
+ )
+ self.assertEqual(
+ resp['headers']['X-Http-Host'],
+ 'exmaple.com:',
+ 'Host port empty HTTP_HOST',
+ )
def test_http_header_host_literal(self):
self.load('host')
- resp = self.get(headers={
- 'Host': '127.0.0.1',
- 'Connection': 'close'
- })
+ resp = self.get(headers={'Host': '127.0.0.1', 'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host literal status')
- self.assertEqual(resp['headers']['X-Server-Name'], '127.0.0.1',
- 'Host literal SERVER_NAME')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ '127.0.0.1',
+ 'Host literal SERVER_NAME',
+ )
def test_http_header_host_literal_ipv6(self):
self.load('host')
- resp = self.get(headers={
- 'Host': '[::1]:7080',
- 'Connection': 'close'
- })
+ resp = self.get(headers={'Host': '[::1]:7080', 'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host literal ipv6 status')
- self.assertEqual(resp['headers']['X-Server-Name'], '[::1]',
- 'Host literal ipv6 SERVER_NAME')
- self.assertEqual(resp['headers']['X-Http-Host'], '[::1]:7080',
- 'Host literal ipv6 HTTP_HOST')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ '[::1]',
+ 'Host literal ipv6 SERVER_NAME',
+ )
+ self.assertEqual(
+ resp['headers']['X-Http-Host'],
+ '[::1]:7080',
+ 'Host literal ipv6 HTTP_HOST',
+ )
def test_http_header_host_trailing_period(self):
self.load('host')
- resp = self.get(headers={
- 'Host': '127.0.0.1.',
- 'Connection': 'close'
- })
+ resp = self.get(headers={'Host': '127.0.0.1.', 'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host trailing period status')
- self.assertEqual(resp['headers']['X-Server-Name'], '127.0.0.1',
- 'Host trailing period SERVER_NAME')
- self.assertEqual(resp['headers']['X-Http-Host'], '127.0.0.1.',
- 'Host trailing period HTTP_HOST')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ '127.0.0.1',
+ 'Host trailing period SERVER_NAME',
+ )
+ self.assertEqual(
+ resp['headers']['X-Http-Host'],
+ '127.0.0.1.',
+ 'Host trailing period HTTP_HOST',
+ )
def test_http_header_host_trailing_period_2(self):
self.load('host')
- resp = self.get(headers={
- 'Host': 'EXAMPLE.COM.',
- 'Connection': 'close'
- })
+ resp = self.get(
+ headers={'Host': 'EXAMPLE.COM.', 'Connection': 'close'}
+ )
self.assertEqual(resp['status'], 200, 'Host trailing period 2 status')
- self.assertEqual(resp['headers']['X-Server-Name'], 'example.com',
- 'Host trailing period 2 SERVER_NAME')
- self.assertEqual(resp['headers']['X-Http-Host'], 'EXAMPLE.COM.',
- 'Host trailing period 2 HTTP_HOST')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ 'example.com',
+ 'Host trailing period 2 SERVER_NAME',
+ )
+ self.assertEqual(
+ resp['headers']['X-Http-Host'],
+ 'EXAMPLE.COM.',
+ 'Host trailing period 2 HTTP_HOST',
+ )
def test_http_header_host_case_insensitive(self):
self.load('host')
- resp = self.get(headers={
- 'Host': 'EXAMPLE.COM',
- 'Connection': 'close'
- })
+ resp = self.get(headers={'Host': 'EXAMPLE.COM', 'Connection': 'close'})
self.assertEqual(resp['status'], 200, 'Host case insensitive')
- self.assertEqual(resp['headers']['X-Server-Name'], 'example.com',
- 'Host case insensitive SERVER_NAME')
+ self.assertEqual(
+ resp['headers']['X-Server-Name'],
+ 'example.com',
+ 'Host case insensitive SERVER_NAME',
+ )
def test_http_header_host_double_dot(self):
self.load('empty')
- self.assertEqual(self.get(headers={
- 'Host': '127.0.0..1',
- 'Connection': 'close'
- })['status'], 400, 'Host double dot')
+ self.assertEqual(
+ self.get(headers={'Host': '127.0.0..1', 'Connection': 'close'})[
+ 'status'
+ ],
+ 400,
+ 'Host double dot',
+ )
def test_http_header_host_slash(self):
self.load('empty')
- self.assertEqual(self.get(headers={
- 'Host': '/localhost',
- 'Connection': 'close'
- })['status'], 400, 'Host slash')
+ self.assertEqual(
+ self.get(headers={'Host': '/localhost', 'Connection': 'close'})[
+ 'status'
+ ],
+ 400,
+ 'Host slash',
+ )
def test_http_header_host_multiple_fields(self):
self.load('empty')
- self.assertEqual(self.get(headers={
- 'Host': ['localhost', 'example.com'],
- 'Connection': 'close'
- })['status'], 400, 'Host multiple fields')
+ self.assertEqual(
+ self.get(
+ headers={
+ 'Host': ['localhost', 'example.com'],
+ 'Connection': 'close',
+ }
+ )['status'],
+ 400,
+ 'Host multiple fields',
+ )
+
if __name__ == '__main__':
TestUnitHTTPHeader.main()