def testInsertBaseInSetResultUpdatesContentLength(self):
# Make sure that the Content-Length header is updated to account
# for an inserted <base> tag.
response = BrowserResponse()
response.setHeader('content-type', 'text/html')
base = 'http://localhost/folder/'
response.setBase(base)
inserted_text = '\n<base href="%s" />\n' % base
html_page = b"""<html>
<head></head>
<blah>
</html>
"""
response.setResult(html_page)
self.assertEqual(int(response.getHeader('content-length')),
len(html_page) + len(inserted_text))
def test_not_DWIM_for_304_response(self):
# Don't guess the content type with 304 responses which MUST NOT /
# SHOULD NOT include it.
#
# http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html#sec10.3.5
#
# NOTE: The content type is sill guessed if status us set after
# the result :(
response = BrowserResponse()
response.setStatus(304)
response.setResult(b'')
self.assertEqual(response.getHeader('content-type'), None)
def testInsertBase(self):
response = BrowserResponse()
response.setHeader('content-type', 'text/html')
insertBase = response._BrowserResponse__insertBase
# Make sure that bases are inserted
response.setBase('http://localhost/folder/')
self.assert_('<base href="http://localhost/folder/" />' in insertBase(
'<html><head></head><body>Page</body></html>'))
# Ensure that unicode bases work as well
response.setBase(u'http://localhost/folder/')
body = insertBase('<html><head></head><body>Page</body></html>')
self.assert_(isinstance(body, str))
self.assert_('<base href="http://localhost/folder/" />' in body)
# Ensure that encoded bodies work, when a base is inserted.
response.setBase('http://localhost/folder')
result = insertBase(
'<html><head></head><body>\xc3\x9bung</body></html>')
self.assert_(isinstance(body, str))
self.assert_('<base href="http://localhost/folder" />' in result)
def testInsertBaseInSetResultUpdatesContentLength(self):
# Make sure that the Content-Length header is updated to account
# for an inserted <base> tag.
response = BrowserResponse()
response.setHeader('content-type', 'text/html')
base = 'http://localhost/folder/'
response.setBase(base)
inserted_text = '\n<base href="%s" />\n' % base
html_page = """<html>
<head></head>
<blah>
</html>
"""
response.setResult(html_page)
self.assertEquals(
int(response.getHeader('content-length')),
len(html_page) + len(inserted_text))
def test_not_DWIM_for_304_response(self):
# Don't guess the content type with 304 responses which MUST NOT /
# SHOULD NOT include it.
#
# http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html#sec10.3.5
#
# NOTE: The content type is sill guessed if status us set after
# the result :(
response = BrowserResponse()
response.setStatus(304)
response.setResult(b'')
self.assertEqual(response.getHeader('content-type'), None)
def testInsertBase(self):
response = BrowserResponse()
response.setHeader('content-type', 'text/html')
insertBase = response._BrowserResponse__insertBase
# Make sure that bases are inserted
response.setBase('http://localhost/folder/')
self.assert_(
'<base href="http://localhost/folder/" />' in
insertBase('<html><head></head><body>Page</body></html>'))
# Ensure that unicode bases work as well
response.setBase(u'http://localhost/folder/')
body = insertBase('<html><head></head><body>Page</body></html>')
self.assert_(isinstance(body, str))
self.assert_('<base href="http://localhost/folder/" />' in body)
# Ensure that encoded bodies work, when a base is inserted.
response.setBase('http://localhost/folder')
result = insertBase(
'<html><head></head><body>\xc3\x9bung</body></html>')
self.assert_(isinstance(body, str))
self.assert_('<base href="http://localhost/folder" />' in result)
def test_handleException(self):
response = BrowserResponse()
try:
raise ValueError(1)
except:
exc_info = sys.exc_info()
response.handleException(exc_info)
self.assertEqual(response.getHeader("content-type"),
"text/html;charset=utf-8")
self.assertEqual(response.getStatus(), 500)
self.assertTrue(response.consumeBody() in [
b"<html><head><title><type 'exceptions.ValueError'></title></head>\n"
b"<body><h2><type 'exceptions.ValueError'></h2>\n"
b"A server error occurred.\n"
b"</body></html>\n",
b"<html><head><title>ValueError</title></head>\n"
b"<body><h2>ValueError</h2>\n"
b"A server error occurred.\n"
b"</body></html>\n"
])
def test_handleException(self):
response = BrowserResponse()
try:
raise ValueError(1)
except:
exc_info = sys.exc_info()
response.handleException(exc_info)
self.assertEquals(response.getHeader("content-type"),
"text/html;charset=utf-8")
self.assertEquals(response.getStatus(), 500)
self.assert_(response.consumeBody() in
["<html><head><title><type 'exceptions.ValueError'></title></head>\n"
"<body><h2><type 'exceptions.ValueError'></h2>\n"
"A server error occurred.\n"
"</body></html>\n",
"<html><head><title>ValueError</title></head>\n"
"<body><h2>ValueError</h2>\n"
"A server error occurred.\n"
"</body></html>\n"]
)
def test_contentType_DWIM_in_setResult(self):
response = BrowserResponse()
response.setResult("""<html>
<blah>
</html>
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/html"))
response = BrowserResponse()
response.setResult("""<html foo="1"
bar="x">
<blah>
</html>
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/html"))
response = BrowserResponse()
response.setResult("""<html foo="1"
bar="x">
<blah>
</html>
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/html"))
response = BrowserResponse()
response.setResult("""<!doctype html>
<html foo="1"
bar="x">
<blah>
</html>
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/html"))
response = BrowserResponse()
response.setResult("""Hello world
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/plain"))
response = BrowserResponse()
response.setResult("""<p>Hello world
""")
self.assertTrue(
response.getHeader('content-type').startswith("text/plain"))
def test_interface(self):
rp = BrowserResponse()
verifyObject(IHTTPResponse, rp)
verifyObject(IHTTPApplicationResponse, rp)
verifyObject(IResponse, rp)
def test_contentType_DWIM_in_setResult(self):
response = BrowserResponse()
response.setResult(
"""<html>
<blah>
</html>
""")
self.assert_(response.getHeader('content-type').startswith("text/html")
)
response = BrowserResponse()
response.setResult(
"""<html foo="1"
bar="x">
<blah>
</html>
""")
self.assert_(response.getHeader('content-type').startswith("text/html")
)
response = BrowserResponse()
response.setResult(
"""<html foo="1"
bar="x">
<blah>
</html>
""")
self.assert_(response.getHeader('content-type').startswith("text/html")
)
response = BrowserResponse()
response.setResult(
"""<!doctype html>
<html foo="1"
bar="x">
<blah>
</html>
""")
self.assert_(response.getHeader('content-type').startswith("text/html")
)
response = BrowserResponse()
response.setResult(
"""Hello world
""")
self.assert_(response.getHeader('content-type').startswith(
"text/plain")
)
response = BrowserResponse()
response.setResult(
"""<p>Hello world
""")
self.assert_(
response.getHeader('content-type').startswith("text/plain")
)
