From 9104ebe5aeb1474cb163c303e35d40542673d5e8 Mon Sep 17 00:00:00 2001 From: Zsolt Dollenstein Date: Fri, 8 May 2020 14:54:30 +0100 Subject: [PATCH 1/1] better test for mono executor --- tests/test_black.py | 32 +++++++++++--------------------- 1 file changed, 11 insertions(+), 21 deletions(-) diff --git a/tests/test_black.py b/tests/test_black.py index a0e57dc..8d424ef 100644 --- a/tests/test_black.py +++ b/tests/test_black.py @@ -1273,27 +1273,6 @@ class BlackTestCase(unittest.TestCase): self.assertIn(one, cache) self.assertIn(two, cache) - @patch("black.ProcessPoolExecutor", autospec=True) - def test_works_in_mono_process_only_environment(self, mock_executor) -> None: - mock_executor.side_effect = OSError() - mode = black.FileMode() - with cache_dir() as workspace: - one = (workspace / "one.py").resolve() - with one.open("w") as fobj: - fobj.write("print('hello')") - two = (workspace / "two.py").resolve() - with two.open("w") as fobj: - fobj.write("print('hello')") - black.write_cache({}, [one], mode) - self.invokeBlack([str(workspace)]) - with one.open("r") as fobj: - self.assertEqual(fobj.read(), "print('hello')") - with two.open("r") as fobj: - self.assertEqual(fobj.read(), 'print("hello")\n') - cache = black.read_cache(mode) - self.assertIn(one, cache) - self.assertIn(two, cache) - def test_no_cache_when_writeback_diff(self) -> None: mode = black.FileMode() with cache_dir() as workspace: @@ -1375,6 +1354,17 @@ class BlackTestCase(unittest.TestCase): 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(): -- 2.39.5