diff --git a/webapp/tests/test_feed.py b/webapp/tests/test_feed.py index a7d85ff..c329e22 100644 --- a/webapp/tests/test_feed.py +++ b/webapp/tests/test_feed.py @@ -1,6 +1,6 @@ from django.contrib.auth.models import User from rest_framework import status -from rest_framework.test import APITestCase +from rest_framework.test import APITestCase, APIRequestFactory from webapp.models import Feed from webapp.serializers import FeedSerializer @@ -27,7 +27,9 @@ class FeedTestCase(APITestCase): self.assertTrue(status.is_success(response.status_code)) feeds = Feed.objects.all() - serializer = FeedSerializer(feeds, many=True) + serializer = FeedSerializer(feeds, many=True, context={ + "request": APIRequestFactory().get(r"http://testserver/api/feed/") + }) self.assertEqual(response.data["results"], serializer.data) def test_post_feed(self):