def test_level_limit_set_only_filtered_levels_returned(self):
"""
Testing if level_limit is set the parent field queryset should filter and remove levels underneath.
"""
item_admin = ItemAdmin(TestItem, self.site)
item_admin.level_limit = 1
form = item_admin.get_form(request)
queryset = form.base_fields["parent"]._queryset
self.assertEqual(len(form.base_fields["parent"]._queryset), 1)
self.assertNotIn(self.menu_child, queryset)
def test_level_limit_not_set_all_levels_returned(self):
"""
Testing if level_limit is not set, all levels should be returned in the parent select.
"""
item_admin = ItemAdmin(TestItem, self.site)
form = item_admin.get_form(request)
queryset = form.base_fields["parent"]._queryset
self.assertEqual(len(queryset), 2)
self.assertIn(self.menu_parent, queryset)
self.assertIn(self.menu_child, queryset)
