fix test for page size

CityOfZion · Feb 16, 2018 · b011b83 · b011b83
1 parent f3590b1
commit b011b83
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/neo/api/REST/test_notification_rest_api.py b/neo/api/REST/test_notification_rest_api.py
@@ -182,11 +182,19 @@ def test_pagination_for_addr_results(self):
         jsn = json.loads(res)
         self.assertEqual(jsn['total'], 1027)
         results = jsn['results']
-        self.assertEqual(len(results), 1000)
+        self.assertEqual(len(results), 500)
 
         mock_req = requestMock(path=b'/addr/AFmseVrdL9f9oyCzZefL9tG6UbvhPbdYzM?page=1')
         res = self.app.get_by_addr(mock_req, 'AFmseVrdL9f9oyCzZefL9tG6UbvhPbdYzM')
         jsn = json.loads(res)
         self.assertEqual(jsn['total'], 1027)
         results = jsn['results']
+        self.assertEqual(len(results), 500)
+
+
+        mock_req = requestMock(path=b'/addr/AFmseVrdL9f9oyCzZefL9tG6UbvhPbdYzM?page=2')
+        res = self.app.get_by_addr(mock_req, 'AFmseVrdL9f9oyCzZefL9tG6UbvhPbdYzM')
+        jsn = json.loads(res)
+        self.assertEqual(jsn['total'], 1027)
+        results = jsn['results']
         self.assertEqual(len(results), 27)