def test_tree_item_admin_item_move(self): main_tree = Tree.objects.get(alias='main') admin = TreeItemAdmin(TreeItem, site) new_item_1 = TreeItem(title='title_1', sort_order=1, tree_id=main_tree.pk) new_item_1.save() new_item_2 = TreeItem(title='title_2', sort_order=2, tree_id=main_tree.pk) new_item_2.save() new_item_3 = TreeItem(title='title_3', sort_order=3, tree_id=main_tree.pk) new_item_3.save() admin.item_move(None, None, new_item_2.id, 'up') self.assertEqual(TreeItem.objects.get(pk=new_item_1.id).sort_order, 2) self.assertEqual(TreeItem.objects.get(pk=new_item_2.id).sort_order, 1) self.assertEqual(TreeItem.objects.get(pk=new_item_3.id).sort_order, 3) admin.item_move(None, None, new_item_1.id, 'down') self.assertEqual(TreeItem.objects.get(pk=new_item_1.id).sort_order, 3) self.assertEqual(TreeItem.objects.get(pk=new_item_2.id).sort_order, 1) self.assertEqual(TreeItem.objects.get(pk=new_item_3.id).sort_order, 2)
def test_tree_item_admin_save_model(self): main_tree = Tree.objects.get(alias='main') tree_item = TreeItem.objects.filter(tree__alias='main')[0] admin = TreeItemAdmin(TreeItem, site) admin.tree = main_tree admin.save_model(MockRequest(), tree_item, None, change=True) self.assertIs(tree_item.tree, admin.tree)
def test_tree_item_admin(self): admin = TreeItemAdmin(TreeItem, site) admin.tree = Tree.objects.get(pk=1) form = admin.get_form(MockRequest()) self.assertEqual(len(form.known_url_names), 3) self.assertIn('articles_list', form.known_url_names) self.assertIn('articles_detailed', form.known_url_names) self.assertIn('url', form.known_url_names)
def test_tree_item_admin(self): admin = TreeItemAdmin(TreeItem, site) admin.tree = Tree.objects.get(alias='main') form = admin.get_form(MockRequest()) self.assertEqual(len(form.known_url_names), 3) self.assertIn('articles_list', form.known_url_names) self.assertIn('articles_detailed', form.known_url_names) self.assertIn('url', form.known_url_names)
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 get_item_admin(): from sitetree.models import TreeItem from sitetree.admin import TreeItemAdmin admin = TreeItemAdmin(TreeItem, site) return admin
def test_tree_item_admin_save_model(self): admin = TreeItemAdmin(TreeItem, site) tree_item = TreeItem.objects.get(pk=1) admin.tree = Tree.objects.get(pk=1) admin.save_model(MockRequest(), tree_item, None, change=True) self.assertIs(tree_item.tree, admin.tree)
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')