def event_loop(pytestconfig, request):
if any(marker.name == 'default_loop'
for marker in request.node.own_markers):
set_loop_policy(posix_loop_type='selector',
windows_loop_type='proactor')
else:
loop_type = pytestconfig.getoption("--loop")
if loop_type:
set_loop_policy(posix_loop_type=loop_type,
windows_loop_type=loop_type)
loop = asyncio.new_event_loop()
yield loop
loop.close()
from aionetworking.utils import set_loop_policy
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('conf',
type=str,
nargs='?',
help='Path to .yaml config file')
parser.add_argument('-t',
'--timeout',
default=None,
type=int,
help='stop the server after this time')
parser.add_argument('-l',
'--loop',
default=None,
type=str,
help='loop to use')
parser.add_argument(
'-p',
'--notify-pid',
type=int,
help='pid of process to send USRSIG2 to when server is started')
args, kw = parser.parse_known_args()
if args.loop:
set_loop_policy(posix_loop_type=args.loop, windows_loop_type=args.loop)
run_server_default_tags(args.conf,
notify_pid=args.notify_pid,
duration=args.timeout)
def pytest_configure(config):
loop_type = config.getoption("--loop")
if loop_type:
set_loop_policy(posix_loop_type=loop_type, windows_loop_type=loop_type)
def __init__(self, args):
super().__init__(args)
set_loop_policy(windows_loop_type=self.loop)
self.task = None
self.loop = None
socket.setdefaulttimeout(60)