def test_case_one_egy_and_parent(self): education_group_year_child = EducationGroupYearFactory() education_group_year_parent = EducationGroupYearFactory() GroupElementYearFactory(parent=education_group_year_parent, child_branch=education_group_year_child) self.assertEquals( get_education_group_year_eligible_management_entities( education_group_year_child), [education_group_year_child.management_entity]) self.assertEquals( get_education_group_year_eligible_management_entities( education_group_year_parent), [education_group_year_parent.management_entity])
def test_case_one_egy(self): education_group_year = EducationGroupYearFactory( academic_year=self.academic_year) self.assertEquals( get_education_group_year_eligible_management_entities( education_group_year), [education_group_year.management_entity])
def test_case_one_egy_one_parent_no_entity_on_child(self): education_group_year_child = EducationGroupYearFactory( academic_year=self.academic_year) education_group_year_parent = EducationGroupYearFactory( academic_year=self.academic_year) GroupElementYearFactory(parent=education_group_year_parent, child_branch=education_group_year_child) education_group_year_child.management_entity = None education_group_year_child.save() self.assertEquals( get_education_group_year_eligible_management_entities( education_group_year_child), [education_group_year_parent.management_entity])
def test_case_one_egy_two_parent_no_entity_on_child(self): education_group_year_child = EducationGroupYearFactory() education_group_year_parent1 = EducationGroupYearFactory() GroupElementYearFactory(parent=education_group_year_parent1, child_branch=education_group_year_child) education_group_year_parent2 = EducationGroupYearFactory() GroupElementYearFactory(parent=education_group_year_parent2, child_branch=education_group_year_child) education_group_year_child.management_entity = None education_group_year_child.save() self.assertCountEqual( get_education_group_year_eligible_management_entities( education_group_year_child), [ education_group_year_parent1.management_entity, education_group_year_parent2.management_entity, ])
def test_case_complex_hierarchy(self): education_group_year_child = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_child.management_entity, acronym="CHILD") education_group_year_parent1 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent1.management_entity, acronym="PARENT1") GroupElementYearFactory(parent=education_group_year_parent1, child_branch=education_group_year_child) education_group_year_parent2 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent2.management_entity, acronym="PARENT2") GroupElementYearFactory(parent=education_group_year_parent2, child_branch=education_group_year_child) education_group_year_parent3 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent3.management_entity, acronym="PARENT3") GroupElementYearFactory(parent=education_group_year_parent3, child_branch=education_group_year_parent1) education_group_year_parent4 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent4.management_entity, acronym="PARENT4") GroupElementYearFactory(parent=education_group_year_parent4, child_branch=education_group_year_parent1) education_group_year_parent5 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent5.management_entity, acronym="PARENT5") GroupElementYearFactory(parent=education_group_year_parent5, child_branch=education_group_year_child) GroupElementYearFactory(parent=education_group_year_parent5, child_branch=education_group_year_parent2) education_group_year_parent6 = EducationGroupYearFactory() EntityVersionFactory( entity=education_group_year_parent6.management_entity, acronym="PARENT6") GroupElementYearFactory(parent=education_group_year_parent6, child_branch=education_group_year_parent5) education_group_year_child.management_entity = None education_group_year_child.save() education_group_year_parent1.management_entity = None education_group_year_parent1.save() education_group_year_parent5.management_entity = None education_group_year_parent5.save() self.assertCountEqual( get_education_group_year_eligible_management_entities( education_group_year_child), [ education_group_year_parent2.management_entity, education_group_year_parent3.management_entity, education_group_year_parent4.management_entity, education_group_year_parent6.management_entity, ])