def test__wrap_callback_errors_error(): msg = mock.create_autospec(message.Message, instance=True) callback = mock.Mock(side_effect=ValueError('meep')) streaming_pull_manager._wrap_callback_errors(callback, msg) msg.nack.assert_called_once()
def test__wrap_callback_errors_no_error(): msg = mock.create_autospec(message.Message, instance=True) callback = mock.Mock() streaming_pull_manager._wrap_callback_errors(callback, msg) callback.assert_called_once_with(msg) msg.nack.assert_not_called()
def test__wrap_callback_errors_no_error(): msg = mock.create_autospec(message.Message, instance=True) callback = mock.Mock() on_callback_error = mock.Mock() streaming_pull_manager._wrap_callback_errors(callback, on_callback_error, msg) callback.assert_called_once_with(msg) msg.nack.assert_not_called() on_callback_error.assert_not_called()
def test__wrap_callback_errors_error(): callback_error = ValueError("meep") msg = mock.create_autospec(message.Message, instance=True) callback = mock.Mock(side_effect=callback_error) on_callback_error = mock.Mock() streaming_pull_manager._wrap_callback_errors(callback, on_callback_error, msg) msg.nack.assert_called_once() on_callback_error.assert_called_once_with(callback_error)
def test__wrap_callback_errors_error(): callback_error = ValueError("meep") msg = mock.create_autospec(message.Message, instance=True) callback = mock.Mock(side_effect=callback_error) on_callback_error = mock.Mock() streaming_pull_manager._wrap_callback_errors(callback, on_callback_error, msg) msg.nack.assert_called_once() on_callback_error.assert_called_once_with(callback_error)