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)