def test_with_one_root_id(self): queryset = GroupElementYear.objects.all().select_related( 'child_branch__academic_year', 'child_leaf__academic_year', # [...] other fetch ) result = group_element_year.fetch_all_group_elements_in_tree( self.root, queryset) expected_result = { self.link_1.parent_id: [self.link_1, self.link_1_bis], self.link_2.parent_id: [self.link_2], self.link_3.parent_id: [self.link_3], self.link_4.parent_id: [self.link_4], } self.assertDictEqual(result, expected_result)
def _init_cache(self): return fetch_all_group_elements_in_tree(self.education_group_year, self.get_queryset()) or {}
def test_when_queryset_is_not_from_group_element_year_model(self): wrong_queryset_model = EducationGroupYear.objects.all() with self.assertRaises(AttributeError): group_element_year.fetch_all_group_elements_in_tree( self.root, wrong_queryset_model)
def _init_cache(self): return fetch_all_group_elements_in_tree(self.education_group_year, self.get_queryset()) or {}