def test_tree_item_admin_get_tree(self): main_tree = Tree.objects.get(alias='main') main_tree_item = TreeItem.objects.filter(tree__alias='main')[0] admin = TreeItemAdmin(TreeItem, site) tree = admin.get_tree(MockRequest(), main_tree.pk) self.assertEqual(tree.alias, 'main') tree = admin.get_tree(MockRequest(), None, main_tree_item.pk) self.assertEqual(tree.alias, 'main')
def test_tree_item_admin_get_tree(self): admin = TreeItemAdmin(TreeItem, site) tree = admin.get_tree(MockRequest(), 1) self.assertEqual(tree.alias, 'main') tree = admin.get_tree(MockRequest(), None, 1) self.assertEqual(tree.alias, 'main')