Adding select_related and prefetch_for_includes to CommentViewSet

Author: jsenecal

example/tests/test_performance.py

@@ -48,8 +48,9 @@ def test_query_count_include_author(self):
 
         1. Primary resource COUNT query
         2. Primary resource SELECT
-        3. Author's prefetched
+        3. Authors prefetched
+        3. Entries prefetched
         """
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             response = self.client.get('/comments?include=author&page_size=25')
             self.assertEqual(len(response.data['results']), 25)

example/views.py

@@ -73,8 +73,13 @@ class AuthorViewSet(ModelViewSet):
 
 
 class CommentViewSet(ModelViewSet):
-    queryset = Comment.objects.all()
+    queryset = Comment.objects.select_related('author', 'entry')
     serializer_class = CommentSerializer
+    prefetch_for_includes = {
+        '__all__': [],
+        'author': ['author', 'author__bio', 'author__entries'],
+        'entry': ['author', 'author__bio', 'author__entries']
+    }
 
 
 class CompanyViewset(ModelViewSet):