def test_spawn_ipv4():
cfile = create_config(("listen=127.0.0.1:9999",))
Config(cfile).load()
loop = asyncio.new_event_loop()
with mock.patch("socket.socket.bind"):
supervisor = Supervisor(loop=loop)
assert len(supervisor.socks) == 1
supervisor.close_socks()
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
def test_create():
cfile = create_config(("listen=:9999, :::9999", "workers=2"))
Config(cfile).load()
loop = asyncio.new_event_loop()
with mock.patch("socket.socket.bind"), mock.patch(
"blackhole.worker.Worker.start"
):
supervisor = Supervisor(loop=loop)
supervisor.start_workers()
assert len(supervisor.workers) == 2
supervisor.close_socks()
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
def test_spawn_ipv4_tls():
cert = create_file("cert.pem")
key = create_file("key.key")
cfile = create_config(
(
"listen=127.0.0.1:9998",
"tls_listen=127.0.0.1:9999",
"tls_cert={}".format(cert),
"tls_key={}".format(key),
)
)
conf = Config(cfile).load()
conf.args = Args((("less_secure", True),))
loop = asyncio.new_event_loop()
with mock.patch("socket.socket.bind"), mock.patch(
"ssl.create_default_context"
):
supervisor = Supervisor(loop=loop)
assert len(supervisor.socks) == 2
supervisor.close_socks()
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
def test_spawn_ipv4_tls_dhparams():
cert = create_file("test.pem", cert_data)
key = create_file("test.key", key_data)
dhparams = create_file("dhparams.pem", dhparams_data)
cfile = create_config(
(
"listen=127.0.0.1:9998",
"tls_listen=127.0.0.1:9999",
"tls_cert={}".format(cert),
"tls_key={}".format(key),
"tls_dhparams={}".format(dhparams),
)
)
conf = Config(cfile).load()
conf.args = Args((("less_secure", False),))
loop = asyncio.new_event_loop()
with mock.patch("socket.socket.bind"):
supervisor = Supervisor(loop=loop)
assert len(supervisor.socks) == 2
supervisor.close_socks()
loop.run_until_complete(loop.shutdown_asyncgens())
loop.close()
