This commit is contained in:
Jeff Balogh 2011-07-28 12:02:47 -07:00
Родитель ce506882b6
Коммит 60a30692cb
1 изменённых файлов: 7 добавлений и 0 удалений

Просмотреть файл

@ -64,6 +64,9 @@ class MockRedis(object):
def __init__(self): def __init__(self):
self.kv = StringDict() self.kv = StringDict()
def flushall(self):
self.kv.clear()
def pipeline(self, **kw): def pipeline(self, **kw):
return self return self
@ -109,6 +112,10 @@ class MockRedis(object):
if isinstance(v, set): if isinstance(v, set):
return v return v
def sinter(self, keys):
sets = [self.kv.get(key, set()) for key in keys]
return reduce(lambda x, y: x & y, sets)
def hmget(self, name, keys): def hmget(self, name, keys):
db = self.kv.get(name, StringDict()) db = self.kv.get(name, StringDict())
return [db.get(key) for key in keys] return [db.get(key) for key in keys]