diff options
Diffstat (limited to 'tests/modules/test_helpers.py')
-rw-r--r-- | tests/modules/test_helpers.py | 117 |
1 files changed, 61 insertions, 56 deletions
diff --git a/tests/modules/test_helpers.py b/tests/modules/test_helpers.py index cdfc4d74..f0cebb41 100644 --- a/tests/modules/test_helpers.py +++ b/tests/modules/test_helpers.py @@ -24,86 +24,91 @@ from tests import base class TestCaseTestHelpers(base.BaseTestCase): - def test_convert_mapping_to_xml(self): """ Tests the test_convert_mapping_to_xml_fail_required function """ # Test default values - default_root = XML.Element('testdefault') + default_root = XML.Element("testdefault") default_data = yaml.load("string: hello") - default_mappings = [('default-string', 'defaultString', 'default')] + default_mappings = [("default-string", "defaultString", "default")] convert_mapping_to_xml( - default_root, - default_data, - default_mappings, - fail_required=True) - result = default_root.find('defaultString').text - self.assertThat(result, Equals('default')) + default_root, default_data, default_mappings, fail_required=True + ) + result = default_root.find("defaultString").text + self.assertThat(result, Equals("default")) # Test user input - user_input_root = XML.Element('testUserInput') + user_input_root = XML.Element("testUserInput") user_input_data = yaml.load("user-input-string: hello") - user_input_mappings = [('user-input-string', 'userInputString', - 'user-input')] + user_input_mappings = [("user-input-string", "userInputString", "user-input")] convert_mapping_to_xml( - user_input_root, - user_input_data, - user_input_mappings, - fail_required=True) - result = user_input_root.find('userInputString').text - self.assertThat(result, Equals('hello')) + user_input_root, user_input_data, user_input_mappings, fail_required=True + ) + result = user_input_root.find("userInputString").text + self.assertThat(result, Equals("hello")) # Test missing required input - required_root = XML.Element('testrequired') + required_root = XML.Element("testrequired") required_data = yaml.load("string: hello") - required_mappings = [('required-string', 'requiredString', None)] + required_mappings = [("required-string", "requiredString", None)] - self.assertRaises(MissingAttributeError, - convert_mapping_to_xml, - required_root, - required_data, - required_mappings, - fail_required=True) + self.assertRaises( + MissingAttributeError, + convert_mapping_to_xml, + required_root, + required_data, + required_mappings, + fail_required=True, + ) # Test invalid user input for list - user_input_root = XML.Element('testUserInput') + user_input_root = XML.Element("testUserInput") user_input_data = yaml.load("user-input-string: bye") - valid_inputs = ['hello'] - user_input_mappings = [('user-input-string', 'userInputString', - 'user-input', valid_inputs)] - - self.assertRaises(InvalidAttributeError, - convert_mapping_to_xml, - user_input_root, - user_input_data, - user_input_mappings) + valid_inputs = ["hello"] + user_input_mappings = [ + ("user-input-string", "userInputString", "user-input", valid_inputs) + ] + + self.assertRaises( + InvalidAttributeError, + convert_mapping_to_xml, + user_input_root, + user_input_data, + user_input_mappings, + ) # Test invalid user input for dict - user_input_root = XML.Element('testUserInput') + user_input_root = XML.Element("testUserInput") user_input_data = yaml.load("user-input-string: later") - valid_inputs = {'hello': 'world'} - user_input_mappings = [('user-input-string', 'userInputString', - 'user-input', valid_inputs)] - - self.assertRaises(InvalidAttributeError, - convert_mapping_to_xml, - user_input_root, - user_input_data, - user_input_mappings) + valid_inputs = {"hello": "world"} + user_input_mappings = [ + ("user-input-string", "userInputString", "user-input", valid_inputs) + ] + + self.assertRaises( + InvalidAttributeError, + convert_mapping_to_xml, + user_input_root, + user_input_data, + user_input_mappings, + ) # Test invalid key for dict - user_input_root = XML.Element('testUserInput') + user_input_root = XML.Element("testUserInput") user_input_data = yaml.load("user-input-string: world") - valid_inputs = {'hello': 'world'} - user_input_mappings = [('user-input-string', 'userInputString', - 'user-input', valid_inputs)] - - self.assertRaises(InvalidAttributeError, - convert_mapping_to_xml, - user_input_root, - user_input_data, - user_input_mappings) + valid_inputs = {"hello": "world"} + user_input_mappings = [ + ("user-input-string", "userInputString", "user-input", valid_inputs) + ] + + self.assertRaises( + InvalidAttributeError, + convert_mapping_to_xml, + user_input_root, + user_input_data, + user_input_mappings, + ) |