From eb11b3c131789ebce5b4b3b0cc760ee03370a2d8 Mon Sep 17 00:00:00 2001 From: Andreas Burbach <aburbach@compeon.de> Date: Tue, 7 Dec 2021 11:23:53 +0100 Subject: [PATCH] add more tests --- social_network/the_social_network/tests.py | 18 +++++++++++++++++- .../the_social_network/views/contentViews.py | 2 +- 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/social_network/the_social_network/tests.py b/social_network/the_social_network/tests.py index b3a3f66..0bea209 100644 --- a/social_network/the_social_network/tests.py +++ b/social_network/the_social_network/tests.py @@ -618,7 +618,7 @@ class TestGetStatementFeed(APITestCase): def test_feed_contains_correct_data_pagination(self): self.client.credentials(HTTP_AUTHORIZATION='Token ' + str(self.token_beate)) - response: Response = self.client.get(path="/contents/statements/feed/pagination/?page=1&size=5") + response: Response = self.client.get(path="/contents/statements/feed/pagination/?page=1&size=3") result = response.data["data"] result_total = response.data["total"] self.assertTrue(len(result), 3) @@ -630,6 +630,22 @@ class TestGetStatementFeed(APITestCase): self.assertTrue(result[0].get("created") > result[1].get("created")) self.assertTrue(result[1].get("id") > result[2].get("id")) self.assertTrue(result[1].get("created") > result[2].get("created")) + + def test_feed_load_over_size_pagination(self): + self.client.credentials(HTTP_AUTHORIZATION='Token ' + str(self.token_beate)) + response: Response = self.client.get(path="/contents/statements/feed/pagination/?page=1&size=4") + result = response.data["data"] + result_total = response.data["total"] + self.assertTrue(len(result), 3) + self.assertTrue(result_total, 3) + + def test_feed_load_under_size_pagination(self): + self.client.credentials(HTTP_AUTHORIZATION='Token ' + str(self.token_beate)) + response: Response = self.client.get(path="/contents/statements/feed/pagination/?page=1&size=2") + result = response.data["data"] + result_total = response.data["total"] + self.assertTrue(len(result), 2) + self.assertTrue(result_total, 3) def test_feed_contains_no_data_pagination(self): self.client.credentials(HTTP_AUTHORIZATION='Token ' + str(self.token_beate)) diff --git a/social_network/the_social_network/views/contentViews.py b/social_network/the_social_network/views/contentViews.py index 0af4708..9347182 100644 --- a/social_network/the_social_network/views/contentViews.py +++ b/social_network/the_social_network/views/contentViews.py @@ -127,7 +127,7 @@ class ShowStatementFeedPagination(APIView): else: if last >= total: last = None - feed = feed[first:last] + feed = feed[first:last] serializer: StatementSerializer = StatementSerializer(instance=feed, many=True) return Response(status=status.HTTP_200_OK, data={"data": serializer.data, "total": total}) -- GitLab