Skip to content

Commit

Permalink
better test for mono executor
Browse files Browse the repository at this point in the history
  • Loading branch information
zsol committed May 8, 2020
1 parent ede23ae commit 9104ebe
Showing 1 changed file with 11 additions and 21 deletions.
32 changes: 11 additions & 21 deletions tests/test_black.py
Original file line number Diff line number Diff line change
Expand Up @@ -1273,27 +1273,6 @@ def test_cache_multiple_files(self) -> None:
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:
Expand Down Expand Up @@ -1375,6 +1354,17 @@ def test_write_cache_write_fail(self) -> None:
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():
Expand Down

0 comments on commit 9104ebe

Please sign in to comment.