Remove click pin in diff-shades workflow (#2979)

[etc/vim.git] / tests / test_blackd.py

diff --git a/tests/test_blackd.py b/tests/test_blackd.py
index 37431fcad0000690964ade29403d26a5a056e803..6174c4538b94d7438d513383f9e6977885855ac6 100644 (file)
--- a/tests/test_blackd.py
+++ b/tests/test_blackd.py
@@ -1,4 +1,5 @@
 import re
 import re

+from typing import Any

 from unittest.mock import patch
 
 from click.testing import CliRunner
 from unittest.mock import patch
 
 from click.testing import CliRunner


@@ -8,12 +9,18 @@ from tests.util import read_data, DETERMINISTIC_HEADER
 
 try:
     import blackd
 
 try:
     import blackd

-    from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop
+    from aiohttp.test_utils import AioHTTPTestCase

     from aiohttp import web
     from aiohttp import web

+except ImportError as e:
+    raise RuntimeError("Please install Black with the 'd' extra") from e
+
+try:
+    from aiohttp.test_utils import unittest_run_loop

 except ImportError:
 except ImportError:

-    has_blackd_deps = False
-else:
-    has_blackd_deps = True
+    # unittest_run_loop is unnecessary and a no-op since aiohttp 3.8, and aiohttp 4
+    # removed it. To maintain compatibility we can make our own no-op decorator.
+    def unittest_run_loop(func: Any, *args: Any, **kwargs: Any) -> Any:
+        return func

 
 
 @pytest.mark.blackd
 
 
 @pytest.mark.blackd