Make the use of a ThreadPoolExecutor explicit

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

diff --git a/tests/test_black.py b/tests/test_black.py
index 11bba1fea0759089a6caf9c9d56000c765362b9e..8d424ef37256d11354ab1b9307468f2e160b5389 100644 (file)
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -1354,6 +1354,17 @@ class BlackTestCase(unittest.TestCase):
             mock.side_effect = OSError
             black.write_cache({}, [], mode)
 
             mock.side_effect = OSError
             black.write_cache({}, [], mode)
 

+    @patch("black.ProcessPoolExecutor", autospec=True)
+    def test_works_in_mono_process_only_environment(self, executor: MagicMock) -> None:
+        executor.side_effect = OSError()
+        with cache_dir() as workspace:
+            for f in [
+                (workspace / "one.py").resolve(),
+                (workspace / "two.py").resolve(),
+            ]:
+                f.write_text("print('hello')")
+            self.invokeBlack([str(workspace)])
+

     @event_loop(close=False)
     def test_check_diff_use_together(self) -> None:
         with cache_dir():
     @event_loop(close=False)
     def test_check_diff_use_together(self) -> None:
         with cache_dir():