From 30bc83f1c96c2982b0a13e929c17dfa26f74f2d8 Mon Sep 17 00:00:00 2001 From: Kevin Tindall Date: Sun, 2 Jan 2022 21:12:01 -0600 Subject: [PATCH] Run formatter --- mcstatus/tests/test_timeout.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/mcstatus/tests/test_timeout.py b/mcstatus/tests/test_timeout.py index 1c1b50e..69e6835 100644 --- a/mcstatus/tests/test_timeout.py +++ b/mcstatus/tests/test_timeout.py @@ -13,11 +13,14 @@ class FakeAsyncStream(asyncio.StreamReader): return bytes([0] * n) -def fake_asyncio_asyncio_open_connection(hostname, port): +async def fake_asyncio_asyncio_open_connection(hostname, port): return FakeAsyncStream(), None -@pytest.mark.skipif(sys.platform.startswith("win"), reason="async bug on Windows https://github.com/Dinnerbone/mcstatus/issues/140") +@pytest.mark.skipif( + sys.platform.startswith("win"), + reason="async bug on Windows https://github.com/Dinnerbone/mcstatus/issues/140", +) class TestAsyncSocketConnection: def setup_method(self): self.tcp_async_socket = TCPAsyncSocketConnection() @@ -29,8 +32,7 @@ class TestAsyncSocketConnection: from asyncio import TimeoutError loop = asyncio.get_event_loop() - with patch("asyncio.open_connection") as open_conn: - open_conn.return_value = (FakeAsyncStream(), None) + with patch("asyncio.open_connection", fake_asyncio_asyncio_open_connection) as open_conn: loop.run_until_complete(self.tcp_async_socket.connect("dummy_address", timeout=0.01)) with pytest.raises(TimeoutError):