def test_get_related_xml_values_with_no_matching_id(self):
"""
Test returning default value if id wasn't found in XML file.
"""
names = utils.get_related_xml_values([121])
self.assertEqual(len(names.keys()), 1)
self.assertEqual(names[121], 'User 121')
def test_get_related_xml_values_with_empty_items(self, etree_from_string):
"""
Test returning empty dict if ids list is also empty.
"""
etree_from_string.return_value = ''
names = utils.get_related_xml_values([])
self.assertDictEqual(names, {})
def users_view():
"""
Users listing for dropdown.
"""
data = get_related_xml_values(get_data().keys())
users = [{"user_id": user_id, "name": user_name} for user_id, user_name in data.items()]
return sorted(users, key=lambda x: x["name"], cmp=locale.strcoll)
def test_get_related_xml_values_processing_error(self, mock_etree):
"""
Test returning none if xml file is corrupted.
"""
data = utils.get_data()
mock_etree.return_value = 'wrong_string'
names = utils.get_related_xml_values(data.keys())
self.assertFalse(names)
def test_get_related_xml_values(self):
"""
Test returning proper list of names according to the ids list.
"""
data = utils.get_data()
names = utils.get_related_xml_values(data.keys())
self.assertEqual(len(names.keys()), 6)
self.assertEqual(names[10], 'Adrian K.')
self.assertEqual(names[13], 'Agata J.')
