Strict tests: fail on any unexpected call of loop.call_exception_handler

Author: 1st1

tests/test_base.py

@@ -370,6 +370,8 @@ async def foo():
         self.assertIn('test_debug_slow_task_callbacks', msg)
 
     def test_default_exc_handler_callback(self):
+        self.loop.set_exception_handler(None)
+
         self.loop._process_events = mock.Mock()
 
         def zero_error(fut):
@@ -399,6 +401,7 @@ def zero_error(fut):
                 exc_info=mock.ANY)
 
     def test_set_exc_handler_custom(self):
+        self.loop.set_exception_handler(None)
         logger = logging.getLogger('asyncio')
 
         def run_loop():

tests/test_tcp.py

@@ -1048,6 +1048,10 @@ async def start_server():
             client.stop()
 
     def test_create_connection_ssl_1(self):
+        if self.implementation == 'asyncio':
+            # Don't crash on asyncio errors
+            self.loop.set_exception_handler(None)
+
         CNT = 0
         TOTAL_CNT = 25
 

uvloop/_testbase.py

@@ -8,6 +8,7 @@
 import gc
 import logging
 import os
+import pprint
 import re
 import select
 import socket
@@ -71,11 +72,17 @@ def is_asyncio_loop(self):
     def run_loop_briefly(self, *, delay=0.01):
         self.loop.run_until_complete(asyncio.sleep(delay, loop=self.loop))
 
+    def loop_exception_handler(self, loop, context):
+        self.__unhandled_exceptions.append(context)
+
     def setUp(self):
         self.loop = self.new_loop()
         asyncio.set_event_loop(None)
         self._check_unclosed_resources_in_debug = True
 
+        self.loop.set_exception_handler(self.loop_exception_handler)
+        self.__unhandled_exceptions = []
+
         if hasattr(asyncio, '_get_running_loop'):
             # Disable `_get_running_loop`.
             self._get_running_loop = asyncio.events._get_running_loop
@@ -84,6 +91,12 @@ def setUp(self):
     def tearDown(self):
         self.loop.close()
 
+        if self.__unhandled_exceptions:
+            print('Unexpected calls to loop.call_exception_handler():')
+            pprint.pprint(self.__unhandled_exceptions)
+            self.fail('unexpected calls to loop.call_exception_handler()')
+            return
+
         if hasattr(asyncio, '_get_running_loop'):
             asyncio.events._get_running_loop = self._get_running_loop