def test_get_subclass_queries_helper_induction(self): """ Test the recursive case of get_subclass_queries_helper, where there are descendants """ r, = [x for x in get_subclass_relations(Page) if x.var_name == 'richtextpage'] result = get_subclass_queries_helper(r, 'test__') self.assertIn('test__richtextpage__foopage', result) self.assertIn('test__richtextpage__barpage', result)
def test_get_subclass_queries_helper_base(self): """ Test base case of get_subclass_queries_helper """ # We pass it a relation of a class that has no descendents # We use "Link" for this r, = [x for x in get_subclass_relations(Page) if x.var_name == 'link'] result = get_subclass_queries_helper(r, 'test__') self.assertEqual(result, ['test__link'])