Making sure the author API is properly filtered

theonion · Sep 23, 2014 · 7415633 · 7415633
1 parent 2a53bda
commit 7415633
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/bulbs/api/views.py b/bulbs/api/views.py
@@ -270,6 +270,8 @@ class AuthorViewSet(UncachedResponse, viewsets.ReadOnlyModelViewSet):
 
     serializer_class = UserSerializer
     model = User
+    filter_backends = (filters.SearchFilter, filters.OrderingFilter)
+    search_fields = ("first_name", "last_name", "username")
 
     def get_queryset(self):
         author_filter = getattr(settings, "BULBS_AUTHOR_FILTER", Q(is_staff=True))

diff --git a/tests/api/test_author_api.py b/tests/api/test_author_api.py
@@ -45,3 +45,9 @@ def test_author_search(self):
         author_endpoint = reverse("author-list")
         response = client.get(author_endpoint, content_type="application/json")
         self.assertEqual(len(response.data), 5)
+
+        response = client.get(author_endpoint, {"search": "chris"}, content_type="application/json")
+        self.assertEqual(len(response.data), 1)
+
+        response = client.get(author_endpoint, {"search": "ok"}, content_type="application/json")
+        self.assertEqual(len(response.data), 2)