96 lines
4.0 KiB
Python
96 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
# This is run by the "run-tests" script.
|
|
import unittest
|
|
import os
|
|
from test import WWWROOT, TestHelper, parse, random_bytes
|
|
|
|
class TestCustomHeaders(TestHelper):
|
|
def setUp(self):
|
|
self.datalen = 2345
|
|
self.data = random_bytes(self.datalen)
|
|
self.url = '/data.jpeg'
|
|
self.not_found = '/not_found.jpeg'
|
|
self.fn = WWWROOT + self.url
|
|
with open(self.fn, 'wb') as f:
|
|
f.write(self.data)
|
|
|
|
def tearDown(self):
|
|
os.unlink(self.fn)
|
|
|
|
def test_custom_headers(self):
|
|
resp = self.get(self.url)
|
|
status, hdrs, body = parse(resp)
|
|
self.assertContains(status, '200 OK')
|
|
self.assertEqual(hdrs["Accept-Ranges"], "bytes")
|
|
self.assertEqual(hdrs["Content-Length"], str(self.datalen))
|
|
self.assertEqual(hdrs["Content-Type"], "image/jpeg")
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
self.assertContains(hdrs["Server"], "darkhttpd/")
|
|
assert body == self.data, [self.url, resp, status, hdrs, body]
|
|
self.assertEqual(body, self.data)
|
|
|
|
def test_custom_headers_not_found(self):
|
|
resp = self.get(self.not_found)
|
|
status, hdrs, body = parse(resp)
|
|
self.assertContains(status, '404 Not Found')
|
|
self.assertContains(hdrs["Server"], "darkhttpd/")
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
|
|
def test_custom_headers_listing(self):
|
|
resp = self.get("/")
|
|
status, hdrs, body = parse(resp)
|
|
self.assertContains(status, '200 OK')
|
|
self.assertContains(body, '<a href="data.jpeg">data.jpeg</a>')
|
|
self.assertContains(body, 'Generated by darkhttpd/')
|
|
self.assertEqual(hdrs["Accept-Ranges"], "bytes")
|
|
self.assertEqual(hdrs["Content-Type"], "text/html; charset=UTF-8")
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
self.assertContains(hdrs["Server"], "darkhttpd/")
|
|
|
|
def test_custom_headers_range(self):
|
|
resp = self.get(self.url, req_hdrs={'Range': 'bytes=0-99'})
|
|
status, hdrs, body = parse(resp)
|
|
self.assertContains(status, '206 Partial Content')
|
|
self.assertEqual(hdrs["Accept-Ranges"], "bytes")
|
|
self.assertEqual(hdrs["Content-Length"], '100')
|
|
self.assertEqual(hdrs["Content-Type"], "image/jpeg")
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
self.assertContains(hdrs["Server"], "darkhttpd/")
|
|
assert body == self.data[0:100], [self.url, resp, status, hdrs, body]
|
|
self.assertEqual(body, self.data[0:100])
|
|
|
|
def test_custom_header_if_modified_since(self):
|
|
resp1 = self.get(self.url, method="HEAD")
|
|
status, hdrs, body = parse(resp1)
|
|
lastmod = hdrs["Last-Modified"]
|
|
|
|
resp2 = self.get(self.url, method="GET", req_hdrs=
|
|
{"If-Modified-Since": lastmod })
|
|
status, hdrs, body = parse(resp2)
|
|
self.assertContains(status, "304 Not Modified")
|
|
self.assertEqual(hdrs["Accept-Ranges"], "bytes")
|
|
self.assertFalse("Last-Modified" in hdrs)
|
|
self.assertFalse("Content-Length" in hdrs)
|
|
self.assertFalse("Content-Type" in hdrs)
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
|
|
def test_custom_header_forward(self):
|
|
resp = self.get('/', req_hdrs={'Host': 'example.com'})
|
|
status, hdrs, body = parse(resp)
|
|
self.assertEqual(hdrs["X-Header-A"], "First Value")
|
|
self.assertEqual(hdrs["X-Header-B"], "Second Value")
|
|
self.assertContains(status, "301 Moved Permanently")
|
|
expect = "http://www.example.com/"
|
|
self.assertEqual(hdrs["Location"], expect)
|
|
self.assertContains(body, expect)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# vim:set ts=4 sw=4 et:
|