Skip to content

Commit a941003

Browse files
committed
Fix
1 parent 4d4cf08 commit a941003

File tree

2 files changed

+12
-8
lines changed

2 files changed

+12
-8
lines changed

tests/test_connector.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -179,16 +179,17 @@ async def test_connection_del_loop_debug() -> None:
179179
exc_handler.assert_called_with(loop, msg)
180180

181181

182-
async def test_connection_del_loop_closed() -> None:
183-
loop = asyncio.get_running_loop()
182+
def test_connection_del_loop_closed(
183+
event_loop: asyncio.AbstractEventLoop,
184+
) -> None:
184185
connector = mock.Mock()
185186
key = mock.Mock()
186187
protocol = mock.Mock()
187-
loop.set_debug(True)
188-
conn = Connection(connector, key, protocol, loop=loop)
188+
event_loop.set_debug(True)
189+
conn = Connection(connector, key, protocol, loop=event_loop)
189190
exc_handler = mock.Mock()
190-
loop.set_exception_handler(exc_handler)
191-
loop.close()
191+
event_loop.set_exception_handler(exc_handler)
192+
event_loop.close()
192193

193194
with pytest.warns(ResourceWarning):
194195
del conn
@@ -261,7 +262,6 @@ def test_del_with_closed_loop( # type: ignore[misc]
261262
async def make_conn() -> aiohttp.BaseConnector:
262263
return aiohttp.BaseConnector()
263264

264-
event_loop = asyncio.get_running_loop()
265265
conn = event_loop.run_until_complete(make_conn())
266266
transp = create_mocked_conn(event_loop)
267267
conn._conns[key] = deque([(transp, 123)])

tests/test_resolver.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -314,9 +314,13 @@ async def test_close_for_async_resolver() -> None:
314314

315315

316316
def test_default_loop_for_threaded_resolver() -> None:
317+
async def create_resolver() -> ThreadedResolver:
318+
"""Create resolver in async context."""
319+
return ThreadedResolver()
320+
317321
loop = asyncio.new_event_loop()
318322
asyncio.set_event_loop(loop)
319-
resolver = ThreadedResolver()
323+
resolver = loop.run_until_complete(create_resolver())
320324
assert resolver._loop is loop
321325
loop.close()
322326

0 commit comments

Comments
 (0)