5
5
6
6
7
7
@asyncio .coroutine
8
- def async_coro (loop ):
9
- yield from asyncio .sleep (0 , loop = loop )
8
+ async def async_coro (loop ):
9
+ await asyncio .sleep (0 , loop = loop )
10
10
return 'ok'
11
11
12
12
@@ -40,8 +40,7 @@ async def closer(_, writer):
40
40
await server1 .wait_closed ()
41
41
42
42
43
- @pytest .mark .asyncio
44
- async def test_unused_port_factory_fixture (unused_tcp_port_factory , event_loop ):
43
+ def test_unused_port_factory_fixture (unused_tcp_port_factory , event_loop ):
45
44
"""Test the unused TCP port factory fixture."""
46
45
47
46
async def closer (_ , writer ):
@@ -50,28 +49,34 @@ async def closer(_, writer):
50
49
port1 , port2 , port3 = (unused_tcp_port_factory (), unused_tcp_port_factory (),
51
50
unused_tcp_port_factory ())
52
51
53
- server1 = await asyncio .start_server (closer , host = 'localhost' ,
54
- port = port1 ,
55
- loop = event_loop )
56
- server2 = await asyncio .start_server (closer , host = 'localhost' ,
57
- port = port2 ,
58
- loop = event_loop )
59
- server3 = await asyncio .start_server (closer , host = 'localhost' ,
60
- port = port3 ,
61
- loop = event_loop )
52
+ async def run_test ():
53
+ server1 = await asyncio .start_server (closer , host = 'localhost' ,
54
+ port = port1 ,
55
+ loop = event_loop )
56
+ server2 = await asyncio .start_server (closer , host = 'localhost' ,
57
+ port = port2 ,
58
+ loop = event_loop )
59
+ server3 = await asyncio .start_server (closer , host = 'localhost' ,
60
+ port = port3 ,
61
+ loop = event_loop )
62
62
63
- for port in port1 , port2 , port3 :
64
- with pytest .raises (IOError ):
65
- await asyncio .start_server (closer , host = 'localhost' ,
66
- port = port ,
67
- loop = event_loop )
63
+ for port in port1 , port2 , port3 :
64
+ with pytest .raises (IOError ):
65
+ await asyncio .start_server (closer , host = 'localhost' ,
66
+ port = port ,
67
+ loop = event_loop )
68
68
69
- server1 .close ()
70
- await server1 .wait_closed ()
71
- server2 .close ()
72
- await server2 .wait_closed ()
73
- server3 .close ()
74
- await server3 .wait_closed ()
69
+ server1 .close ()
70
+ await server1 .wait_closed ()
71
+ server2 .close ()
72
+ await server2 .wait_closed ()
73
+ server3 .close ()
74
+ await server3 .wait_closed ()
75
+
76
+ event_loop .run_until_complete (run_test ())
77
+
78
+ event_loop .stop ()
79
+ event_loop .close ()
75
80
76
81
77
82
class Test :
@@ -82,3 +87,8 @@ async def test_asyncio_marker_method(self, event_loop):
82
87
"""Test the asyncio pytest marker in a Test class."""
83
88
ret = await async_coro (event_loop )
84
89
assert ret == 'ok'
90
+
91
+
92
+ def test_async_close_loop (event_loop ):
93
+ event_loop .close ()
94
+ return 'ok'
0 commit comments