Added a test for Client.generic() data coercion.

The smart_str() call (now force_bytes()) added in
e73838b6dd is otherwise untested.
This commit is contained in:
Tim Graham 2018-02-12 10:50:43 -05:00
parent b8a41a2872
commit a6fb81750a
2 changed files with 11 additions and 1 deletions

View File

@ -117,6 +117,13 @@ class ClientTest(TestCase):
self.assertTrue(mock_encoder.called) self.assertTrue(mock_encoder.called)
self.assertTrue(mock_encoding.encode.called) self.assertTrue(mock_encoding.encode.called)
def test_put(self):
response = self.client.put('/put_view/', {'foo': 'bar'})
self.assertEqual(response.status_code, 200)
self.assertEqual(response.templates[0].name, 'PUT Template')
self.assertEqual(response.context['data'], "{'foo': 'bar'}")
self.assertEqual(response.context['Content-Length'], 14)
def test_trace(self): def test_trace(self):
"""TRACE a view""" """TRACE a view"""
response = self.client.trace('/trace_view/') response = self.client.trace('/trace_view/')

View File

@ -53,7 +53,10 @@ def trace_view(request):
def put_view(request): def put_view(request):
if request.method == 'PUT': if request.method == 'PUT':
t = Template('Data received: {{ data }} is the body.', name='PUT Template') t = Template('Data received: {{ data }} is the body.', name='PUT Template')
c = Context({'data': request.body.decode()}) c = Context({
'Content-Length': request.META['CONTENT_LENGTH'],
'data': request.body.decode(),
})
else: else:
t = Template('Viewing GET page.', name='Empty GET Template') t = Template('Viewing GET page.', name='Empty GET Template')
c = Context() c = Context()