Example #1
0
 def get_consumer(self, no_hub=False, **kwargs):
     consumer = Consumer(
         on_task_request=Mock(),
         init_callback=Mock(),
         pool=Mock(),
         app=self.app,
         timer=Mock(),
         controller=Mock(),
         hub=None if no_hub else Mock(),
         **kwargs
     )
     consumer.blueprint = Mock()
     consumer._restart_state = Mock()
     consumer.connection = Mock()
     consumer.connection_errors = (socket.error, OSError, )
     return consumer
Example #2
0
 def get_consumer(self, no_hub=False, **kwargs):
     consumer = Consumer(
         on_task_request=Mock(),
         init_callback=Mock(),
         pool=Mock(),
         app=self.app,
         timer=Mock(),
         controller=Mock(),
         hub=None if no_hub else Mock(),
         **kwargs
     )
     consumer.blueprint = Mock()
     consumer._restart_state = Mock()
     consumer.connection = _amqp_connection()
     consumer.connection_errors = (socket.error, OSError, )
     return consumer