コード例 #1
0
 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)
コード例 #2
0
 def _init_cache(self):
     return fetch_all_group_elements_in_tree(self.education_group_year,
                                             self.get_queryset()) or {}
コード例 #3
0
 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)
コード例 #4
0
 def _init_cache(self):
     return fetch_all_group_elements_in_tree(self.education_group_year, self.get_queryset()) or {}