Mock connection error

fakeredis.py

@@ -2231,11 +2231,13 @@ class FakePubSub(object):
     PATTERN_MESSAGE_TYPES = ['psubscribe', 'punsubscribe']
     LISTEN_DELAY = 0.1          # delay between listen loops (seconds)
 
-    def __init__(self, decode_responses=False, *args, **kwargs):
+    def __init__(self, connected=True, decode_responses=False, *args, **kwargs):
         self.channels = {}
         self.patterns = {}
         self._q = Queue()
         self.subscribed = False
+        self.connected = connected
+        _patch_responses(self, _check_conn)
         if decode_responses:
             _patch_responses(self, _make_decode_func)
         self._decode_responses = decode_responses

test_fakeredis.py

@@ -4403,5 +4403,42 @@ def test_hscan_iter(self):
             self.redis.hscan_iter('name')
 
 
+class TestPubSubConnected(unittest.TestCase):
+
+    def setUp(self):
+        self.redis = fakeredis.FakePubSub(connected=False)
+
+    def tearDown(self):
+        del self.redis
+
+    def test_basic_subscript(self):
+        with self.assertRaises(redis.ConnectionError):
+            self.redis.subscribe('logs')
+
+    def test_subscript_conn_lost(self):
+        self.redis.connected = True
+        self.redis.subscribe('logs')
+        self.redis.connected = False
+        with self.assertRaises(redis.ConnectionError):
+            self.redis.get_message()
+
+    def test_put_listen(self):
+        self.redis.connected = True
+        count = self.redis.put('logs', 'mymessage', 'subscribe')
+        self.assertEqual(count, 1, 'Message could should be 1')
+        self.redis.connected = False
+        with self.assertRaises(redis.ConnectionError):
+            self.redis.get_message()
+        self.redis.connected = True
+        msg = self.redis.get_message()
+        check = {
+            'type': 'subscribe',
+            'pattern': None,
+            'channel': b'logs',
+            'data': 'mymessage'
+        }
+        self.assertEqual(msg, check, 'Message was not published to channel')
+
+
 if __name__ == '__main__':
     unittest.main()