add encoding in client functions (#232)

1 files changed, 6 insertions, 4 deletions

diff --git a/pymemcache/test/utils.py b/pymemcache/test/utils.py
index a30e98c..826e052 100644
--- a/pymemcache/test/utils.py
+++ b/pymemcache/test/utils.py
@@ -27,7 +27,8 @@ class MockMemcacheClient(object):
                  no_delay=False,
                  ignore_exc=False,
                  default_noreply=True,
-                 allow_unicode_keys=False):
+                 allow_unicode_keys=False,
+                 encoding='ascii'):
 
         self._contents = {}
 
@@ -41,6 +42,7 @@ class MockMemcacheClient(object):
         self.timeout = timeout
         self.no_delay = no_delay
         self.ignore_exc = ignore_exc
+        self.encoding = encoding
 
     def get(self, key, default=None):
         if not self.allow_unicode_keys:
@@ -80,15 +82,15 @@ class MockMemcacheClient(object):
             if isinstance(key, six.string_types):
                 try:
                     if isinstance(key, bytes):
-                        key = key.decode().encode('ascii')
+                        key = key.decode().encode()
                     else:
-                        key = key.encode('ascii')
+                        key = key.encode(self.encoding)
                 except (UnicodeEncodeError, UnicodeDecodeError):
                     raise MemcacheIllegalInputError
         if (isinstance(value, six.string_types) and
                 not isinstance(value, six.binary_type)):
             try:
-                value = value.encode('ascii')
+                value = value.encode(self.encoding)
             except (UnicodeEncodeError, UnicodeDecodeError):
                 raise MemcacheIllegalInputError