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)