From 6157cebcc99c012a1bd5f9358d8b4e026e4b65a3 Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Sat, 2 Apr 2016 21:33:39 -0700 Subject: [PATCH] testutils: when mock patching FakeStrictRedis, use a subclass with a shared pubsub (to match real redis) --- webagg/test/testutils.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/webagg/test/testutils.py b/webagg/test/testutils.py index a5d8677d..d0fb361e 100644 --- a/webagg/test/testutils.py +++ b/webagg/test/testutils.py @@ -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