def test_ack(self): def wrapped_f(*args, **kwargs): pass f = handler.callback_wrapper(wrapped_f) f(mock.Mock(), self.message, "search.index") self.message.channel.basic_ack.assert_called_once_with(self.delivery_tag)
def test_ack(self): def wrapped_f(*args, **kwargs): pass f = handler.callback_wrapper(wrapped_f) _handler = handler.Handler() f(_handler, self.message, "search.index") # Process messages to ack _handler.process_messages() self.message.channel.basic_ack.assert_called_once_with(self.delivery_tag)
def test_search_failed_on_mb_retries_zero(self): def wrapped_f(*args, **kwargs): raise ValueError() self.message.application_headers["mb-retries"] = 0 f = handler.callback_wrapper(wrapped_f) f(self.handler, self.message, "search.index") self.channel.basic_reject.assert_called_once_with(self.delivery_tag, requeue=False) self.channel.basic_publish.assert_called_once_with( self.message, exchange="search.failed", routing_key=self.routing_key) self.assertEqual(self.message.application_headers["mb-retries"], 0)
def test_reject_on_exception(self): def wrapped_f(*args, **kwargs): raise ValueError() f = handler.callback_wrapper(wrapped_f) f(mock.Mock(), self.message, "search.index") self.message.channel.basic_reject.assert_called_once_with( self.delivery_tag, requeue=False) self.message.channel.basic_publish.assert_called_once_with( self.message, exchange="search.retry", routing_key=self.routing_key) self.assertEqual(self.message.application_headers["mb-retries"], handler._DEFAULT_MB_RETRIES - 1) self.assertFalse(self.message.channel.basic_ack.called)
def test_search_failed_on_mb_retries_zero(self): def wrapped_f(*args, **kwargs): raise ValueError() self.message.application_headers["mb-retries"] = 0 f = handler.callback_wrapper(wrapped_f) f(mock.Mock(), self.message, "search.index") self.message.channel.basic_reject.assert_called_once_with( self.delivery_tag, requeue=False) self.message.channel.basic_publish.assert_called_once_with( self.message, exchange="search.failed", routing_key=self.routing_key) self.assertEqual( self.message.application_headers["mb-retries"], 0)
def test_reject_on_exception(self): def wrapped_f(*args, **kwargs): raise ValueError() f = handler.callback_wrapper(wrapped_f) f(mock.Mock(), self.message, "search.index") self.message.channel.basic_reject.assert_called_once_with( self.delivery_tag, requeue=False) self.message.channel.basic_publish.assert_called_once_with( self.message, exchange="search.retry", routing_key=self.routing_key) self.assertEqual( self.message.application_headers["mb-retries"], handler._DEFAULT_MB_RETRIES - 1) self.assertFalse(self.message.channel.basic_ack.called)