test_views.py

importjson

fromdjango.testimportRequestFactory
fromdjango.utilsimporttimezone
fromrest_framework.reverseimportreverse

fromrest_framework.testimportAPITestCase
fromrest_framework.testimportforce_authenticate

fromrest_framework_json_api.utilsimportformat_resource_type
fromexample.modelsimportBlog, Entry, Comment, Author

from .. importviews
from . importTestBase


classTestRelationshipView(APITestCase):
defsetUp(self):
self.author=Author.objects.create(name='Super powerful superhero', email='i.am@lost.com')
self.blog=Blog.objects.create(name='Some Blog', tagline="It's a blog")
self.other_blog=Blog.objects.create(name='Other blog', tagline="It's another blog")
self.first_entry=Entry.objects.create(
blog=self.blog,
headline='headline one',
body_text='body_text two',
pub_date=timezone.now(),
mod_date=timezone.now(),
n_comments=0,
n_pingbacks=0,
rating=3
 )
self.second_entry=Entry.objects.create(
blog=self.blog,
headline='headline two',
body_text='body_text one',
pub_date=timezone.now(),
mod_date=timezone.now(),
n_comments=0,
n_pingbacks=0,
rating=1
 )
self.first_comment=Comment.objects.create(entry=self.first_entry, body="This entry is cool", author=None)
self.second_comment=Comment.objects.create(
entry=self.second_entry,
body="This entry is not cool",
author=self.author
 )

deftest_get_entry_relationship_blog(self):
url=reverse('entry-relationships', kwargs={'pk': self.first_entry.id, 'related_field': 'blog'})
response=self.client.get(url)
expected_data= {'type': format_resource_type('Blog'), 'id': str(self.first_entry.blog.id)}

assertresponse.data==expected_data

deftest_get_entry_relationship_invalid_field(self):
response=self.client.get('/entries/{}/relationships/invalid_field'.format(self.first_entry.id))

assertresponse.status_code==404

deftest_get_blog_relationship_entry_set(self):
response=self.client.get('/blogs/{}/relationships/entry_set'.format(self.blog.id))
expected_data= [{'type': format_resource_type('Entry'), 'id': str(self.first_entry.id)},
 {'type': format_resource_type('Entry'), 'id': str(self.second_entry.id)}]

assertresponse.data==expected_data

