3613587536
memcached's ASCII protocol supports unicode keys, so lets support them as well. Since using unicode keys for memcache is uncommon and to preserve the previous behavior disable support by default.
166 lines
4.8 KiB
Python
166 lines
4.8 KiB
Python
"""
|
|
Useful testing utilities.
|
|
|
|
This module is considered public API.
|
|
|
|
"""
|
|
|
|
import time
|
|
|
|
import six
|
|
|
|
from pymemcache.exceptions import MemcacheIllegalInputError
|
|
|
|
|
|
class MockMemcacheClient(object):
|
|
"""
|
|
A (partial) in-memory mock for Clients.
|
|
|
|
"""
|
|
|
|
def __init__(self,
|
|
server=None,
|
|
serializer=None,
|
|
deserializer=None,
|
|
connect_timeout=None,
|
|
timeout=None,
|
|
no_delay=False,
|
|
ignore_exc=False,
|
|
default_noreply=True,
|
|
allow_unicode_keys=False):
|
|
|
|
self._contents = {}
|
|
|
|
self.serializer = serializer
|
|
self.deserializer = deserializer
|
|
self.allow_unicode_keys = allow_unicode_keys
|
|
|
|
# Unused, but present for interface compatibility
|
|
self.server = server
|
|
self.connect_timeout = connect_timeout
|
|
self.timeout = timeout
|
|
self.no_delay = no_delay
|
|
self.ignore_exc = ignore_exc
|
|
|
|
def get(self, key, default=None):
|
|
if not self.allow_unicode_keys:
|
|
if isinstance(key, six.text_type):
|
|
raise MemcacheIllegalInputError(key)
|
|
if isinstance(key, six.string_types):
|
|
try:
|
|
key = key.encode('ascii')
|
|
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
raise MemcacheIllegalInputError
|
|
|
|
if key not in self._contents:
|
|
return default
|
|
|
|
expire, value, flags = self._contents[key]
|
|
if expire and expire < time.time():
|
|
del self._contents[key]
|
|
return default
|
|
|
|
if self.deserializer:
|
|
return self.deserializer(key, value, flags)
|
|
return value
|
|
|
|
def get_many(self, keys):
|
|
out = {}
|
|
for key in keys:
|
|
value = self.get(key)
|
|
if value is not None:
|
|
out[key] = value
|
|
return out
|
|
|
|
get_multi = get_many
|
|
|
|
def set(self, key, value, expire=0, noreply=True):
|
|
if not self.allow_unicode_keys:
|
|
if isinstance(key, six.text_type):
|
|
raise MemcacheIllegalInputError(key)
|
|
if isinstance(key, six.string_types):
|
|
try:
|
|
key = key.encode('ascii')
|
|
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
raise MemcacheIllegalInputError
|
|
if isinstance(value, six.text_type):
|
|
raise MemcacheIllegalInputError(value)
|
|
if isinstance(value, six.string_types):
|
|
try:
|
|
value = value.encode('ascii')
|
|
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
raise MemcacheIllegalInputError
|
|
|
|
flags = 0
|
|
if self.serializer:
|
|
value, flags = self.serializer(key, value)
|
|
|
|
if expire:
|
|
expire += time.time()
|
|
|
|
self._contents[key] = expire, value, flags
|
|
return True
|
|
|
|
def set_many(self, values, expire=None, noreply=True):
|
|
for key, value in six.iteritems(values):
|
|
self.set(key, value, expire, noreply)
|
|
return True
|
|
|
|
set_multi = set_many
|
|
|
|
def incr(self, key, value, noreply=False):
|
|
current = self.get(key)
|
|
present = current is not None
|
|
if present:
|
|
self.set(key, current + value, noreply=noreply)
|
|
return None if noreply or not present else current + value
|
|
|
|
def decr(self, key, value, noreply=False):
|
|
current = self.get(key)
|
|
if current is None:
|
|
return
|
|
|
|
self.set(key, current - value, noreply=noreply)
|
|
return current - value
|
|
|
|
def add(self, key, value, expire=None, noreply=True):
|
|
current = self.get(key)
|
|
present = current is not None
|
|
if not present:
|
|
self.set(key, value, expire, noreply)
|
|
return not present
|
|
|
|
def delete(self, key, noreply=True):
|
|
current = self._contents.pop(key, None)
|
|
present = current is not None
|
|
return noreply or present
|
|
|
|
def delete_many(self, keys, noreply=True):
|
|
for key in keys:
|
|
self.delete(key, noreply)
|
|
return True
|
|
|
|
delete_multi = delete_many
|
|
|
|
def stats(self):
|
|
# I make no claim that these values make any sense, but the format
|
|
# of the output is the same as for pymemcache.client.Client.stats()
|
|
return {
|
|
"version": "MockMemcacheClient",
|
|
"rusage_user": 1.0,
|
|
"rusage_system": 1.0,
|
|
"hash_is_expanding": False,
|
|
"slab_reassign_running": False,
|
|
"inter": "in-memory",
|
|
"evictions": False,
|
|
"growth_factor": 1.0,
|
|
"stat_key_prefix": "",
|
|
"umask": 0o644,
|
|
"detail_enabled": False,
|
|
"cas_enabled": False,
|
|
"auth_enabled_sasl": False,
|
|
"maxconns_fast": False,
|
|
"slab_reassign": False,
|
|
"slab_automove": False,
|
|
}
|