# coding: utf-8
from .autopath import do_autopath
do_autopath()
from .config import config_type, get_config


import pytest
from tiramisu import IntOption, StrOption, OptionDescription, Config
from tiramisu.error import PropertiesOptionError, ConfigError
from tiramisu.storage import list_sessions, delete_session


def teardown_function(function):
    assert list_sessions() == [], 'session list is not empty when leaving "{}"'.format(function.__name__)


def make_description():
    u1 = IntOption('u1', '', properties=('frozen', 'mandatory', 'disabled', ))
    u2 = IntOption('u2', '', properties=('frozen', 'mandatory', 'disabled', ))
    return OptionDescription('od1', '', [u1, u2])


@pytest.mark.asyncio
async def test_permissive(config_type):
    descr = make_description()
    cfg_ori = await Config(descr)
    await cfg_ori.property.read_write()
    await cfg_ori.property.read_write()
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.unrestraint.permissive.add('disabled')
    await cfg_ori.unrestraint.permissive.pop('hidden')
    assert await cfg_ori.unrestraint.permissive.get() == frozenset(['disabled'])
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.add('permissive')
    cfg = await get_config(cfg_ori, config_type)
    await cfg.option('u1').value.get()
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.pop('permissive')
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}


@pytest.mark.asyncio
async def test_permissive_add(config_type):
    descr = make_description()
    cfg_ori = await Config(descr)
    await cfg_ori.property.read_write()
    await cfg_ori.property.read_write()
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.unrestraint.permissive.add('disabled')
    assert await cfg_ori.unrestraint.permissive.get() == frozenset(['hidden', 'disabled'])
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.add('permissive')
    cfg = await get_config(cfg_ori, config_type)
    await cfg.option('u1').value.get()
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.pop('permissive')
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}


@pytest.mark.asyncio
async def test_permissive_pop():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    await cfg.property.read_write()
    props = frozenset()
    try:
        await cfg.forcepermissive.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    await cfg.unrestraint.permissive.add('disabled')
    assert await cfg.unrestraint.permissive.get() == frozenset(['hidden', 'disabled'])
    await cfg.forcepermissive.option('u1').value.get()
    await cfg.unrestraint.permissive.pop('disabled')
    props = frozenset()
    try:
        await cfg.forcepermissive.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}


@pytest.mark.asyncio
async def test_permissive_reset():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    assert await cfg.unrestraint.permissive.get() == frozenset(['hidden'])
    #
    await cfg.unrestraint.permissive.add('disabled')
    await cfg.unrestraint.permissive.pop('hidden')
    assert await cfg.unrestraint.permissive.get() == frozenset(['disabled'])
    #
    await cfg.unrestraint.permissive.reset()
    assert await cfg.unrestraint.permissive.get() == frozenset()


@pytest.mark.asyncio
async def test_permissive_mandatory():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_only()
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])
    await cfg.unrestraint.permissive.add('mandatory')
    await cfg.unrestraint.permissive.add('disabled')
    assert await cfg.unrestraint.permissive.get() == frozenset(['mandatory', 'disabled'])
    await cfg.property.add('permissive')
    await cfg.option('u1').value.get()
    await cfg.property.pop('permissive')
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])


@pytest.mark.asyncio
async def test_permissive_frozen():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    await cfg.unrestraint.permissive.pop('hidden')
    await cfg.unrestraint.permissive.add('frozen')
    await cfg.unrestraint.permissive.add('disabled')
    assert await cfg.unrestraint.permissive.get() == frozenset(['frozen', 'disabled'])
    assert await cfg.permissive.get() == frozenset(['frozen', 'disabled'])
    try:
        await cfg.option('u1').value.set(1)
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])
    await cfg.property.add('permissive')
    await cfg.option('u1').value.set(1)
    assert await cfg.option('u1').value.get() == 1
    await cfg.property.pop('permissive')
    try:
        await cfg.option('u1').value.set(1)
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])


@pytest.mark.asyncio
async def test_invalid_permissive():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    # FIXME with pytest.raises(TypeError): 
    #    await cfg.unrestraint.permissive.set(['frozen', 'disabled'])")


@pytest.mark.asyncio
async def test_forbidden_permissive():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    with pytest.raises(ConfigError): 
        await cfg.permissive.add('force_default_on_freeze')
    with pytest.raises(ConfigError): 
        await cfg.permissive.add('force_metaconfig_on_freeze')


