Tests for blog, create, update and delete

Author: Alig1493

example/serializers.py

@@ -77,8 +77,9 @@ def get_root_meta(self, resource, many):
 
     class Meta:
         model = Blog
-        fields = ('name', 'url', 'tags')
+        fields = ('name', 'url', 'tags', 'copyright')
         read_only_fields = ('tags',)
+        meta_fields = ('copyright',)
 
 
 class EntrySerializer(serializers.ModelSerializer):

example/tests/unit/test_default_drf_serializers.py

@@ -1,5 +1,4 @@
 import pytest
-import factory
 
 import json
 
@@ -9,7 +8,7 @@
 
 from rest_framework_json_api.renderers import JSONRenderer
 
-from example.models import Comment, Entry
+from example.models import Comment, Entry, Blog
 
 
 # serializers
@@ -80,10 +79,51 @@ def test_render_format_keys(settings):
     assert result['data']['attributes']['json-field'] == {'json-key': 'JsonValue'}
 
 
+@pytest.mark.django_db
+def test_blog_create(client):
+
+    url = reverse('drf-entry-blog-list')
+    name = "Dummy Name"
+
+    request_data = {
+        'data': {
+            'attributes': {'name': name},
+            'type': 'blogs'
+        },
+    }
+
+    resp = client.post(url, request_data)
+
+    # look for created blog in database
+    blog = Blog.objects.filter(name=name)
+
+    # check if blog exists in database
+    assert blog.exists()
+    assert blog.count() == 1
+
+    # get created blog from database
+    blog = blog[0]
+
+    expected = {
+        'data': {
+            'attributes': {'name': blog.name},
+            'id': '{}'.format(blog.id),
+            'links': {'self': 'http://testserver/blogs/{}'.format(blog.id)},
+            'meta': {'copyright': 2018},
+            'relationships': {'tags': {'data': []}},
+            'type': 'blogs'
+        },
+        'meta': {'apiDocs': '/docs/api/blogs'}
+    }
+
+    assert resp.status_code == 201
+    assert resp.json() == expected
+
+
 @pytest.mark.django_db
 def test_get_object_gives_correct_blog(client, blog, entry):
 
-    url = reverse('drf-entry-blog', kwargs={'entry_pk': entry.id})
+    url = reverse('drf-entry-blog-detail', kwargs={'entry_pk': entry.id})
     resp = client.get(url)
     expected = {
         'data': {
@@ -100,26 +140,49 @@ def test_get_object_gives_correct_blog(client, blog, entry):
     assert got == expected
 
 
-# @pytest.mark.django_db
-# def test_get_object_updates_correct_blog(client, blog, entry):
-#
-#     url = reverse('drf-entry-blog', kwargs={'entry_pk': entry.id})
-#     new_name = blog.name + " update"
-#     assert not new_name == blog.name
-#
-#     resp = client.patch(url, {"name": new_name})
-#     print(resp)
-#
-#     expected = {
-#         'data': {
-#             'attributes': {'name': new_name},
-#             'id': '{}'.format(blog.id),
-#             'links': {'self': 'http://testserver/blogs/{}'.format(blog.id)},
-#             'meta': {'copyright': 2018},
-#             'relationships': {'tags': {'data': []}},
-#             'type': 'blogs'
-#         },
-#         'meta': {'apiDocs': '/docs/api/blogs'}
-#     }
-#     got = resp.json()
-#     assert got == expected
+@pytest.mark.django_db
+def test_get_object_patches_correct_blog(client, blog, entry):
+
+    url = reverse('drf-entry-blog-detail', kwargs={'entry_pk': entry.id})
+    new_name = blog.name + " update"
+    assert not new_name == blog.name
+
+    request_data = {
+        'data': {
+            'attributes': {'name': new_name},
+            'id': '{}'.format(blog.id),
+            'links': {'self': 'http://testserver/blogs/{}'.format(blog.id)},
+            'meta': {'copyright': 2018},
+            'relationships': {'tags': {'data': []}},
+            'type': 'blogs'
+        },
+        'meta': {'apiDocs': '/docs/api/blogs'}
+    }
+
+    resp = client.patch(url, data=request_data)
+
+    assert resp.status_code == 200
+
+    expected = {
+        'data': {
+            'attributes': {'name': new_name},
+            'id': '{}'.format(blog.id),
+            'links': {'self': 'http://testserver/blogs/{}'.format(blog.id)},
+            'meta': {'copyright': 2018},
+            'relationships': {'tags': {'data': []}},
+            'type': 'blogs'
+        },
+        'meta': {'apiDocs': '/docs/api/blogs'}
+    }
+    got = resp.json()
+    assert got == expected
+
+
+@pytest.mark.django_db
+def test_get_object_deletes_correct_blog(client, blog, entry):
+
+    url = reverse('drf-entry-blog-detail', kwargs={'entry_pk': entry.id})
+
+    resp = client.delete(url)
+
+    assert resp.status_code == 204

example/urls_test.py

@@ -23,7 +23,7 @@
 
 router.register(r'blogs', BlogViewSet)
 # router to test default DRF functionalities
-router.register(r'blogs', DRFBlogViewSet)
+router.register(r'drf-blogs', DRFBlogViewSet, 'drf-entry-blog')
 router.register(r'entries', EntryViewSet)
 # these "flavors" of entries are used for various tests:
 router.register(r'nopage-entries', NonPaginatedEntryViewSet, 'nopage-entry')
@@ -50,11 +50,6 @@
         BlogViewSet.as_view({'get': 'retrieve'}),
         name='entry-blog'
         ),
-    # get/patch url to test default DRF functionalities
-    url(r'^entries/(?P<entry_pk>[^/.]+)/blog',
-        DRFBlogViewSet.as_view({'get': 'retrieve'}),
-        name='drf-entry-blog'
-        ),
     url(r'^entries/(?P<entry_pk>[^/.]+)/comments',
         CommentViewSet.as_view({'get': 'list'}),
         name='entry-comments'

example/views.py

@@ -43,9 +43,10 @@ def get_object(self):
 class DRFBlogViewSet(viewsets.ModelViewSet):
     queryset = Blog.objects.all()
     serializer_class = BlogDRFSerializer
+    lookup_url_kwarg = 'entry_pk'
 
     def get_object(self):
-        entry_pk = self.kwargs.get('entry_pk', None)
+        entry_pk = self.kwargs.get(self.lookup_url_kwarg, None)
         if entry_pk is not None:
             return Entry.objects.get(id=entry_pk).blog