def test_none_media_type(self):
m = mock.Mock()
r = ice.Response(m)
r.media_type = None
r.response()
m.assert_called_with('200 OK', [
('Content-Length', '0'),
])
def test_start_with_body(self):
m = mock.Mock()
r = ice.Response(m)
r.body = 'foo'
r.response()
m.assert_called_with('200 OK',
[('Content-Type', 'text/html; charset=UTF-8'),
('Content-Length', '3')])
def test_non_text_media_type(self):
m = mock.Mock()
r = ice.Response(m)
r.media_type = 'image/png'
r.response()
m.assert_called_with('200 OK', [
('Content-Type', 'image/png'),
('Content-Length', '0'),
])
def test_text_media_type(self):
m = mock.Mock()
r = ice.Response(m)
r.media_type = 'text/css'
r.response()
m.assert_called_with('200 OK', [
('Content-Type', 'text/css; charset=UTF-8'),
('Content-Length', '0'),
])
def test_cookies(self):
m = mock.Mock()
r = ice.Response(m)
r.set_cookie('a', 'foo')
r.set_cookie('b', 'bar', {'path': '/blog'})
r.set_cookie('c', 'baz', {'secure': True, 'httponly': True})
r.set_cookie('d', 'qux', {'PaTh': '/blog', 'SeCuRe': True})
r.response()
# The mock is called with the following arguments.
#
# ('200 OK', [
# ('Set-Cookie', 'a=foo'),
# ('Set-Cookie', 'b=bar; Path=/blog'),
# ('Set-Cookie', 'c=baz; HttpOnly; Secure'),
# ('Set-Cookie', 'd=qux; Path=/blog; Secure')
# ('Content-Type', 'text/html; charset=UTF-8'),
# ('Content-Length', '0')
# ])
#
# However, "HttpOnly" and "Secure" may occur in lowercase prior
# to version 3.4.2. See the following URLs for more details.
#
# - https://docs.python.org/3.4/whatsnew/changelog.html
# - http://bugs.python.org/issue23250
# First item in call_args contains positional arguments and the
# second contains keyword arguments. We pick up the positional
# arguments and then within that we pick up the list of headers.
headers = m.call_args[0][1]
# Convert all cookies to lower case for case-insensitive
# matching.
cookies = []
for field, value in headers:
if field == 'Set-Cookie':
cookies.append(value.lower())
# Verify that expected cookies are present in the response.
self.assertIn('a=foo', cookies)
self.assertIn('b=bar; path=/blog', cookies)
self.assertIn('c=baz; httponly; secure', cookies)
self.assertIn('d=qux; path=/blog; secure', cookies)
def test_status_phrase(self):
r = ice.Response(mock.Mock())
r.status = 400
self.assertEqual(r.status_detail,
'Bad request syntax or unsupported method')
def test_status_line(self):
r = ice.Response(mock.Mock())
r.status = 400
self.assertEqual(r.status_line, '400 Bad Request')
def test_response_return_value_with_bytes_body(self):
r = ice.Response(mock.Mock())
r.body = b'foo'
self.assertEqual(r.response(), [b'foo'])
def test_response_return_value_with_no_body(self):
self.assertEqual(ice.Response(mock.Mock()).response(), [b''])