def test_wrap_response_str_err(self):
r = _wrap_response(400, 'fake_message', 'ERR_FAKE')
self.assertEqual({
'message': 'fake_message',
'errorCode': 'ERR_FAKE'
}, json.loads(r.body))
self.assertEqual(400, r.status_code)
def test_wrap_response_bytes_text(self):
r = _wrap_response(200,
b'fake_bytes_message',
content_type='text/plain')
self.assertEqual(b'fake_bytes_message', r.body)
self.assertEqual(200, r.status_code)
self.assertEqual('text/plain', r.media_type)
def test_wrap_response_obj(self):
fake_data = {'message': 'ok'}
r = _wrap_response(200, fake_data)
self.assertEqual(fake_data, json.loads(r.body))
self.assertEqual(200, r.status_code)
def test_wrap_response_str(self):
r = _wrap_response(200, 'fake_message')
self.assertEqual({'message': 'fake_message'}, json.loads(r.body))
self.assertEqual(200, r.status_code)