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)
@@ -631,6 +631,22 @@ class TestGetStatementFeed(APITestCase):
         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))
         response: Response = self.client.get(path="/contents/statements/feed/pagination/?page=2&size=2")