1
0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-15 08:04:49 +01:00

testutils: when mock patching FakeStrictRedis, use a subclass with a shared pubsub (to match real redis)

This commit is contained in:
Ilya Kreymer 2016-04-02 21:33:39 -07:00
parent ddee9236c6
commit 6157cebcc9

View File

@ -41,12 +41,21 @@ class BaseTestClass(object):
pass
# ============================================================================
PUBSUBS = []
class FakeStrictRedisSharedPubSub(FakeStrictRedis):
def __init__(self, *args, **kwargs):
super(FakeStrictRedisSharedPubSub, self).__init__(*args, **kwargs)
self._pubsubs = PUBSUBS
# ============================================================================
class FakeRedisTests(object):
@classmethod
def setup_class(cls):
super(FakeRedisTests, cls).setup_class()
cls.redismock = patch('redis.StrictRedis', FakeStrictRedis)
cls.redismock = patch('redis.StrictRedis', FakeStrictRedisSharedPubSub)
cls.redismock.start()
@staticmethod