test_format_keys.py

fromdjango.contrib.authimportget_user_model
fromdjango.core.urlresolversimportreverse
fromdjango.utilsimportencoding

fromexample.testsimportTestBase
fromexample.tests.utilsimportload_json


classFormatKeysSetTests(TestBase):
"""
 Test that camelization and underscoring of key names works if they are activated.
 """
list_url=reverse('user-list')

defsetUp(self):
super(FormatKeysSetTests, self).setUp()
self.detail_url=reverse('user-detail', kwargs={'pk': self.miles.pk})


deftest_camelization(self):
"""
 Test that camelization works.
 """
response=self.client.get(self.list_url)
self.assertEqual(response.status_code, 200)

user=get_user_model().objects.all()[0]
expected= {
'data': [
 {
'type': 'users',
'id': encoding.force_text(user.pk),
'attributes': {
'firstName': user.first_name,
'lastName': user.last_name,
'email': user.email
 },
 }
 ],
'links': {
'first': 'http://testserver/identities?page=1',
'last': 'http://testserver/identities?page=2',
'next': 'http://testserver/identities?page=2',
'prev': None
 },
'meta': {
'pagination': {
'page': 1,
'pages': 2,
'count': 2
 }
 }
 }

parsed_content=load_json(response.content)

assertexpected==parsed_content