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:
parent
ddee9236c6
commit
6157cebcc9
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user