Browse Source

Tests passing

Ask Solem 12 years ago
parent
commit
d99d2b50a0
3 changed files with 8 additions and 10 deletions
  1. 0 1
      celery/backends/amqrpc.py
  2. 1 0
      celery/events/__init__.py
  3. 7 9
      celery/tests/backends/test_amqp.py

+ 0 - 1
celery/backends/amqrpc.py

@@ -13,7 +13,6 @@ import kombu
 from threading import local
 
 from kombu.common import maybe_declare, oid_from
-from kombu.utils import cached_property
 
 from celery import current_task
 from celery.backends import amqp

+ 1 - 0
celery/events/__init__.py

@@ -182,6 +182,7 @@ class EventReceiver(ConsumerMixin):
                            routing_key=self.routing_key,
                            auto_delete=True,
                            durable=False)
+        self.clock = self.app.clock
 
     def get_exchange(self):
         return get_exchange(self.connection)

+ 7 - 9
celery/tests/backends/test_amqp.py

@@ -5,6 +5,8 @@ import socket
 from datetime import timedelta
 from Queue import Empty, Queue
 
+from mock import patch
+
 from celery import current_app
 from celery import states
 from celery.app import app_or_default
@@ -246,15 +248,11 @@ class test_AMQPBackend(AppCase):
             next(b.get_many(['id1']))
 
     def test_test_get_many_raises_inner_block(self):
-
-        class Backend(AMQPBackend):
-
-            def drain_events(self, *args, **kwargs):
-                raise KeyError('foo')
-
-        b = Backend()
-        with self.assertRaises(KeyError):
-            next(b.get_many(['id1']))
+        with patch('kombu.connection.Connection.drain_events') as drain:
+            drain.side_effect = KeyError('foo')
+            b = AMQPBackend()
+            with self.assertRaises(KeyError):
+                next(b.get_many(['id1']))
 
     def test_no_expires(self):
         b = self.create_backend(expires=None)