def test_case_one_egy_one_parent_no_entity_on_child(self):
education_group_year_child = EducationGroupYearFactory()
education_group_year_parent = EducationGroupYearFactory()
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(
academic_year=self.academic_year)
education_group_year_parent1 = EducationGroupYearFactory(
academic_year=self.academic_year)
GroupElementYearFactory(parent=education_group_year_parent1,
child_branch=education_group_year_child)
education_group_year_parent2 = EducationGroupYearFactory(
academic_year=self.academic_year)
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,
])
