def test_get_address_host(): f = get_address_host assert f('tcp://127.0.0.1:123') == '127.0.0.1' assert f('inproc://%s/%d/123' % (get_ip(), os.getpid())) == get_ip() if is_zmq_enabled(): assert f('zmq://192.168.2.3:456') == '192.168.2.3'
def test_get_local_address_for(): f = get_local_address_for assert f('tcp://127.0.0.1:80') == 'tcp://127.0.0.1' assert f('tcp://8.8.8.8:4444') == 'tcp://' + get_ip() if has_ipv6(): assert f('tcp://[::1]:123') == 'tcp://[::1]' if is_zmq_enabled(): assert f('zmq://127.0.0.1:80') == 'zmq://127.0.0.1' assert f('zmq://8.8.8.8:4444') == 'zmq://' + get_ip() if has_ipv6(): assert f('zmq://[::1]:123') == 'zmq://[::1]' inproc_arg = 'inproc://%s/%d/444' % (get_ip(), os.getpid()) inproc_res = f(inproc_arg) assert inproc_res.startswith('inproc://') assert inproc_res != inproc_arg
def test_resolve_address(): f = resolve_address assert f('tcp://127.0.0.1:123') == 'tcp://127.0.0.1:123' assert f('127.0.0.2:789') == 'tcp://127.0.0.2:789' assert f('tcp://0.0.0.0:456') == 'tcp://0.0.0.0:456' assert f('tcp://0.0.0.0:456') == 'tcp://0.0.0.0:456' if has_ipv6(): assert f('tcp://[::1]:123') == 'tcp://[::1]:123' # OS X returns '::0.0.0.2' as canonical representation assert f('[::2]:789') in ('tcp://[::2]:789', 'tcp://[::0.0.0.2]:789') assert f('tcp://[::]:123') == 'tcp://[::]:123' assert f('localhost:123') == 'tcp://127.0.0.1:123' assert f('tcp://localhost:456') == 'tcp://127.0.0.1:456' if is_zmq_enabled(): assert f('zmq://127.0.0.1:456') == 'zmq://127.0.0.1:456' assert f('zmq://localhost:789') == 'zmq://127.0.0.1:789' if has_ipv6(): assert f('zmq://[::1]:456') == 'zmq://[::1]:456'
from distributed.core import pingpong from distributed.metrics import time from distributed.utils import get_ip, get_ipv6 from distributed.utils_test import (slow, loop, gen_test, gen_cluster, requires_ipv6, has_ipv6, get_cert) from distributed.protocol import (loads, dumps, to_serialize, Serialized, serialize, deserialize) from distributed.comm import (tcp, inproc, connect, listen, CommClosedError, is_zmq_enabled, parse_address, parse_host_port, unparse_host_port, resolve_address, get_address_host, get_local_address_for) if is_zmq_enabled(): from distributed.comm import zmq def requires_zmq(test_func): return test_func else: zmq = None requires_zmq = pytest.mark.skip("need to enable zmq using special config option") EXTERNAL_IP4 = get_ip() if has_ipv6(): EXTERNAL_IP6 = get_ipv6() ca_file = get_cert('tls-ca-cert.pem')