From 4809e365d43ff5f837f9dacbc1f3d04975fe35e8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jim=20Br=C3=A4nnlund?= Date: Thu, 29 Nov 2018 22:43:22 +0100 Subject: [PATCH] Silence expected stderr (#621) * Silence expected stderr output during test * Change based on PR comment --- tests/test_black.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/tests/test_black.py b/tests/test_black.py index 30bb6fdd73..1d759f2537 100644 --- a/tests/test_black.py +++ b/tests/test_black.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 import asyncio from concurrent.futures import ThreadPoolExecutor -from contextlib import contextmanager +from contextlib import contextmanager, redirect_stderr from functools import partial, wraps from io import BytesIO, TextIOWrapper import os @@ -1454,15 +1454,16 @@ async def test_blackd_py36(self) -> None: @unittest.skipUnless(has_blackd_deps, "blackd's dependencies are not installed") @async_test async def test_blackd_fast(self) -> None: - app = blackd.make_app() - async with TestClient(TestServer(app)) as client: - response = await client.post("/", data=b"ur'hello'") - self.assertEqual(response.status, 500) - self.assertIn("failed to parse source file", await response.text()) - response = await client.post( - "/", data=b"ur'hello'", headers={blackd.FAST_OR_SAFE_HEADER: "fast"} - ) - self.assertEqual(response.status, 200) + with open(os.devnull, "w") as dn, redirect_stderr(dn): + app = blackd.make_app() + async with TestClient(TestServer(app)) as client: + response = await client.post("/", data=b"ur'hello'") + self.assertEqual(response.status, 500) + self.assertIn("failed to parse source file", await response.text()) + response = await client.post( + "/", data=b"ur'hello'", headers={blackd.FAST_OR_SAFE_HEADER: "fast"} + ) + self.assertEqual(response.status, 200) @unittest.skipUnless(has_blackd_deps, "blackd's dependencies are not installed") @async_test