tiramisu/tests/test_config_ip.py

from .autopath import do_autopath
do_autopath()
from .config import config_type, get_config, value_list, global_owner, event_loop

import warnings
import pytest
from tiramisu import Config, IPOption, NetworkOption, NetmaskOption, \
                     PortOption, BroadcastOption, OptionDescription
from tiramisu.error import ValueWarning
from tiramisu.storage import list_sessions


@pytest.mark.asyncio
async def test_ip(config_type):
    a = IPOption('a', '')
    b = IPOption('b', '', private_only=True)
    d = IPOption('d', '', warnings_only=True, private_only=True)
    warnings.simplefilter("always", ValueWarning)
    od = OptionDescription('od', '', [a, b, d])
    async with await Config(od) as cfg:
        cfg = await get_config(cfg, config_type)
        await cfg.option('a').value.set('192.168.1.1')
        await cfg.option('a').value.set('192.168.1.0')
        await cfg.option('a').value.set('88.88.88.88')
        await cfg.option('a').value.set('0.0.0.0')
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('255.255.255.0')
        await cfg.option('b').value.set('192.168.1.1')
        await cfg.option('b').value.set('192.168.1.0')
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('b').value.set('88.88.88.88')
        await cfg.option('b').value.set('0.0.0.0')
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('b').value.set('255.255.255.0')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('333.0.1.20')

        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                IPOption('a', 'ip', default='192.000.023.01')
            with warnings.catch_warnings(record=True) as w:
                await cfg.option('d').value.set('88.88.88.88')
            assert len(w) == 1
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_ip_cidr():
    b = IPOption('b', '', private_only=True, cidr=True)
    c = IPOption('c', '', private_only=True)
    warnings.simplefilter("always", ValueWarning)
    od = OptionDescription('od', '', [b, c])
    async with await Config(od) as cfg:
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('192.168.1.1')
        await cfg.option('b').value.set('192.168.1.1/24')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('192.168.1.1/32')
        #
        await cfg.option('c').value.set('192.168.1.1')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('192.168.1.1/24')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('192.168.1.1/32')
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_ip_default():
    a = IPOption('a', '', '88.88.88.88')
    od = OptionDescription('od', '', [a])
    async with await Config(od) as cfg:
        await cfg.option('a').value.get() == '88.88.88.88'
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_ip_reserved(config_type):
    a = IPOption('a', '')
    b = IPOption('b', '', allow_reserved=True)
    c = IPOption('c', '', warnings_only=True)
    od = OptionDescription('od', '', [a, b, c])
    warnings.simplefilter("always", ValueWarning)
    async with await Config(od) as cfg:
        cfg = await get_config(cfg, config_type)
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('240.94.1.1')
        await cfg.option('b').value.set('240.94.1.1')
        if config_type != 'tiramisu-api':
            # FIXME
            with warnings.catch_warnings(record=True) as w:
                await cfg.option('c').value.set('240.94.1.1')
            assert len(w) == 1
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_network(config_type):
    a = NetworkOption('a', '')
    b = NetworkOption('b', '', warnings_only=True)
    od = OptionDescription('od', '', [a, b])
    warnings.simplefilter("always", ValueWarning)
    async with await Config(od) as cfg:
        cfg = await get_config(cfg, config_type)
        await cfg.option('a').value.set('192.168.1.1')
        await cfg.option('a').value.set('192.168.1.0')
        await cfg.option('a').value.set('88.88.88.88')
        await cfg.option('a').value.set('0.0.0.0')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set(1)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('1.1.1.1.1')
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('255.255.255.0')
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('192.168.001.0')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('333.168.1.1')
        if config_type != 'tiramisu-api':
            # FIXME
            with warnings.catch_warnings(record=True) as w:
                await cfg.option('b').value.set('255.255.255.0')
            assert len(w) == 1
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_network_cidr(config_type):
    a = NetworkOption('a', '', cidr=True)
    od = OptionDescription('od', '', [a])
    async with await Config(od) as cfg:
        # FIXME cfg = await get_config(cfg, config_type)
        await cfg.option('a').value.set('192.168.1.1/32')
        await cfg.option('a').value.set('192.168.1.0/24')
        await cfg.option('a').value.set('88.88.88.88/32')
        await cfg.option('a').value.set('0.0.0.0/0')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.1.1')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.1.1/24')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('2001:db00::0/24')
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_network_invalid():
    with pytest.raises(ValueError):
        NetworkOption('a', '', default='toto')


