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(name='blueprint')
     consumer._restart_state = Mock(name='_restart_state')
     consumer.connection = _amqp_connection()
     consumer.connection_errors = (socket.error, OSError,)
     consumer.conninfo = consumer.connection
     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(name='blueprint')
     consumer._restart_state = Mock(name='_restart_state')
     consumer.connection = _amqp_connection()
     consumer.connection_errors = (socket.error, OSError,)
     consumer.conninfo = consumer.connection
     return consumer