mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-24 06:59:52 +01:00
testutils: when mock patching FakeStrictRedis, use a subclass with a shared pubsub (to match real redis)
This commit is contained in:
parent
ddee9236c6
commit
6157cebcc9
@ -41,12 +41,21 @@ class BaseTestClass(object):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
# ============================================================================
|
||||||
|
PUBSUBS = []
|
||||||
|
|
||||||
|
class FakeStrictRedisSharedPubSub(FakeStrictRedis):
|
||||||
|
def __init__(self, *args, **kwargs):
|
||||||
|
super(FakeStrictRedisSharedPubSub, self).__init__(*args, **kwargs)
|
||||||
|
self._pubsubs = PUBSUBS
|
||||||
|
|
||||||
|
|
||||||
# ============================================================================
|
# ============================================================================
|
||||||
class FakeRedisTests(object):
|
class FakeRedisTests(object):
|
||||||
@classmethod
|
@classmethod
|
||||||
def setup_class(cls):
|
def setup_class(cls):
|
||||||
super(FakeRedisTests, cls).setup_class()
|
super(FakeRedisTests, cls).setup_class()
|
||||||
cls.redismock = patch('redis.StrictRedis', FakeStrictRedis)
|
cls.redismock = patch('redis.StrictRedis', FakeStrictRedisSharedPubSub)
|
||||||
cls.redismock.start()
|
cls.redismock.start()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
|
Loading…
x
Reference in New Issue
Block a user