class TestResponseAttributes(unittest.TestCase):
"""
test if response set default attributes as intended
"""
def setUp(self):
self.environ = {
'CONTENT_LENGTH': '0',
'PATH_INFO': '',
'HTTP_COOKIE': '',
'REQUEST_METHOD': 'GET',
'wsgi.url_scheme': 'http',
'HTTP_HOST': 'localhost',
'SCRIPT_NAME': '',
'QUERY_STRING': ''
}
request = Request(self.environ)
self.response = Response(request)
def test_init_headers(self):
expected = '200 OK'
self.assertEqual(expected, self.response.status)
expected = [('Content-Type', 'text/html; charset=UTF-8')]
self.assertEqual(expected, self.response.header)
def test_init_body(self):
actual = self.response.body
expected = ''
self.assertEqual(actual, expected)
def test_change_body(self):
expected = ('test changing the response body', 'add another')
self.response.add(expected[0])
self.assertEqual(self.response.body, expected[0])
self.response.add(expected[1])
self.assertEqual(self.response.body, ''.join(expected))
def setUp(self):
self.environ = {
'CONTENT_LENGTH': '0',
'PATH_INFO': '',
'HTTP_COOKIE': '',
'REQUEST_METHOD': 'GET',
'wsgi.url_scheme': 'http',
'HTTP_HOST': 'localhost',
'SCRIPT_NAME': '',
'QUERY_STRING': ''
}
request = Request(self.environ)
self.response = Response(request)
