mirrors/granian
1
0
Fork 0
mirror of https://github.com/emmett-framework/granian.git synced 2025-12-23 05:36:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enhance cmd spawn reliability in tests

Browse source
This commit is contained in:
Giovanni Barillari 2024-01-19 15:46:57 +01:00
parent ce592e448a
commit 8e4f699d07
1 changed files with 32 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

44
tests/conftest.py
View file

@ -14,19 +14,39 @@ async def _server(interface, port, threading_mode, tls=False):
tls_opts = (
(f"--ssl-certificate {certs_path / 'cert.pem'} " f"--ssl-keyfile {certs_path / 'key.pem'} ") if tls else ''
)
proc = await asyncio.create_subprocess_shell(
''.join(
[
f'granian --interface {interface} --port {port} ',
f'--threads 1 --threading-mode {threading_mode} ',
tls_opts,
'--opt ' if os.getenv('LOOP_OPT') else '',
f'tests.apps.{interface}:app',
]
),
env=dict(os.environ),
cmd = ''.join(
[
f'granian --interface {interface} --port {port} ',
f'--threads 1 --threading-mode {threading_mode} ',
tls_opts,
'--opt ' if os.getenv('LOOP_OPT') else '',
f'tests.apps.{interface}:app',
]
)
await asyncio.sleep(1)
succeeded, spawn_failures = False, 0
while spawn_failures < 3:
proc = await asyncio.create_subprocess_shell(cmd, env=dict(os.environ))
conn_failures = 0
while conn_failures < 3:
try:
await asyncio.sleep(1.5)
sock = socket.create_connection(('127.0.0.1', port), timeout=1)
sock.close()
succeeded = True
break
except Exception:
conn_failures += 1
if succeeded:
break
proc.terminate()
await proc.wait()
spawn_failures += 1
if not succeeded:
raise RuntimeError('Cannot bind server')
try:
yield port
finally: