24 lines
870 B
Python
24 lines
870 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from util.config.validators import ConfigValidationException
|
||
|
|
from util.config.validators.validate_timemachine import TimeMachineValidator
|
||
|
|
from util.morecollections import AttrDict
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('default_exp,options,expected_exception', [
|
||
|
|
('2d', ['1w', '2d'], None),
|
||
|
|
|
||
|
|
('2d', ['1w'], 'Default expiration must be in expiration options set'),
|
||
|
|
('2d', ['2d', '1M'], 'Invalid tag expiration option: 1M'),
|
||
|
|
])
|
||
|
|
def test_validate(default_exp, options, expected_exception):
|
||
|
|
config = {}
|
||
|
|
config['DEFAULT_TAG_EXPIRATION'] = default_exp
|
||
|
|
config['TAG_EXPIRATION_OPTIONS'] = options
|
||
|
|
|
||
|
|
if expected_exception is not None:
|
||
|
|
with pytest.raises(ConfigValidationException) as cve:
|
||
|
|
TimeMachineValidator.validate(config, None, None)
|
||
|
|
assert str(cve.value) == str(expected_exception)
|
||
|
|
else:
|
||
|
|
TimeMachineValidator.validate(config, None, None)
|