Fix tests on Python < 3.10.

Author: aaugustin

tests/asyncio/test_connection.py

@@ -44,6 +44,22 @@ async def asyncTearDown(self):
         await self.remote_connection.close()
         await self.connection.close()
 
+    if sys.version_info[:2] < (3, 10):  # pragma: no cover
+        @contextlib.contextmanager
+        def assertNoLogs(self, logger="websockets", level=logging.ERROR):
+            """
+            No message is logged on the given logger with at least the given level.
+
+            """
+            with self.assertLogs(logger, level) as logs:
+                # We want to test that no log message is emitted
+                # but assertLogs expects at least one log message.
+                logging.getLogger(logger).log(level, "dummy")
+                yield
+
+            level_name = logging.getLevelName(level)
+            self.assertEqual(logs.output, [f"{level_name}:{logger}:dummy"])
+
     # Test helpers built upon RecordingProtocol and InterceptingConnection.
 
     async def assertFrameSent(self, frame):

tests/legacy/utils.py

@@ -56,21 +56,21 @@ def run_loop_once(self):
         self.loop.call_soon(self.loop.stop)
         self.loop.run_forever()
 
-    # Remove when dropping Python < 3.10
-    @contextlib.contextmanager
-    def assertNoLogs(self, logger="websockets", level=logging.ERROR):
-        """
-        No message is logged on the given logger with at least the given level.
-
-        """
-        with self.assertLogs(logger, level) as logs:
-            # We want to test that no log message is emitted
-            # but assertLogs expects at least one log message.
-            logging.getLogger(logger).log(level, "dummy")
-            yield
-
-        level_name = logging.getLevelName(level)
-        self.assertEqual(logs.output, [f"{level_name}:{logger}:dummy"])
+    if sys.version_info[:2] < (3, 10):  # pragma: no cover
+        @contextlib.contextmanager
+        def assertNoLogs(self, logger="websockets", level=logging.ERROR):
+            """
+            No message is logged on the given logger with at least the given level.
+
+            """
+            with self.assertLogs(logger, level) as logs:
+                # We want to test that no log message is emitted
+                # but assertLogs expects at least one log message.
+                logging.getLogger(logger).log(level, "dummy")
+                yield
+
+            level_name = logging.getLevelName(level)
+            self.assertEqual(logs.output, [f"{level_name}:{logger}:dummy"])
 
     def assertDeprecationWarnings(self, recorded_warnings, expected_warnings):
         """