from __future__ import print_function, division, absolute_import

import pytest
pytest.importorskip('requests')

from distributed.cli.utils import uri_from_host_port
from distributed.utils import get_ip


external_ip = get_ip()


def test_uri_from_host_port():
    f = uri_from_host_port

    assert f('', 456, None) == 'tcp://:456'
    assert f('', 456, 123) == 'tcp://:456'
    assert f('', None, 123) == 'tcp://:123'
    assert f('', None, 0) == 'tcp://'
    assert f('', 0, 123) == 'tcp://'

    assert f('localhost', 456, None) == 'tcp://localhost:456'
    assert f('localhost', 456, 123) == 'tcp://localhost:456'
    assert f('localhost', None, 123) == 'tcp://localhost:123'
    assert f('localhost', None, 0) == 'tcp://localhost'

    assert f('192.168.1.2', 456, None) == 'tcp://192.168.1.2:456'
    assert f('192.168.1.2', 456, 123) == 'tcp://192.168.1.2:456'
    assert f('192.168.1.2', None, 123) == 'tcp://192.168.1.2:123'
    assert f('192.168.1.2', None, 0) == 'tcp://192.168.1.2'

    assert f('tcp://192.168.1.2', 456, None) == 'tcp://192.168.1.2:456'
    assert f('tcp://192.168.1.2', 456, 123) == 'tcp://192.168.1.2:456'
    assert f('tcp://192.168.1.2', None, 123) == 'tcp://192.168.1.2:123'
    assert f('tcp://192.168.1.2', None, 0) == 'tcp://192.168.1.2'

    assert f('tcp://192.168.1.2:456', None, None) == 'tcp://192.168.1.2:456'
    assert f('tcp://192.168.1.2:456', 0, 0) == 'tcp://192.168.1.2:456'
    assert f('tcp://192.168.1.2:456', 0, 123) == 'tcp://192.168.1.2:456'
    assert f('tcp://192.168.1.2:456', 456, 123) == 'tcp://192.168.1.2:456'

    with pytest.raises(ValueError):
        # Two incompatible port values
        f('tcp://192.168.1.2:456', 123, None)

    assert f('tls://192.168.1.2:456', None, None) == 'tls://192.168.1.2:456'
    assert f('tls://192.168.1.2:456', 0, 0) == 'tls://192.168.1.2:456'
    assert f('tls://192.168.1.2:456', 0, 123) == 'tls://192.168.1.2:456'
    assert f('tls://192.168.1.2:456', 456, 123) == 'tls://192.168.1.2:456'

    assert f('tcp://[::1]:456', None, None) == 'tcp://[::1]:456'

    assert f('tls://[::1]:456', None, None) == 'tls://[::1]:456'