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.')