def _clean_attribute_initial_value(attribute_name, attribute_value):
clean_attribute_value = attribute_value
if attribute_name == "campus":
clean_attribute_value = campus.find_by_id(attribute_value)
elif attribute_name == "language":
clean_attribute_value = language.find_by_id(attribute_value)
return clean_attribute_value
def clean_attribute_initial_value(attribute_name, attribute_value):
# TODO : clean this function ; it could make up to 6 hits DB
clean_attribute_value = attribute_value
if attribute_name == "campus":
clean_attribute_value = campus.find_by_id(attribute_value)
elif attribute_name == "language":
clean_attribute_value = language.find_by_id(attribute_value)
elif attribute_name in ['requirement_entity', 'allocation_entity', 'additional_entity_1', 'additional_entity_2']:
clean_attribute_value = get_by_internal_id(attribute_value)
return clean_attribute_value
def test_find_by_id(self):
campus_id = self.campus.id
self.assertEqual(self.campus, campus.find_by_id(campus_id))
self.campus.delete()
self.assertIsNone(campus.find_by_id(campus_id))