@pytest.mark.asyncio
async def test_netmask(config_type):
    a = NetmaskOption('a', '')
    od = OptionDescription('od', '', [a])
    async with await Config(od) as cfg:
        cfg = await get_config(cfg, config_type)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.1.1.1')
        if config_type != 'tiramisu-api':
            # FIXME
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('192.168.1.1')
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('192.168.1.0')
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('88.88.88.88')
            with pytest.raises(ValueError):
                await cfg.option('a').value.set('255.255.255.000')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set(2)
        await cfg.option('a').value.set('0.0.0.0')
        await cfg.option('a').value.set('255.255.255.0')
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_broadcast(config_type):
    a = BroadcastOption('a', '')
    od = OptionDescription('od', '', [a])
    async with await Config(od) as cfg:
        # FIXME cfg = await get_config(cfg, config_type)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.1.255.1')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.001.255')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('192.168.0.300')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set(1)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set(2)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('2001:db8::1')
        await cfg.option('a').value.set('0.0.0.0')
        await cfg.option('a').value.set('255.255.255.0')
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_port(config_type):
    a = PortOption('a', '')
    b = PortOption('b', '', allow_zero=True)
    c = PortOption('c', '', allow_zero=True, allow_registred=False)
    d = PortOption('d', '', allow_zero=True, allow_wellknown=False, allow_registred=False)
    e = PortOption('e', '', allow_zero=True, allow_private=True)
    f = PortOption('f', '', allow_private=True)
    od = OptionDescription('od', '', [a, b, c, d, e, f])
    async with await Config(od) as cfg:
        # FIXME cfg = await get_config(cfg, config_type)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('0')
        await cfg.option('a').value.set('1')
        await cfg.option('a').value.set('1023')
        await cfg.option('a').value.set('1024')
        await cfg.option('a').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('65536')

        await cfg.option('b').value.set('0')
        await cfg.option('b').value.set('1')
        await cfg.option('b').value.set('1023')
        await cfg.option('b').value.set('1024')
        await cfg.option('b').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('65536')

        await cfg.option('c').value.set('0')
        await cfg.option('c').value.set('1')
        await cfg.option('c').value.set('1023')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('1024')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('65536')

        await cfg.option('d').value.set('0')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1023')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1024')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('65536')

        await cfg.option('e').value.set('0')
        await cfg.option('e').value.set('1')
        await cfg.option('e').value.set('1023')
        await cfg.option('e').value.set('1024')
        await cfg.option('e').value.set('49151')
        await cfg.option('e').value.set('49152')
        await cfg.option('e').value.set('65535')

        with pytest.raises(ValueError):
            await cfg.option('f').value.set('0')
        await cfg.option('f').value.set('1')
        await cfg.option('f').value.set('1023')
        await cfg.option('f').value.set('1024')
        await cfg.option('f').value.set('49151')
        await cfg.option('f').value.set('49152')
        await cfg.option('f').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('f').value.set('65536')
    assert not await list_sessions()


@pytest.mark.asyncio
async def test_port_range(config_type):
    a = PortOption('a', '', allow_range=True)
    b = PortOption('b', '', allow_range=True, allow_zero=True)
    c = PortOption('c', '', allow_range=True, allow_zero=True, allow_registred=False)
    d = PortOption('d', '', allow_range=True, allow_zero=True, allow_wellknown=False, allow_registred=False)
    e = PortOption('e', '', allow_range=True, allow_zero=True, allow_private=True)
    f = PortOption('f', '', allow_range=True, allow_private=True)
    od = OptionDescription('od', '', [a, b, c, d, e, f])
    async with await Config(od) as cfg:
        # FIXME cfg = await get_config(cfg, config_type)
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('0')
        await cfg.option('a').value.set('1')
        await cfg.option('a').value.set('1023')
        await cfg.option('a').value.set('1024')
        await cfg.option('a').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('65536')
        await cfg.option('a').value.set('1:49151')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('0:49151')
        with pytest.raises(ValueError):
            await cfg.option('a').value.set('1:49152')

        await cfg.option('b').value.set('0')
        await cfg.option('b').value.set('1')
        await cfg.option('b').value.set('1023')
        await cfg.option('b').value.set('1024')
        await cfg.option('b').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('65536')
        await cfg.option('b').value.set('0:49151')
        with pytest.raises(ValueError):
            await cfg.option('b').value.set('0:49152')

        await cfg.option('c').value.set('0')
        await cfg.option('c').value.set('1')
        await cfg.option('c').value.set('1023')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('1024')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('65536')
        await cfg.option('c').value.set('0:1023')
        with pytest.raises(ValueError):
            await cfg.option('c').value.set('0:1024')

        await cfg.option('d').value.set('0')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1023')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('1024')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('49151')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('49152')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('65536')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('0:0')
        with pytest.raises(ValueError):
            await cfg.option('d').value.set('0:1')

        await cfg.option('e').value.set('0')
        await cfg.option('e').value.set('1')
        await cfg.option('e').value.set('1023')
        await cfg.option('e').value.set('1024')
        await cfg.option('e').value.set('49151')
        await cfg.option('e').value.set('49152')
        await cfg.option('e').value.set('65535')
        await cfg.option('e').value.set('0:65535')
        with pytest.raises(ValueError):
            await cfg.option('e').value.set('0:65536')

        with pytest.raises(ValueError):
            await cfg.option('f').value.set('0')
        await cfg.option('f').value.set('1')
        await cfg.option('f').value.set('1023')
        await cfg.option('f').value.set('1024')
        await cfg.option('f').value.set('49151')
        await cfg.option('f').value.set('49152')
        await cfg.option('f').value.set('65535')
        with pytest.raises(ValueError):
            await cfg.option('f').value.set('65536')
        await cfg.option('f').value.set('1:65535')
        await cfg.option('f').value.set('3:4')
        with pytest.raises(ValueError):
            await cfg.option('f').value.set('0:65535')
        with pytest.raises(ValueError):
            await cfg.option('f').value.set('4:3')
    assert not await list_sessions()