Use custom encoder (if provided) for all view key params not just keys (#501)

Author: mojito317

CHANGES.md

@@ -5,6 +5,7 @@
 - [FIXED] Fixed result paging for grouped view queries.
 - [FIXED] Incorrect use of username as account name in `Cloudant.bluemix()`.
 - [IMPROVED] Documented use of None account name and url override for `Cloudant.iam()`.
+- [FIXED] Use custom encoder (if provided) for all view `key` params not just `keys`.
 
 # 2.14.0 (2020-08-17)
 

src/cloudant/_common_util.py

@@ -160,7 +160,7 @@ def feed_arg_types(feed_type):
         return _COUCH_DB_UPDATES_ARG_TYPES
     return _CHANGES_ARG_TYPES
 
-def python_to_couch(options):
+def python_to_couch(options, encoder=None):
     """
     Translates query options from python style options into CouchDB/Cloudant
     query options.  For example ``{'include_docs': True}`` will
@@ -171,13 +171,14 @@ def python_to_couch(options):
     :func:`~cloudant.view.View.__call__` callable, both used to retrieve data.
 
     :param dict options: Python style parameters to be translated.
+    :param encoder: Custom encoder, defaults to None
 
     :returns: Dictionary of translated CouchDB/Cloudant query parameters
     """
     translation = dict()
     for key, val in iteritems_(options):
         py_to_couch_validate(key, val)
-        translation.update(_py_to_couch_translate(key, val))
+        translation.update(_py_to_couch_translate(key, val, encoder))
     return translation
 
 def py_to_couch_validate(key, val):
@@ -201,14 +202,16 @@ def py_to_couch_validate(key, val):
         if val not in ('ok', 'update_after'):
             raise CloudantArgumentError(135, val)
 
-def _py_to_couch_translate(key, val):
+def _py_to_couch_translate(key, val, encoder=None):
     """
     Performs the conversion of the Python parameter value to its CouchDB
     equivalent.
     """
     try:
         if key in ['keys', 'endkey_docid', 'startkey_docid', 'stale', 'update']:
             return {key: val}
+        if key in ['endkey', 'key', 'startkey']:
+            return {key: json.dumps(val, cls=encoder)}
         if val is None:
             return {key: None}
         arg_converter = TYPE_CONVERTERS.get(type(val))
@@ -249,7 +252,7 @@ def get_docs(r_session, url, encoder=None, headers=None, **params):
     keys = None
     if keys_list is not None:
         keys = json.dumps({'keys': keys_list}, cls=encoder)
-    f_params = python_to_couch(params)
+    f_params = python_to_couch(params, encoder)
     resp = None
     if keys is not None:
         # If we're using POST we are sending JSON so add the header

tests/unit/database_tests.py

@@ -512,6 +512,15 @@ def test_all_docs_get_with_long_type(self):
         data = self.db.all_docs(limit=1, skip=LONG_NUMBER)
         self.assertEqual(len(data.get('rows')), 1)
 
+    def test_all_docs_get_uses_custom_encoder(self):
+        """
+        Test that all_docs uses the custom encoder.
+        """
+        self.set_up_client(auto_connect=True, encoder="AEncoder")
+        database = self.client[self.test_dbname]
+        with self.assertRaises(CloudantArgumentError):
+            database.all_docs(endkey=['foo', 10])
+
     def test_custom_result_context_manager(self):
         """
         Test using the database custom result context manager

tests/unit/param_translation_tests.py

@@ -36,17 +36,17 @@ def test_valid_descending(self):
             {'descending': 'true'}
         )
         self.assertEqual(
-            python_to_couch({'descending': False}), 
+            python_to_couch({'descending': False}),
             {'descending': 'false'}
         )
 
     def test_valid_endkey(self):
         """
         Test endkey translation is successful.
         """
-        self.assertEqual(python_to_couch({'endkey': 10}), {'endkey': 10})
+        self.assertEqual(python_to_couch({'endkey': 10}), {'endkey': '10'})
         # Test with long type
-        self.assertEqual(python_to_couch({'endkey': LONG_NUMBER}), {'endkey': LONG_NUMBER})
+        self.assertEqual(python_to_couch({'endkey': LONG_NUMBER}), {'endkey': str(LONG_NUMBER)})
         self.assertEqual(
             python_to_couch({'endkey': 'foo'}),
             {'endkey': '"foo"'}
@@ -120,9 +120,9 @@ def test_valid_key(self):
         """
         Test key translation is successful.
         """
-        self.assertEqual(python_to_couch({'key': 10}), {'key': 10})
+        self.assertEqual(python_to_couch({'key': 10}), {'key': '10'})
         # Test with long type
-        self.assertEqual(python_to_couch({'key': LONG_NUMBER}), {'key': LONG_NUMBER})
+        self.assertEqual(python_to_couch({'key': LONG_NUMBER}), {'key': str(LONG_NUMBER)})
         self.assertEqual(python_to_couch({'key': 'foo'}), {'key': '"foo"'})
         self.assertEqual(
             python_to_couch({'key': ['foo', 10]}),
@@ -194,9 +194,9 @@ def test_valid_startkey(self):
         """
         Test startkey translation is successful.
         """
-        self.assertEqual(python_to_couch({'startkey': 10}), {'startkey': 10})
+        self.assertEqual(python_to_couch({'startkey': 10}), {'startkey': '10'})
         # Test with long type
-        self.assertEqual(python_to_couch({'startkey': LONG_NUMBER}), {'startkey': LONG_NUMBER})
+        self.assertEqual(python_to_couch({'startkey': LONG_NUMBER}), {'startkey': str(LONG_NUMBER)})
         self.assertEqual(
             python_to_couch({'startkey': 'foo'}),
             {'startkey': '"foo"'}