Add all() function to the ResultSet API

Author: aboudreault

CHANGELOG.rst

@@ -1,3 +1,11 @@
+3.22.0
+======
+UNRELEASED
+
+Features
+--------
+* Add all() function to the ResultSet API (PYTHON-1203)
+
 3.21.0
 ======
 January 15, 2020
@@ -31,6 +39,7 @@ Others
 * Remove *read_repair_chance table options (PYTHON-1140)
 * Avoid warnings about unspecified load balancing policy when connecting to a cloud cluster (PYTHON-1177)
 * Add new DSE CQL keywords (PYTHON-1122)
+* Publish binary wheel distributions (PYTHON-1013)
 
 Deprecations
 ------------

cassandra/cluster.py

@@ -4934,6 +4934,15 @@ def current_rows(self):
         """
         return self._current_rows or []
 
+    def all(self):
+        """
+        Returns all the remaining rows as a list. This is basically
+        a convenient shortcut to `list(result_set)`.
+
+        This function is not recommended for queries that return a large number of elements.
+        """
+        return list(self)
+
     def one(self):
         """
         Return a single row of the results or None if empty. This is basically

tests/unit/test_resultset.py

@@ -195,6 +195,13 @@ def test_one(self):
 
         self.assertEqual(rs.one(), first)
 
+    def test_all(self):
+        first, second = Mock(), Mock()
+        rs1 = ResultSet(Mock(has_more_pages=False), [first, second])
+        rs2 = ResultSet(Mock(has_more_pages=False), [first, second])
+
+        self.assertEqual(rs1.all(), list(rs2))
+
     @patch('cassandra.cluster.warn')
     def test_indexing_deprecation(self, mocked_warn):
         # normally we'd use catch_warnings to test this, but that doesn't work