def test_publish_callback_exc(self, metrics):
fut = Future()
fut.set_exception(Exception())
published_callback("topic", time.time(), fut)
metrics["PUBLISHED_MESSAGES"].labels.assert_called_with(
stream_id="topic", partition=-1, error="Exception")
metrics["PUBLISHED_MESSAGES"].labels().inc()
def _execute_http_client_method(self, host, uri, client_method, waited, callback):
if waited and (self.is_finished() or self.finish_group.is_finished()):
handler_logger.info(
'attempted to make waited http request to %s %s in finished handler, ignoring', host, uri
)
future = Future()
future.set_exception(AbortAsyncGroup())
return future
if waited and callable(callback):
callback = self.check_finished(callback)
future = client_method(callback)
if waited:
self.finish_group.add_future(future)
return future