Tests passing

celery · Oct 6, 2015 · 1e023aa · 1e023aa
1 parent 88e5be7
commit 1e023aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/kombu/tests/transport/test_redis.py b/kombu/tests/transport/test_redis.py
@@ -482,12 +482,12 @@ def test_receive(self):
         self.assertDictEqual(payload, {'hello': 'world'})
         self.assertEqual(queue, 'b')
 
-    def test_receive_raises(self):
+    def test_receive_raises_for_connection_error(self):
         self.channel._in_listen = True
         s = self.channel.subclient = Mock()
         s.parse_response.side_effect = KeyError('foo')
 
-        with self.assertRaises(redis.Empty):
+        with self.assertRaises(KeyError):
             self.channel._receive()
         self.assertFalse(self.channel._in_listen)
 
@@ -509,7 +509,7 @@ def test_brpop_read_raises(self):
         c = self.channel.client = Mock()
         c.parse_response.side_effect = KeyError('foo')
 
-        with self.assertRaises(redis.Empty):
+        with self.assertRaises(KeyError):
             self.channel._brpop_read()
 
         c.connection.disconnect.assert_called_with()