tiramisu/tests/test_choice_option.py

# coding: utf-8
from py.test import raises
import pytest

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

from tiramisu import ChoiceOption, StrOption, OptionDescription, Config, owners, Calculation, \
                     undefined, Params, ParamValue, ParamOption, list_sessions
from tiramisu.error import ConfigError


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


def return_val(val):
    return val


def return_list():
    return ['val1', 'val2']


def return_calc_list(val):
    return [val]


def return_error(*args, **kwargs):
    raise Exception('test')


@pytest.mark.asyncio
async def test_choiceoption(config_type):
    choice = ChoiceOption('choice', '', values=('val1', 'val2'))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    cfg = await get_config(cfg, config_type)
    owner = await global_owner(cfg, config_type)
    assert await cfg.option('choice').owner.get() == owners.default
    assert await cfg.option('choice').owner.isdefault()
    #
    await cfg.option('choice').value.set('val1')
    assert await cfg.option('choice').owner.get() == owner
    assert not await cfg.option('choice').owner.isdefault()
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.get() == owners.default
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.get() == owners.default
    assert await cfg.option('choice').owner.isdefault()
    #
    assert value_list(await cfg.option('choice').value.list()) == ('val1', 'val2')


@pytest.mark.asyncio
async def test_choiceoption_function(config_type):
    choice = ChoiceOption('choice', '', values=Calculation(return_list))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    cfg = await get_config(cfg, config_type)
    owner = await global_owner(cfg, config_type)
    assert await cfg.option('choice').owner.isdefault()
    #
    await cfg.option('choice').value.set('val1')
    assert await cfg.option('choice').owner.get() == owner
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.isdefault()
    #
    assert value_list(await cfg.option('choice').value.list()) == ('val1', 'val2')


@pytest.mark.asyncio
async def test_choiceoption_function_error():
    choice = ChoiceOption('choice', '', values=Calculation(return_error))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    with pytest.raises(ConfigError):
        await cfg.option('choice').value.set('val1')


@pytest.mark.asyncio
async def test_choiceoption_function_error_args():
    choice = ChoiceOption('choice', '', values=Calculation(return_error, Params(ParamValue('val1'))))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    with pytest.raises(ConfigError):
        await cfg.option('choice').value.set('val1')


@pytest.mark.asyncio
async def test_choiceoption_function_error_kwargs():
    choice = ChoiceOption('choice', '', values=Calculation(return_error, Params(kwargs={'kwargs': ParamValue('val1')})))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    with pytest.raises(ConfigError):
        await cfg.option('choice').value.set('val1')


@pytest.mark.asyncio
async def test_choiceoption_calc_function(config_type):
    choice = ChoiceOption('choice', "", values=Calculation(return_calc_list, Params(ParamValue('val1'))))
    odesc = OptionDescription('od', '', [choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    cfg = await get_config(cfg, config_type)
    owner = await global_owner(cfg, config_type)
    assert await cfg.option('choice').owner.isdefault()
    #
    await cfg.option('choice').value.set('val1')
    assert await cfg.option('choice').owner.get() == owner
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.isdefault()


@pytest.mark.asyncio
async def test_choiceoption_calc_opt_function(config_type):
    str_ = StrOption('str', '', 'val1')
    choice = ChoiceOption('choice',
                          "",
                          values=Calculation(return_calc_list, Params(ParamOption(str_))))
    odesc = OptionDescription('od', '', [str_, choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    owner = await cfg.owner.get()
    cfg = await get_config(cfg, config_type)
    assert await cfg.option('choice').owner.isdefault()
    #
    await cfg.option('choice').value.set('val1')
    assert await cfg.option('choice').owner.get() == owner
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.isdefault()


@pytest.mark.asyncio
async def test_choiceoption_calc_opt_function_propertyerror():
    str_ = StrOption('str', '', 'val1', properties=('disabled',))
    choice = ChoiceOption('choice',
                          "",
                          values=Calculation(return_calc_list, Params(ParamOption(str_))))
    odesc = OptionDescription('od', '', [str_, choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    with pytest.raises(ConfigError):
        await cfg.option('choice').value.set('no')


#def test_choiceoption_calc_opt_multi_function(config_type):
@pytest.mark.asyncio
async def test_choiceoption_calc_opt_multi_function():
    # FIXME
    config_type = 'tiramisu'
    str_ = StrOption('str', '', ['val1'], multi=True)
    choice = ChoiceOption('choice',
                          "",
                          default_multi='val2',
                          values=Calculation(return_val, Params(ParamOption(str_))),
                          multi=True)
    ch2 = ChoiceOption('ch2',
                       "",
                       default=['val2'],
                       values=Calculation(return_val, Params(ParamOption(str_))),
                       multi=True)
    odesc = OptionDescription('od', '', [str_, choice, ch2])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    owner = await cfg.owner.get()
    cfg = await get_config(cfg, config_type, True)
    assert await cfg.option('choice').owner.isdefault()
    assert await cfg.option('choice').value.get() == []
    #
    await cfg.option('choice').value.set(['val1'])
    assert await cfg.option('choice').owner.get() == owner
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set([undefined])
    #
    await cfg.option('choice').value.set(['val1'])
    assert await cfg.option('choice').owner.get() == owner
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError):
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError): 
        await cfg.option('ch2').value.get()


@pytest.mark.asyncio
async def test_choiceoption_calc_opt_multi_function_kwargs(config_type):
    str_ = StrOption('str', '', ['val1'], multi=True)
    choice = ChoiceOption('choice',
                          "",
                          default_multi='val2',
                          values=Calculation(return_val, Params(kwargs={'val': ParamOption(str_)})),
                          multi=True)
    ch2 = ChoiceOption('ch2',
                       "",
                       default=['val2'],
                       values=Calculation(return_val, Params(kwargs={'val': ParamOption(str_)})),
                       multi=True)
    odesc = OptionDescription('od', '', [str_, choice, ch2])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    owner = await cfg.owner.get()
    # FIXME cfg = await get_config(cfg, config_type)
    assert await cfg.option('choice').owner.isdefault()
    assert await cfg.option('choice').value.get() == []
    #
    await cfg.option('choice').value.set(['val1'])
    assert await cfg.option('choice').owner.get() == owner
    #
    with pytest.raises(ValueError): 
        await cfg.option('choice').value.set([undefined])
    #
    await cfg.option('choice').value.set(['val1'])
    assert await cfg.option('choice').owner.get() == owner
    #
    await cfg.option('choice').value.reset()
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError): 
        await cfg.option('choice').value.set('no')
    assert await cfg.option('choice').owner.isdefault()
    #
    with pytest.raises(ValueError): 
        await cfg.option('ch2').value.get()


@pytest.mark.asyncio
async def test_choiceoption_calc_not_list():
    str_ = StrOption('str', '', 'val1')
    choice = ChoiceOption('choice',
                          "",
                          default_multi='val2',
                          values=Calculation(return_val, Params(ParamOption(str_))),
                          multi=True)
    odesc = OptionDescription('od', '', [str_, choice])
    cfg = await Config(odesc)
    await cfg.property.read_write()
    with pytest.raises(ConfigError): 
        await cfg.option('choice').value.set(['val1'])