def __init__(self, app, heartbeat=None, on_task_message=None, transport_driver_type=None): hub = Hub() ( self.obj, self.connection, self.consumer, self.blueprint, self.hub, self.qos, self.heartbeat, self.clock, ) = self.args = [Mock(name='obj'), Mock(name='connection'), Mock(name='consumer'), Mock(name='blueprint'), hub, Mock(name='qos'), heartbeat, Mock(name='clock')] self.connection.supports_heartbeats = True self.connection.get_heartbeat_interval.side_effect = ( lambda: self.heartbeat ) self.consumer.callbacks = [] self.obj.strategies = {} self.connection.connection_errors = (socket.error, ) if transport_driver_type: self.connection.transport.driver_type = transport_driver_type self.hub.readers = {} self.hub.writers = {} self.hub.consolidate = set() self.hub.timer = Mock(name='hub.timer') self.hub.timer._queue = [Mock()] self.hub.fire_timers = Mock(name='hub.fire_timers') self.hub.fire_timers.return_value = 1.7 self.hub.poller = Mock(name='hub.poller') self.hub.close = Mock(name='hub.close()') # asynloop calls hub.close self.Hub = self.hub self.blueprint.state = RUN # need this for create_task_handler _consumer = Consumer(Mock(), timer=Mock(), app=app) _consumer.on_task_message = on_task_message or [] self.obj.create_task_handler = _consumer.create_task_handler self.on_unknown_message = self.obj.on_unknown_message = Mock( name='on_unknown_message', ) _consumer.on_unknown_message = self.on_unknown_message self.on_unknown_task = self.obj.on_unknown_task = Mock( name='on_unknown_task', ) _consumer.on_unknown_task = self.on_unknown_task self.on_invalid_task = self.obj.on_invalid_task = Mock( name='on_invalid_task', ) _consumer.on_invalid_task = self.on_invalid_task _consumer.strategies = self.obj.strategies
def __init__(self, app, heartbeat=None, on_task_message=None, transport_driver_type=None): hub = Hub() ( self.obj, self.connection, self.consumer, self.blueprint, self.hub, self.qos, self.heartbeat, self.clock, ) = self.args = [Mock(name='obj'), Mock(name='connection'), Mock(name='consumer'), Mock(name='blueprint'), hub, Mock(name='qos'), heartbeat, Mock(name='clock')] self.connection.supports_heartbeats = True self.connection.get_heartbeat_interval.side_effect = ( lambda: self.heartbeat ) self.consumer.callbacks = [] self.obj.strategies = {} self.connection.connection_errors = (socket.error,) if transport_driver_type: self.connection.transport.driver_type = transport_driver_type self.hub.readers = {} self.hub.writers = {} self.hub.consolidate = set() self.hub.timer = Mock(name='hub.timer') self.hub.timer._queue = [Mock()] self.hub.fire_timers = Mock(name='hub.fire_timers') self.hub.fire_timers.return_value = 1.7 self.hub.poller = Mock(name='hub.poller') self.hub.close = Mock(name='hub.close()') # asynloop calls hub.close self.Hub = self.hub self.blueprint.state = RUN # need this for create_task_handler _consumer = Consumer(Mock(), timer=Mock(), app=app) _consumer.on_task_message = on_task_message or [] self.obj.create_task_handler = _consumer.create_task_handler self.on_unknown_message = self.obj.on_unknown_message = Mock( name='on_unknown_message', ) _consumer.on_unknown_message = self.on_unknown_message self.on_unknown_task = self.obj.on_unknown_task = Mock( name='on_unknown_task', ) _consumer.on_unknown_task = self.on_unknown_task self.on_invalid_task = self.obj.on_invalid_task = Mock( name='on_invalid_task', ) _consumer.on_invalid_task = self.on_invalid_task _consumer.strategies = self.obj.strategies
def __init__(self, app, heartbeat=None, on_task_message=None): hub = Hub() ( self.obj, self.connection, self.consumer, self.blueprint, self.hub, self.qos, self.heartbeat, self.clock, ) = self.args = [ Mock(name="obj"), Mock(name="connection"), Mock(name="consumer"), Mock(name="blueprint"), hub, Mock(name="qos"), heartbeat, Mock(name="clock"), ] self.connection.supports_heartbeats = True self.connection.get_heartbeat_interval.side_effect = lambda: self.heartbeat self.consumer.callbacks = [] self.obj.strategies = {} self.connection.connection_errors = (socket.error,) self.hub.readers = {} self.hub.writers = {} self.hub.consolidate = set() self.hub.timer = Mock(name="hub.timer") self.hub.timer._queue = [Mock()] self.hub.fire_timers = Mock(name="hub.fire_timers") self.hub.fire_timers.return_value = 1.7 self.hub.poller = Mock(name="hub.poller") self.hub.close = Mock(name="hub.close()") # asynloop calls hub.close self.Hub = self.hub self.blueprint.state = RUN # need this for create_task_handler _consumer = Consumer(Mock(), timer=Mock(), app=app) _consumer.on_task_message = on_task_message or [] self.obj.create_task_handler = _consumer.create_task_handler self.on_unknown_message = self.obj.on_unknown_message = Mock(name="on_unknown_message") _consumer.on_unknown_message = self.on_unknown_message self.on_unknown_task = self.obj.on_unknown_task = Mock(name="on_unknown_task") _consumer.on_unknown_task = self.on_unknown_task self.on_invalid_task = self.obj.on_invalid_task = Mock(name="on_invalid_task") _consumer.on_invalid_task = self.on_invalid_task _consumer.strategies = self.obj.strategies
def __init__(self, app, heartbeat=None, on_task=None): ( self.obj, self.connection, self.consumer, self.blueprint, self.hub, self.qos, self.heartbeat, self.clock, ) = self.args = [Mock(name='obj'), Mock(name='connection'), Mock(name='consumer'), Mock(name='blueprint'), Mock(name='Hub'), Mock(name='qos'), heartbeat, Mock(name='clock')] self.connection.supports_heartbeats = True self.consumer.callbacks = [] self.obj.strategies = {} self.connection.connection_errors = (socket.error, ) #hent = self.Hub.__enter__ = Mock(name='Hub.__enter__') #self.Hub.__exit__ = Mock(name='Hub.__exit__') #self.hub = hent.return_value = Mock(name='hub_context') self.hub.on_task = on_task or [] self.hub.readers = {} self.hub.writers = {} self.hub.consolidate = set() self.hub.fire_timers.return_value = 1.7 self.Hub = self.hub # need this for create_task_handler _consumer = Consumer(Mock(), timer=Mock(), app=app) self.obj.create_task_handler = _consumer.create_task_handler self.on_unknown_message = self.obj.on_unknown_message = Mock( name='on_unknown_message', ) _consumer.on_unknown_message = self.on_unknown_message self.on_unknown_task = self.obj.on_unknown_task = Mock( name='on_unknown_task', ) _consumer.on_unknown_task = self.on_unknown_task self.on_invalid_task = self.obj.on_invalid_task = Mock( name='on_invalid_task', ) _consumer.on_invalid_task = self.on_invalid_task _consumer.strategies = self.obj.strategies
def __init__(self, app, heartbeat=None, on_task=None): ( self.obj, self.connection, self.consumer, self.blueprint, self.hub, self.qos, self.heartbeat, self.clock, ) = self.args = [Mock(name='obj'), Mock(name='connection'), Mock(name='consumer'), Mock(name='blueprint'), Mock(name='Hub'), Mock(name='qos'), heartbeat, Mock(name='clock')] self.connection.supports_heartbeats = True self.consumer.callbacks = [] self.obj.strategies = {} self.connection.connection_errors = (socket.error, ) #hent = self.Hub.__enter__ = Mock(name='Hub.__enter__') #self.Hub.__exit__ = Mock(name='Hub.__exit__') #self.hub = hent.return_value = Mock(name='hub_context') self.hub.on_task = on_task or [] self.hub.readers = {} self.hub.writers = {} self.hub.fire_timers.return_value = 1.7 self.Hub = self.hub # need this for create_task_handler _consumer = Consumer(Mock(), timer=Mock(), app=app) self.obj.create_task_handler = _consumer.create_task_handler self.on_unknown_message = self.obj.on_unknown_message = Mock( name='on_unknown_message', ) _consumer.on_unknown_message = self.on_unknown_message self.on_unknown_task = self.obj.on_unknown_task = Mock( name='on_unknown_task', ) _consumer.on_unknown_task = self.on_unknown_task self.on_invalid_task = self.obj.on_invalid_task = Mock( name='on_invalid_task', ) _consumer.on_invalid_task = self.on_invalid_task _consumer.strategies = self.obj.strategies