borgmatic/atticmatic/tests/unit/test_config.py

from flexmock import flexmock
from nose.tools import assert_raises

from atticmatic import config as module


def insert_mock_parser(section_names):
    parser = flexmock()
    parser.should_receive('readfp')
    parser.should_receive('sections').and_return(section_names)
    flexmock(module).open = lambda filename: None
    flexmock(module).ConfigParser = parser

    return parser


def test_parse_configuration_should_return_config_data():
    section_names = (module.CONFIG_SECTION_LOCATION, module.CONFIG_SECTION_RETENTION)
    parser = insert_mock_parser(section_names)

    for section_name in section_names:
        parser.should_receive('options').with_args(section_name).and_return(
            module.CONFIG_FORMAT[section_name],
        )

    expected_config = (
        module.LocationConfig(flexmock(), flexmock()),
        module.RetentionConfig(flexmock(), flexmock(), flexmock()),
    )
    sections = (
        (module.CONFIG_SECTION_LOCATION, expected_config[0], 'get'),
        (module.CONFIG_SECTION_RETENTION, expected_config[1], 'getint'),
    )

    for section_name, section_config, method_name in sections:
        for index, option_name in enumerate(module.CONFIG_FORMAT[section_name]):
            (
                parser.should_receive(method_name).with_args(section_name, option_name)
                .and_return(section_config[index])
            )

    config = module.parse_configuration(flexmock())

    assert config == expected_config


def test_parse_configuration_with_missing_section_should_raise():
    insert_mock_parser((module.CONFIG_SECTION_LOCATION,))

    with assert_raises(ValueError):
        module.parse_configuration(flexmock())


def test_parse_configuration_with_extra_section_should_raise():
    insert_mock_parser((module.CONFIG_SECTION_LOCATION, module.CONFIG_SECTION_RETENTION, 'extra'))

    with assert_raises(ValueError):
        module.parse_configuration(flexmock())


def test_parse_configuration_with_missing_option_should_raise():
    section_names = (module.CONFIG_SECTION_LOCATION, module.CONFIG_SECTION_RETENTION)
    parser = insert_mock_parser(section_names)

    for section_name in section_names:
        parser.should_receive('options').with_args(section_name).and_return(
            module.CONFIG_FORMAT[section_name][:-1],
        )

    with assert_raises(ValueError):
        module.parse_configuration(flexmock())


def test_parse_configuration_with_extra_option_should_raise():
    section_names = (module.CONFIG_SECTION_LOCATION, module.CONFIG_SECTION_RETENTION)
    parser = insert_mock_parser(section_names)

    for section_name in section_names:
        parser.should_receive('options').with_args(section_name).and_return(
            module.CONFIG_FORMAT[section_name] + ('extra',),
        )

    with assert_raises(ValueError):
        module.parse_configuration(flexmock())