@pytest.mark.asyncio
async def test_permissive_option(config_type):
    descr = make_description()
    cfg_ori = await Config(descr)
    await cfg_ori.property.read_write()
    cfg = await get_config(cfg_ori, config_type)

    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.unrestraint.option('u1').permissive.set(frozenset(['disabled']))
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.add('permissive')
    cfg = await get_config(cfg_ori, config_type)
    await cfg.option('u1').value.get()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.property.pop('permissive')
    cfg = await get_config(cfg_ori, config_type)
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}


@pytest.mark.asyncio
async def test_permissive_option_cache():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()

    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    await cfg.unrestraint.option('u1').permissive.set(frozenset(['disabled']))
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    await cfg.property.add('permissive')
    await cfg.option('u1').value.get()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}

    await cfg.property.pop('permissive')
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset()
    props = frozenset()
    try:
        await cfg.option('u2').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert set(props) == {'disabled'}


@pytest.mark.asyncio
async def test_permissive_option_mandatory():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_only()
    props = frozenset()
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])
    await cfg.unrestraint.option('u1').permissive.set(frozenset(['mandatory', 'disabled']))
    assert await cfg.unrestraint.option('u1').permissive.get() == frozenset(['mandatory', 'disabled'])
    await cfg.property.add('permissive')
    await cfg.option('u1').value.get()
    await cfg.property.pop('permissive')
    try:
        await cfg.option('u1').value.get()
    except PropertiesOptionError as err:
        props = err.proptype
    assert frozenset(props) == frozenset(['disabled'])


@pytest.mark.asyncio
async def test_permissive_option_frozen():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    await cfg.unrestraint.option('u1').permissive.set(frozenset(['frozen', 'disabled']))
    await cfg.option('u1').value.set(1)
    assert await cfg.option('u1').value.get() == 1
    await cfg.property.add('permissive')
    assert await cfg.option('u1').value.get() == 1
    await cfg.property.pop('permissive')
    assert await cfg.option('u1').value.get() == 1


@pytest.mark.asyncio
async def test_invalid_option_permissive():
    descr = make_description()
    cfg = await Config(descr)
    await cfg.property.read_write()
    with pytest.raises(TypeError): 
        await cfg.unrestraint.option('u1').permissive.set(['frozen', 'disabled'])


@pytest.mark.asyncio
async def test_remove_option_permissive(config_type):
    var1 = StrOption('var1', '', u'value', properties=('hidden',))
    od1 = OptionDescription('od1', '', [var1])
    rootod = OptionDescription('rootod', '', [od1])
    cfg_ori = await Config(rootod)
    await cfg_ori.property.read_write()
    cfg = await get_config(cfg_ori, config_type)
    with pytest.raises(PropertiesOptionError): 
        await cfg.option('od1.var1').value.get()
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.forcepermissive.option('od1.var1').permissive.set(frozenset(['hidden']))
    assert await cfg_ori.forcepermissive.option('od1.var1').permissive.get() == frozenset(['hidden'])
    cfg = await get_config(cfg_ori, config_type)
    assert await cfg.option('od1.var1').value.get() == 'value'
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.forcepermissive.option('od1.var1').permissive.set(frozenset())
    assert await cfg_ori.forcepermissive.option('od1.var1').permissive.get() == frozenset()
    cfg = await get_config(cfg_ori, config_type)
    with pytest.raises(PropertiesOptionError): 
        await cfg.option('od1.var1').value.get()


@pytest.mark.asyncio
async def test_reset_option_permissive(config_type):
    var1 = StrOption('var1', '', u'value', properties=('hidden',))
    od1 = OptionDescription('od1', '', [var1])
    rootod = OptionDescription('rootod', '', [od1])
    cfg_ori = await Config(rootod)
    await cfg_ori.property.read_write()
    cfg = await get_config(cfg_ori, config_type)
    with pytest.raises(PropertiesOptionError): 
        await cfg.option('od1.var1').value.get()
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.forcepermissive.option('od1.var1').permissive.set(frozenset(['hidden']))
    assert await cfg_ori.forcepermissive.option('od1.var1').permissive.get() == frozenset(['hidden'])
    cfg = await get_config(cfg_ori, config_type)
    assert await cfg.option('od1.var1').value.get() == 'value'
    if config_type == 'tiramisu-api':
        await cfg.send()
    await cfg_ori.forcepermissive.option('od1.var1').permissive.reset()
    assert await cfg_ori.forcepermissive.option('od1.var1').permissive.get() == frozenset()
    cfg = await get_config(cfg_ori, config_type)
    with pytest.raises(PropertiesOptionError): 
        await cfg.option('od1.var1').value.get()