add more tests

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))

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})