HH-102457 Handle exceptions in futures

frontik/futures.py

@@ -1,6 +1,6 @@
 import time
 import logging
-from functools import wraps
+from functools import wraps, partial
 
 from tornado.ioloop import IOLoop
 from tornado.concurrent import Future
@@ -102,8 +102,13 @@ def new_cb(*args, **kwargs):
 
         return new_cb
 
+    @staticmethod
+    def _handle_future(callback, future):
+        future.result()
+        callback()
+
     def add_future(self, future):
-        IOLoop.current().add_future(future, self.add_notification())
+        IOLoop.current().add_future(future, partial(self._handle_future, self.add_notification()))
         return future
 
     def get_finish_future(self):

tests/projects/test_app/pages/error_yield.py

@@ -0,0 +1,17 @@
+from tornado import gen
+
+import frontik.handler
+
+
+@gen.coroutine
+def some_async_function(handler):
+    yield handler.post_url(handler.request.host, handler.request.path)
+    return 1 / 0
+
+
+class Page(frontik.handler.PageHandler):
+    def get_page(self):
+        self.finish_group.add_future(some_async_function(self))
+
+    def post_page(self):
+        self.text = 'result'

tests/test_yield_errors.py

@@ -0,0 +1,9 @@
+import unittest
+
+from .instances import frontik_test_app
+
+
+class TestHandler(unittest.TestCase):
+    def test_error_in_yield(self):
+        response = frontik_test_app.get_page('error_yield')
+        self.assertEqual(response.status_code, 500)