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