1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
from __future__ import annotations
from textwrap import dedent
from pytest import Pytester
def test_event_loop_fixture_respects_event_loop_policy(pytester: Pytester):
pytester.makeconftest(
dedent(
"""\
'''Defines and sets a custom event loop policy'''
import asyncio
from asyncio import DefaultEventLoopPolicy, SelectorEventLoop
class TestEventLoop(SelectorEventLoop):
pass
class TestEventLoopPolicy(DefaultEventLoopPolicy):
def new_event_loop(self):
return TestEventLoop()
# This statement represents a code which sets a custom event loop policy
asyncio.set_event_loop_policy(TestEventLoopPolicy())
"""
)
)
pytester.makepyfile(
dedent(
"""\
'''Tests that any externally provided event loop policy remains unaltered'''
import asyncio
import pytest
@pytest.mark.asyncio
async def test_uses_loop_provided_by_custom_policy():
'''Asserts that test cases use the event loop
provided by the custom event loop policy'''
assert type(asyncio.get_event_loop()).__name__ == "TestEventLoop"
@pytest.mark.asyncio
async def test_custom_policy_is_not_overwritten():
'''
Asserts that any custom event loop policy stays the same
across test cases.
'''
assert type(asyncio.get_event_loop()).__name__ == "TestEventLoop"
"""
)
)
result = pytester.runpytest_subprocess("--asyncio-mode=strict")
result.assert_outcomes(passed=2)
|
