def test_expand_restriction_format(self):
string_restriction = 'settings.some_attribute.value != true'
dict_restriction_long_format = {
'condition': 'settings.some_attribute.value != true',
'message': 'Another attribute required'
}
dict_restriction_short_format = {
'settings.some_attribute.value != true':
'Another attribute required'
}
result = {
'action': 'disable',
'condition': 'settings.some_attribute.value != true',
}
invalid_format = ['some_condition']
# check string format
self.assertDictEqual(
RestrictionMixin._expand_restriction(
string_restriction), result)
result['message'] = 'Another attribute required'
# check long format
self.assertDictEqual(
RestrictionMixin._expand_restriction(
dict_restriction_long_format), result)
# check short format
self.assertDictEqual(
RestrictionMixin._expand_restriction(
dict_restriction_short_format), result)
# check invalid format
self.assertRaises(
errors.InvalidData,
RestrictionMixin._expand_restriction,
invalid_format)