deftest_put_entry_relationship_blog_returns_405(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
response=self.client.put(url, data={})
assertresponse.status_code==405

deftest_patch_invalid_entry_relationship_blog_returns_400(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
response=self.client.patch(url,
data=json.dumps({'data': {'invalid': ''}}),
content_type='application/vnd.api+json')
assertresponse.status_code==400

deftest_relationship_view_errors_format(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
response=self.client.patch(url,
data=json.dumps({'data': {'invalid': ''}}),
content_type='application/vnd.api+json')
assertresponse.status_code==400

result=json.loads(response.content.decode('utf-8'))

assert'data'notinresult
assert'errors'inresult

deftest_get_empty_to_one_relationship(self):
url='/comments/{}/relationships/author'.format(self.first_entry.id)
response=self.client.get(url)
expected_data=None

assertresponse.data==expected_data

deftest_get_to_many_relationship_self_link(self):
url='/authors/{}/relationships/comment_set'.format(self.author.id)

response=self.client.get(url)
expected_data= {
'links': {'self': 'http://testserver/authors/1/relationships/comment_set'},
'data': [{'id': str(self.second_comment.id), 'type': format_resource_type('Comment')}]
 }
assertjson.loads(response.content.decode('utf-8')) ==expected_data

deftest_patch_to_one_relationship(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
request_data= {
'data': {'type': format_resource_type('Blog'), 'id': str(self.other_blog.id)}
 }
response=self.client.patch(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()
assertresponse.data==request_data['data']

response=self.client.get(url)
assertresponse.data==request_data['data']

deftest_patch_one_to_many_relationship(self):
url='/blogs/{}/relationships/entry_set'.format(self.first_entry.id)
request_data= {
'data': [{'type': format_resource_type('Entry'), 'id': str(self.first_entry.id)}, ]
 }
response=self.client.patch(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()
assertresponse.data==request_data['data']

response=self.client.get(url)
assertresponse.data==request_data['data']

deftest_patch_many_to_many_relationship(self):
url='/entries/{}/relationships/authors'.format(self.first_entry.id)
request_data= {
'data': [
 {
'type': format_resource_type('Author'),
'id': str(self.author.id)
 },
 ]
 }
response=self.client.patch(url,
data=json.dumps(request_data),
content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()
assertresponse.data==request_data['data']

response=self.client.get(url)
assertresponse.data==request_data['data']

deftest_post_to_one_relationship_should_fail(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
request_data= {
'data': {'type': format_resource_type('Blog'), 'id': str(self.other_blog.id)}
 }
response=self.client.post(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==405, response.content.decode()

deftest_post_to_many_relationship_with_no_change(self):
url='/entries/{}/relationships/comment_set'.format(self.first_entry.id)
request_data= {
'data': [{'type': format_resource_type('Comment'), 'id': str(self.first_comment.id)}, ]
 }
response=self.client.post(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==204, response.content.decode()

deftest_post_to_many_relationship_with_change(self):
url='/entries/{}/relationships/comment_set'.format(self.first_entry.id)
request_data= {
'data': [{'type': format_resource_type('Comment'), 'id': str(self.second_comment.id)}, ]
 }
response=self.client.post(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()

assertrequest_data['data'][0] inresponse.data

deftest_delete_to_one_relationship_should_fail(self):
url='/entries/{}/relationships/blog'.format(self.first_entry.id)
request_data= {
'data': {'type': format_resource_type('Blog'), 'id': str(self.other_blog.id)}
 }
response=self.client.delete(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==405, response.content.decode()

deftest_delete_relationship_overriding_with_none(self):
url='/comments/{}'.format(self.second_comment.id)
request_data= {
'data': {
'type': 'comments',
'id': self.second_comment.id,
'relationships': {
'author': {
'data': None
 }
 }
 }
 }
response=self.client.patch(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()
assertresponse.data['author'] ==None

deftest_delete_to_many_relationship_with_no_change(self):
url='/entries/{}/relationships/comment_set'.format(self.first_entry.id)
request_data= {
'data': [{'type': format_resource_type('Comment'), 'id': str(self.second_comment.id)}, ]
 }
response=self.client.delete(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==204, response.content.decode()

deftest_delete_one_to_many_relationship_with_not_null_constraint(self):
url='/entries/{}/relationships/comment_set'.format(self.first_entry.id)
request_data= {
'data': [{'type': format_resource_type('Comment'), 'id': str(self.first_comment.id)}, ]
 }
response=self.client.delete(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==409, response.content.decode()

deftest_delete_to_many_relationship_with_change(self):
url='/authors/{}/relationships/comment_set'.format(self.author.id)
request_data= {
'data': [{'type': format_resource_type('Comment'), 'id': str(self.second_comment.id)}, ]
 }
response=self.client.delete(url, data=json.dumps(request_data), content_type='application/vnd.api+json')
assertresponse.status_code==200, response.content.decode()


classTestValidationErrorResponses(TestBase):
deftest_if_returns_error_on_empty_post(self):
view=views.BlogViewSet.as_view({'post': 'create'})
response=self._get_create_response("{}", view)
self.assertEqual(400, response.status_code)
expected= [{'detail': 'Received document does not contain primary data', 'status': '400', 'source': {'pointer': '/data'}}]
self.assertEqual(expected, response.data)

deftest_if_returns_error_on_missing_form_data_post(self):
view=views.BlogViewSet.as_view({'post': 'create'})
response=self._get_create_response('{"data":{"attributes":{},"type":"blogs"}}', view)
self.assertEqual(400, response.status_code)
expected= [{'status': '400', 'detail': 'This field is required.', 'source': {'pointer': '/data/attributes/name'}}]
self.assertEqual(expected, response.data)

deftest_if_returns_error_on_bad_endpoint_name(self):
view=views.BlogViewSet.as_view({'post': 'create'})
response=self._get_create_response('{"data":{"attributes":{},"type":"bad"}}', view)
self.assertEqual(409, response.status_code)
expected= [{'detail': "The resource object's type (bad) is not the type that constitute the collection represented by the endpoint (blogs).", 'source': {'pointer': '/data'}, 'status': '409'}]
self.assertEqual(expected, response.data)

def_get_create_response(self, data, view):
factory=RequestFactory()
request=factory.post('/', data, content_type='application/vnd.api+json')
user=self.create_user('user', 'pass')
force_authenticate(request, user)
returnview(request)