actions/black
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

close event loop for all tests (#1394)

Browse Source
This commit is contained in:
Zsolt Dollenstein 2020-05-08 16:27:50 +01:00 committed by GitHub
parent 362b722004
commit 7f4b275413
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_black.py
View File

@ -82,7 +82,7 @@ def cache_dir(exists: bool = True) -> Iterator[Path]:
@contextmanager
def event_loop(close: bool) -> Iterator[None]:
def event_loop() -> Iterator[None]:
policy = asyncio.get_event_loop_policy()
loop = policy.new_event_loop()
asyncio.set_event_loop(loop)
@ -90,8 +90,7 @@ def event_loop(close: bool) -> Iterator[None]:
yield
finally:
if close:
loop.close()
loop.close()
@contextmanager
@ -1260,7 +1259,7 @@ def test_cache_single_file_already_cached(self) -> None:
with src.open("r") as fobj:
self.assertEqual(fobj.read(), "print('hello')")
@event_loop(close=False)
@event_loop()
def test_cache_multiple_files(self) -> None:
mode = black.FileMode()
with cache_dir() as workspace, patch(
@ -1340,7 +1339,7 @@ def test_write_cache_creates_directory_if_needed(self) -> None:
black.write_cache({}, [], mode)
self.assertTrue(workspace.exists())
@event_loop(close=False)
@event_loop()
def test_failed_formatting_does_not_get_cached(self) -> None:
mode = black.FileMode()
with cache_dir() as workspace, patch(
@ -1363,7 +1362,7 @@ def test_write_cache_write_fail(self) -> None:
mock.side_effect = OSError
black.write_cache({}, [], mode)
@event_loop(close=False)
@event_loop()
@patch("black.ProcessPoolExecutor", MagicMock(side_effect=OSError))
def test_works_in_mono_process_only_environment(self) -> None:
with cache_dir() as workspace:
@ -1374,7 +1373,7 @@ def test_works_in_mono_process_only_environment(self) -> None:
f.write_text('print("hello")\n')
self.invokeBlack([str(workspace)])
@event_loop(close=False)
@event_loop()
def test_check_diff_use_together(self) -> None:
with cache_dir():
# Files which will be reformatted.
@ -1437,7 +1436,7 @@ def test_single_file_force_pyi(self) -> None:
self.assertNotIn(path, normal_cache)
self.assertEqual(actual, expected)
@event_loop(close=False)
@event_loop()
def test_multi_file_force_pyi(self) -> None:
reg_mode = black.FileMode()
pyi_mode = black.FileMode(is_pyi=True)
@ -1489,7 +1488,7 @@ def test_single_file_force_py36(self) -> None:
self.assertNotIn(path, normal_cache)
self.assertEqual(actual, expected)
@event_loop(close=False)
@event_loop()
def test_multi_file_force_py36(self) -> None:
reg_mode = black.FileMode()
py36_mode = black.FileMode(target_versions=black.PY36_VERSIONS)