def test_find_entities_by_person_with_multiple_person_entity_no_duplicate(self):
person = PersonFactory()
PersonEntityFactory(person=person, entity=self.sst_entity, with_child=True)
PersonEntityFactory(person=person, entity=self.agro_entity, with_child=False)
entities = person_entity.find_entities_by_person(person)
self.assertIsInstance(entities, list)
self.assertEqual(len(entities), 3)
def filter_by_attached_entities(person, entity_queryset):
entities_attached = person_entity.find_entities_by_person(person)
field_path = MAP_ENTITY_FIELD.get(entity_queryset.model)
if not field_path:
raise ObjectDoesNotExist
field_filter = "{}__in".format(field_path)
return entity_queryset.filter(**{field_filter: entities_attached})
def test_find_entities_by_person_with_child_true(self):
person = PersonFactory()
PersonEntityFactory(person=person,
entity=self.root_entity,
with_child=True)
entities = person_entity.find_entities_by_person(person)
self.assertIsInstance(entities, list)
self.assertEqual(len(entities), 8)
def test_find_entities_by_person_with_child_false(self):
person = PersonFactory()
PersonEntityFactory(person=person,
entity=self.root_entity,
with_child=False)
entities = person_entity.find_entities_by_person(person)
self.assertIsInstance(entities, list)
self.assertEqual(len(entities), 1) # We take only root, no child
def can_delete_learning_unit_year(person, learning_unit_year):
# Check person_entity linked
requirement_entity_version = entity_container_year.find_requirement_entity(
learning_unit_year.learning_container_year)
entities_linked = person_entity.find_entities_by_person(person)
if not requirement_entity_version or requirement_entity_version.entity not in entities_linked:
return False
return _can_delete_learning_unit_year_according_type(
person.user, learning_unit_year)
def test_find_entities_by_person_with_no_link(self):
person = PersonFactory()
entities = person_entity.find_entities_by_person(person)
self.assertIsInstance(entities, list)
self.assertFalse(entities)
def entities(self):
return find_entities_by_person(self)
