Merge pull request #487 from cloudant/456-key-error-grouped-view

Fix result paging for grouped view queries

Author: ricellis

CHANGES.md

@@ -2,6 +2,7 @@
 - [REMOVED] Removed Python 2 compatibility from the supported environments.
 - [FIXED] Fixed the documentation for `bookmarks`.
 - [FIXED] Also exit `follow_replication` for `failed` state.
+- [FIXED] Fixed result paging for grouped view queries.
 
 # 2.14.0 (2020-08-17)
 

src/cloudant/result.py

@@ -397,9 +397,10 @@ def _iterator(self, response):
 
                 # if we are in a view, keys could be duplicate so we
                 # need to start from the right docid
-                if last['id']:
+                last_doc_id = last.get('id')
+                if last_doc_id is not None:
                     response = self._call(startkey=last['key'],
-                                          startkey_docid=last['id'])
+                                          startkey_docid=last_doc_id)
                 # reduce result keys are unique by definition
                 else:
                     response = self._call(startkey=last['key'])

tests/unit/database_tests.py

@@ -408,6 +408,34 @@ def test_retrieve_view_results(self):
         self.assertIsInstance(rslt, Result)
         self.assertEqual(rslt[:1], rslt['julia099'])
 
+    def test_retrieve_grouped_view_result_with_page_size(self):
+        """
+        Test retrieving Result wrapped output from a design document grouped view
+        that uses a custom page size
+
+        The view used here along with group=True will generate rows of
+        data where each key will be grouped into groups of 2.  Such as:
+        {'key': 0, 'value': 2},
+        {'key': 1, 'value': 2},
+        ...
+        """
+        map_func = 'function(doc) {\n emit(Math.floor(doc.age / 2), 1); \n}'
+        data = {'_id': '_design/ddoc01','views': {'view01': {"map": map_func, "reduce": "_count"}}}
+        self.db.create_document(data)
+        self.populate_db_with_documents(5)
+
+        rslt = self.db.get_view_result(
+            '_design/ddoc01',
+            'view01',
+            group=True,
+            page_size=1)
+        self.assertIsInstance(rslt, Result)
+        i = 0
+        for row in rslt:
+            self.assertIsNotNone(row)
+            self.assertEqual(row['key'], i)
+            i += 1
+
     def test_retrieve_raw_view_results(self):
         """
         Test retrieving raw output from a design document view