Adding a bunch of Unit Tests

This commit is contained in:
Raphael Michel
2014-10-07 17:12:06 +02:00
parent d633f4317c
commit d004d7c49a
12 changed files with 227 additions and 19 deletions

View File

@@ -38,6 +38,9 @@ class EventRelatedCache:
key = hashlib.sha256(key.encode("UTF-8")).hexdigest()
return key
def _strip_prefix(self, key):
return key.split(":", maxsplit=3)[-1] if 'event:' in key else key
def clear(self):
try:
prefix = self.cache.incr(self.prefixkey, 1)
@@ -52,25 +55,29 @@ class EventRelatedCache:
return self.cache.get(self._prefix_key(key))
def get_many(self, keys):
return self.cache.get_many([self._prefix_key(key) for key in keys])
values = self.cache.get_many([self._prefix_key(key) for key in keys])
newvalues = {}
for k, v in values.items():
newvalues[self._strip_prefix(k)] = v
return newvalues
def set_many(self, values, timeout=3600):
newvalues = {}
for i in values.items():
newvalues[self._prefix_key(i[0])] = i[1]
return self.cache.set_many([newvalues], timeout)
for k, v in values.items():
newvalues[self._prefix_key(k)] = v
return self.cache.set_many(newvalues, timeout)
def delete(self, key):
def delete(self, key): # NOQA
return self.cache.delete(self._prefix_key(key))
def delete_many(self, keys):
def delete_many(self, keys): # NOQA
return self.cache.delete_many([self._prefix_key(key) for key in keys])
def incr(self, key, by=1):
def incr(self, key, by=1): # NOQA
return self.cache.incr(self._prefix_key(key), by)
def decr(self, key, by=1):
def decr(self, key, by=1): # NOQA
return self.cache.decr(self._prefix_key(key), by)
def close(self):
def close(self): # NOQA
pass