Conform stream_id to go-mplex

This commit is contained in:
mhchia
2019-08-28 21:43:34 +08:00
parent 9b60e1757d
commit d35b8ffc64
5 changed files with 55 additions and 46 deletions

View File

@ -23,14 +23,16 @@ async def hosts(num_hosts):
await asyncio.gather(
*[_host.get_network().listen(LISTEN_MADDR) for _host in _hosts]
)
yield _hosts
# Clean up
listeners = []
for _host in _hosts:
for listener in _host.get_network().listeners.values():
listener.server.close()
listeners.append(listener)
await asyncio.gather(*[listener.server.wait_closed() for listener in listeners])
try:
yield _hosts
finally:
# Clean up
listeners = []
for _host in _hosts:
for listener in _host.get_network().listeners.values():
listener.server.close()
listeners.append(listener)
await asyncio.gather(*[listener.server.wait_closed() for listener in listeners])
@pytest.fixture