def test_convert_must_raise_if_template_name_not_found(
self, template_from_dict_mock, is_valid_templates_dict):
policy_templates_dict = {"Templates": {"key1": "value1"}}
parameter_values = {"a": "b"}
processor = PolicyTemplatesProcessor(policy_templates_dict)
with self.assertRaises(TemplateNotFoundException):
processor.convert("key2", parameter_values)
def test_convert_must_raise_if_template_name_not_found(self, template_from_dict_mock, is_valid_templates_dict):
policy_templates_dict = {
"Templates": {
"key1": "value1"
}
}
parameter_values = {
"a": "b"
}
processor = PolicyTemplatesProcessor(policy_templates_dict)
with self.assertRaises(TemplateNotFoundException):
processor.convert("key2", parameter_values)
def test_convert_must_bubble_exceptions(self, template_from_dict_mock,
is_valid_templates_dict):
policy_templates_dict = {"Templates": {"key1": "value1"}}
parameter_values = {"a": "b"}
template_obj_mock = Mock()
template_from_dict_mock.return_value = template_obj_mock
template_obj_mock.to_statement.side_effect = TypeError
processor = PolicyTemplatesProcessor(policy_templates_dict)
with self.assertRaises(TypeError):
processor.convert("key1", parameter_values)
def test_convert_must_bubble_exceptions(self, template_from_dict_mock, is_valid_templates_dict):
policy_templates_dict = {
"Templates": {
"key1": "value1"
}
}
parameter_values = {
"a": "b"
}
template_obj_mock = Mock()
template_from_dict_mock.return_value = template_obj_mock
template_obj_mock.to_statement.side_effect = TypeError
processor = PolicyTemplatesProcessor(policy_templates_dict)
with self.assertRaises(TypeError):
processor.convert("key1", parameter_values)
def test_convert_must_work_for_known_template_names(
self, template_from_dict_mock, is_valid_templates_dict):
policy_templates_dict = {"Templates": {"key1": "value1"}}
parameter_values = {"a": "b"}
template_obj_mock = Mock()
template_from_dict_mock.return_value = template_obj_mock
expected = {"stmt": "result"}
template_obj_mock.to_statement.return_value = expected
processor = PolicyTemplatesProcessor(policy_templates_dict)
result = processor.convert("key1", parameter_values)
self.assertEquals(result, expected)
template_obj_mock.to_statement.assert_called_once_with(
parameter_values)
def test_convert_must_work_for_known_template_names(self, template_from_dict_mock, is_valid_templates_dict):
policy_templates_dict = {
"Templates": {
"key1": "value1"
}
}
parameter_values = {
"a": "b"
}
template_obj_mock = Mock()
template_from_dict_mock.return_value = template_obj_mock
expected = {"stmt": "result"}
template_obj_mock.to_statement.return_value = expected
processor = PolicyTemplatesProcessor(policy_templates_dict)
result = processor.convert("key1", parameter_values)
self.assertEquals(result, expected)
template_obj_mock.to_statement.assert_called_once_with(parameter_values)
