Example #1
0
    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)
Example #2
0
    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)
Example #3
0
    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)
Example #4
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)
Example #5
0
    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)
Example #6
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)