83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
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('read')
|
|
parser.should_receive('sections').and_return(section_names)
|
|
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())
|