diff --git a/webapp/tests.py b/webapp/tests.py index 1d0c016..68d2a43 100644 --- a/webapp/tests.py +++ b/webapp/tests.py @@ -24,7 +24,7 @@ class TagsTestCase(APITestCase): self.assertTrue(status.is_success(response.status_code)) # We dont care about icon, so response is sliced - sliced_response = OrderedDict(islice(response.data[0].items(), 3)) + sliced_response = OrderedDict(islice(response.data['results'][0].items(), 3)) tag1 = Tag.objects.get(slug="Party") self.assertEqual(sliced_response, {'id': tag1.id, 'slug': 'Party', 'name': 'Bileet'}) @@ -39,12 +39,12 @@ class TagsTestCase(APITestCase): # We dont care about icon, so response is sliced tag1 = Tag.objects.get(slug="Party") - sliced_response = OrderedDict(islice(response.data[0].items(), 3)) + sliced_response = OrderedDict(islice(response.data['results'][0].items(), 3)) self.assertEqual(sliced_response, {'id': tag1.id, 'slug': 'Party', 'name': 'Bileet'}) - sliced_response = OrderedDict(islice(response.data[1].items(), 3)) + sliced_response = OrderedDict(islice(response.data['results'][1].items(), 3)) tag2 = Tag.objects.get(slug="Freshmen") self.assertEqual(sliced_response, {'id': tag2.id, 'slug': 'Freshmen', 'name': 'Fuksit'}) - sliced_response = OrderedDict(islice(response.data[2].items(), 3)) + sliced_response = OrderedDict(islice(response.data['results'][2].items(), 3)) tag3 = Tag.objects.get(slug="International") self.assertEqual(sliced_response, {'id': tag3.id, 'slug': 'International', 'name': 'Ulkkarit'}) @@ -83,7 +83,7 @@ class FeedTestCase(APITestCase): feeds = Feed.objects.all() serializer = FeedSerializer(feeds, many=True) - self.assertEqual(response.data, serializer.data) + self.assertEqual(response.data['results'], serializer.data) def test_post_feed(self): Tag.objects.create(slug="test1", name="testsds")